add multipart filememe

ArtisanCloud · Jan 7, 2023 · 8dfdf3e · 8dfdf3e
1 parent 91252d2
commit 8dfdf3e
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/http/contract/dataflowInterface.go b/http/contract/dataflowInterface.go
@@ -52,6 +52,7 @@ type ResponseHelper interface {
 type MultipartDfInterface interface {
 	Boundary(b string) MultipartDfInterface
 	FileByPath(fieldName string, filePath string) MultipartDfInterface
+	FileMem(fieldName string, fileName string, reader io.Reader) MultipartDfInterface
 	Part(header textproto.MIMEHeader, reader io.Reader) MultipartDfInterface
 	FieldValue(fieldName string, value string) MultipartDfInterface
 	Field(fieldName string, reader io.Reader) MultipartDfInterface

diff --git a/http/dataflow/mutlipart.go b/http/dataflow/mutlipart.go
@@ -55,6 +55,23 @@ func (m *MultipartDf) FileByPath(fieldName string, filePath string) contract.Mul
 	return m
 }
 
+func (m *MultipartDf) FileMem(fieldName string, fileName string, reader io.Reader) contract.MultipartDfInterface {
+	writer, err := m.mWriter.CreateFormFile(fieldName, fileName)
+	if err != nil {
+		m.errs = append(m.errs, errors.Wrap(err, "create file part failed"))
+	}
+	var buf bytes.Buffer
+	_, err = buf.ReadFrom(reader)
+	if err != nil {
+		m.errs = append(m.errs, errors.Wrap(err, "create file part failed"))
+	}
+	_, err = buf.WriteTo(writer)
+	if err != nil {
+		m.errs = append(m.errs, errors.Wrap(err, "create file part failed"))
+	}
+	return m
+}
+
 func (m *MultipartDf) Part(header textproto.MIMEHeader, reader io.Reader) contract.MultipartDfInterface {
 	writer, err := m.mWriter.CreatePart(header)
 	if err != nil {