feat(s3): set content type from stream when uploading (#4460)

Co-authored-by: guopeilun <guopl@flatincbr.com>
AlistGo · May 24, 2023 · 6b97b4e · 6b97b4e
1 parent 645c10c
commit 6b97b4e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/drivers/s3/driver.go b/drivers/s3/driver.go
@@ -136,11 +136,13 @@ func (d *S3) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreame
 		uploader.PartSize = stream.GetSize() / (s3manager.MaxUploadParts - 1)
 	}
 	key := getKey(stdpath.Join(dstDir.GetPath(), stream.GetName()), false)
+	contentType := stream.GetMimetype()
 	log.Debugln("key:", key)
 	input := &s3manager.UploadInput{
-		Bucket: &d.Bucket,
-		Key:    &key,
-		Body:   stream,
+		Bucket:      &d.Bucket,
+		Key:         &key,
+		Body:        stream,
+		ContentType: &contentType,
 	}
 	_, err := uploader.UploadWithContext(ctx, input)
 	return err