adding dapr-http-read-buffer-size parameter to cli

dapr · Jun 3, 2022 · 6ba84e5 · 6ba84e5
1 parent 9bfc538
commit 6ba84e5
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 1 deletion.
diff --git a/cmd/run.go b/cmd/run.go
@@ -46,6 +46,7 @@ var (
 	appSSL             bool
 	metricsPort        int
 	maxRequestBodySize int
+	readBufferSize     int
 	unixDomainSocket   string
 	enableAPILogging   bool
 )
@@ -115,6 +116,7 @@ dapr run --app-id myapp --app-port 3000 --app-protocol grpc -- go run main.go
 			AppSSL:             appSSL,
 			MetricsPort:        metricsPort,
 			MaxRequestBodySize: maxRequestBodySize,
+			HTTPReadBufferSize: readBufferSize,
 			UnixDomainSocket:   unixDomainSocket,
 			EnableAPILogging:   enableAPILogging,
 		})
@@ -364,6 +366,7 @@ func init() {
 	RunCmd.Flags().IntVarP(&metricsPort, "metrics-port", "M", -1, "The port of metrics on dapr")
 	RunCmd.Flags().BoolP("help", "h", false, "Print this help message")
 	RunCmd.Flags().IntVarP(&maxRequestBodySize, "dapr-http-max-request-size", "", -1, "Max size of request body in MB")
+	RunCmd.Flags().IntVarP(&readBufferSize, "dapr-http-read-buffer-size", "", -1, "HTTP header read buffer in KB")
 	RunCmd.Flags().StringVarP(&unixDomainSocket, "unix-domain-socket", "u", "", "Path to a unix domain socket dir. If specified, Dapr API servers will use Unix Domain Sockets")
 	RunCmd.Flags().BoolVar(&enableAPILogging, "enable-api-logging", false, "Log API calls at INFO verbosity. Valid values are: true or false")
 

diff --git a/install/install.ps1 b/install/install.ps1
@@ -12,7 +12,7 @@
 # ------------------------------------------------------------
 param (
     [string]$Version,
-    [string]$DaprRoot = "$Env:SystemDrive\dapr",
+    [string]$DaprRoot = "c:\dapr",
     [string]$DaprReleaseJsonUrl = "",
     [scriptblock]$CustomAssetFactory = $null
 )

diff --git a/pkg/standalone/list.go b/pkg/standalone/list.go
@@ -51,6 +51,7 @@ type runData struct {
 	appCmd             string
 	enableMetrics      bool
 	maxRequestBodySize int
+	httpReadBufferSize int
 }
 
 func (d *daprProcess) List() ([]ListOutput, error) {
@@ -134,6 +135,11 @@ func List() ([]ListOutput, error) {
 				continue
 			}
 
+			httpReadBufferSize, err := strconv.Atoi(argumentsMap["--dapr-http-read-buffer-size"])
+			if err != nil {
+				continue
+			}
+
 			run := runData{
 				cliPID:             cliPID,
 				sidecarPID:         proc.Pid(),
@@ -144,6 +150,7 @@ func List() ([]ListOutput, error) {
 				appCmd:             appCmd,
 				enableMetrics:      enableMetrics,
 				maxRequestBodySize: maxRequestBodySize,
+				httpReadBufferSize: httpReadBufferSize,
 			}
 
 			cliToSidecarMap[cliPID] = &run

diff --git a/pkg/standalone/run.go b/pkg/standalone/run.go
@@ -52,6 +52,7 @@ type RunConfig struct {
 	AppSSL             bool   `arg:"app-ssl"`
 	MetricsPort        int    `env:"DAPR_METRICS_PORT" arg:"metrics-port"`
 	MaxRequestBodySize int    `arg:"dapr-http-max-request-size"`
+	HTTPReadBufferSize int    `arg:"dapr-http-read-buffer-size"`
 	UnixDomainSocket   string `arg:"unix-domain-socket"`
 	EnableAPILogging   bool   `arg:"enable-api-logging"`
 }
@@ -158,6 +159,10 @@ func (config *RunConfig) validate() error {
 		config.MaxRequestBodySize = -1
 	}
 
+	if config.HTTPReadBufferSize < 0 {
+		config.HTTPReadBufferSize = -1
+	}
+
 	err = config.validatePlacementHostAddr()
 	if err != nil {
 		return err

diff --git a/pkg/standalone/run_test.go b/pkg/standalone/run_test.go
@@ -90,6 +90,7 @@ func assertCommonArgs(t *testing.T, basicConfig *RunConfig, output *RunOutput) {
 	assertArgumentEqual(t, "app-ssl", "", output.DaprCMD.Args)
 	assertArgumentEqual(t, "metrics-port", "9001", output.DaprCMD.Args)
 	assertArgumentEqual(t, "dapr-http-max-request-size", "-1", output.DaprCMD.Args)
+	assertArgumentEqual(t, "dapr-http-read-buffer-size", "-1", output.DaprCMD.Args)
 }
 
 func assertAppEnv(t *testing.T, config *RunConfig, output *RunOutput) {
@@ -148,6 +149,7 @@ func TestRun(t *testing.T) {
 		AppSSL:             true,
 		MetricsPort:        9001,
 		MaxRequestBodySize: -1,
+		HTTPReadBufferSize: -1,
 		EnableAPILogging:   true,
 	}