fix multipart last part boundary

ArtisanCloud · Jan 7, 2023 · bf9353d · bf9353d
1 parent d56b3ad
commit bf9353d
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/http/contract/dataflowInterface.go b/http/contract/dataflowInterface.go
@@ -56,6 +56,7 @@ type MultipartDfInterface interface {
 	Part(header textproto.MIMEHeader, reader io.Reader) MultipartDfInterface
 	FieldValue(fieldName string, value string) MultipartDfInterface
 	Field(fieldName string, reader io.Reader) MultipartDfInterface
+	Close() error
 	GetBoundary() string
 	GetReader() io.Reader
 	GetContentType() string

diff --git a/http/dataflow/dataflow.go b/http/dataflow/dataflow.go
@@ -220,7 +220,12 @@ func (d *Dataflow) Xml(xmlAny interface{}) contract.RequestDataflowInterface {
 func (d *Dataflow) Multipart(multipartDf func(multipart contract.MultipartDfInterface)) contract.RequestDataflowInterface {
 	multipart := NewMultipartHelper()
 	multipartDf(multipart)
-	err := multipart.Err()
+	err := multipart.Close()
+	if err != nil {
+		d.err = append(d.err, err)
+		return d
+	}
+	err = multipart.Err()
 	if err != nil {
 		d.err = append(d.err, err)
 		return d

diff --git a/http/dataflow/mutlipart.go b/http/dataflow/mutlipart.go
@@ -120,6 +120,10 @@ func (m *MultipartDf) Field(fieldName string, reader io.Reader) contract.Multipa
 	return m
 }
 
+func (m *MultipartDf) Close() error {
+	return m.mWriter.Close()
+}
+
 func (m *MultipartDf) GetBoundary() string {
 	return m.mWriter.Boundary()
 }