diff --git a/executor/http_executor.go b/executor/http_executor.go index e032b3d..3e36ebb 100644 --- a/executor/http_executor.go +++ b/executor/http_executor.go @@ -175,7 +175,7 @@ func (e HttpExecutor) progressReader(text string, completedText string, reader i } func (e HttpExecutor) writeMultipartBody(bodyWriter *io.PipeWriter, parameters []ExecutionParameter, errorChan chan error) (string, int64) { - contentLength := e.calculateMultipartSize(parameters) + multipartSize := e.calculateMultipartSize(parameters) formWriter := multipart.NewWriter(bodyWriter) go func() { defer bodyWriter.Close() @@ -186,7 +186,7 @@ func (e HttpExecutor) writeMultipartBody(bodyWriter *io.PipeWriter, parameters [ return } }() - return formWriter.FormDataContentType(), contentLength + return formWriter.FormDataContentType(), multipartSize } func (e HttpExecutor) writeInputBody(bodyWriter *io.PipeWriter, input utils.Stream, errorChan chan error) { @@ -230,30 +230,39 @@ func (e HttpExecutor) writeJsonBody(bodyWriter *io.PipeWriter, parameters []Exec }() } -func (e HttpExecutor) writeBody(context ExecutionContext, errorChan chan error) (io.Reader, string, int64) { +func (e HttpExecutor) writeBody(context ExecutionContext, errorChan chan error) (io.Reader, string, int64, int64) { if context.Input != nil { reader, writer := io.Pipe() e.writeInputBody(writer, context.Input, errorChan) - return reader, context.ContentType, -1 + contentLength, _ := context.Input.Size() + return reader, context.ContentType, contentLength, contentLength } formParameters := context.Parameters.Form() if len(formParameters) > 0 { reader, writer := io.Pipe() - contentType, contentLength := e.writeMultipartBody(writer, formParameters, errorChan) - return reader, contentType, contentLength + contentType, multipartSize := e.writeMultipartBody(writer, formParameters, errorChan) + return reader, contentType, -1, multipartSize } bodyParameters := context.Parameters.Body() if len(bodyParameters) > 0 && context.ContentType == "application/x-www-form-urlencoded" { reader, writer := io.Pipe() e.writeUrlEncodedBody(writer, bodyParameters, errorChan) - return reader, context.ContentType, -1 + return reader, context.ContentType, -1, -1 } if len(bodyParameters) > 0 { reader, writer := io.Pipe() e.writeJsonBody(writer, bodyParameters, errorChan) - return reader, context.ContentType, -1 + return reader, context.ContentType, -1, -1 } - return bytes.NewReader([]byte{}), context.ContentType, -1 + return bytes.NewReader([]byte{}), context.ContentType, -1, -1 +} + +func (e HttpExecutor) contentLength(context ExecutionContext) int64 { + if context.Input == nil { + return -1 + } + contentLength, _ := context.Input.Size() + return contentLength } func (e HttpExecutor) send(client *http.Client, request *http.Request, errorChan chan error) (*http.Response, error) { @@ -306,9 +315,9 @@ func (e HttpExecutor) call(context ExecutionContext, writer output.OutputWriter, return err } requestError := make(chan error) - bodyReader, contentType, contentLength := e.writeBody(context, requestError) + bodyReader, contentType, contentLength, size := e.writeBody(context, requestError) uploadBar := utils.NewProgressBar(logger) - uploadReader := e.progressReader("uploading...", "completing ", bodyReader, contentLength, uploadBar) + uploadReader := e.progressReader("uploading...", "completing ", bodyReader, size, uploadBar) defer uploadBar.Remove() request, err := http.NewRequest(context.Method, uri.String(), uploadReader) if err != nil { @@ -317,6 +326,9 @@ func (e HttpExecutor) call(context ExecutionContext, writer output.OutputWriter, if contentType != "" { request.Header.Add("Content-Type", contentType) } + if contentLength != -1 { + request.ContentLength = contentLength + } e.addHeaders(request, context.Parameters.Header()) auth, err := e.executeAuthenticators(context.AuthConfig, context.IdentityUri, context.Debug, context.Insecure, request) if err != nil { diff --git a/test/execution_test.go b/test/execution_test.go index 9ea99ac..05122fc 100644 --- a/test/execution_test.go +++ b/test/execution_test.go @@ -1260,6 +1260,10 @@ paths: if contentType != "application/octet-stream" { t.Errorf("Content-Type is not application/octet-stream, got: %v", contentType) } + contentLength := result.RequestHeader["content-length"] + if contentLength != "11" { + t.Errorf("Content-Length is not 11, got: %v", contentLength) + } if result.RequestBody != "hello-world" { t.Errorf("Request body is not as expected, got: %v", result.RequestBody) }