'上传兼容文件[]byte化'

README.md

@@ -5,6 +5,7 @@
 对阿里云oss-golang sdk 进行上传、删除的简单封装，便于使用
 上传使用字节流，内自带 *os.File | *multipart.FileHeader 文件类型转字节流，
 以及使用字符串路径 例如：./test.png 。测试案例oss_test.go中有不同操作的案例
+新版兼容文件[]byte化上传
 
 #install 
 go get github.com/cowardmrx/go_aliyun_oss
@@ -23,8 +24,8 @@ go get github.com/cowardmrx/go_aliyun_oss
 
 	ossClient := ossConfig.CreateOssConnect()
     
-    //put 方法返回完整的oss 可访问地址
-	uri := ossClient.Put("logo/","./File/3HaqWaOzJWD86DDvZD9Pmn9VUEOBOBbuWackGOXb (2).jpeg")
+    //put 方法返回完整的oss 可访问地址 第三个参数 如果文件不是[]byte数组的可以直接传递空，如果是[]byte也可以是空，会默认给出一个png的文件类型，建议实际中给出文件类型
+	uri := ossClient.Put("logo/","./File/3HaqWaOzJWD86DDvZD9Pmn9VUEOBOBbuWackGOXb (2).jpeg","")
     fmt.Println(uri)
 
     //HasExists 方法返回一个bool值 true-存在 false-不存在

file.go

@@ -31,6 +31,7 @@ func (ossFile *OssFile) FileTypeTransForm() (*OssFile,error) {
 
 	switch ossFile.File.(type) {
 	case *os.File:
+
 		ossFile.FileByte,err = ioutil.ReadAll(ossFile.File.(*os.File))
 
 		if err != nil {
@@ -39,6 +40,8 @@ func (ossFile *OssFile) FileTypeTransForm() (*OssFile,error) {
 
 		_,ossFile.FileOldName = filepath.Split(ossFile.File.(*os.File).Name())
 
+		break
+
 	case *multipart.FileHeader:
 
 		fileResources,err := ossFile.File.(*multipart.FileHeader).Open()
@@ -57,6 +60,8 @@ func (ossFile *OssFile) FileTypeTransForm() (*OssFile,error) {
 
 		ossFile.FileOldName = ossFile.File.(*multipart.FileHeader).Filename
 
+		break
+
 	case string:
 		newFile,err := os.Open(ossFile.File.(string))
 
@@ -70,6 +75,21 @@ func (ossFile *OssFile) FileTypeTransForm() (*OssFile,error) {
 
 		_,ossFile.FileOldName = filepath.Split(newFile.Name())
 
+		break
+
+	// 支持[]byte数组传递 因为无法解析文件类型 默认直接给出文件类型为.png
+	case []byte:
+		ossFile.FileByte = ossFile.File.([]byte)
+
+		//判断是否指定了文件的类型 如果没有指定默认为png格式
+		if ossFile.FileType == "" || len(ossFile.FileType) <= 0 {
+			ossFile.FileOldName = uuid.NewV4().String() + ".png"
+		} else {
+			ossFile.FileOldName = uuid.NewV4().String() + ossFile.FileType
+		}
+
+		break
+
 	default:
 		fmt.Println(reflect.TypeOf(ossFile.File))
 		panic("file type is not support" )
@@ -83,12 +103,16 @@ func (ossFile *OssFile) FileTypeTransForm() (*OssFile,error) {
 //split file type and generate file name
 //截取文件类型
 func (ossFile *OssFile) GetFileType() *OssFile {
+
+	// 当没有传递文件类型时去文件名中截取出文件类型
+	if ossFile.FileType == "" || len(ossFile.FileType) <= 0 {
 		//from oldFileName split file type
 		fileTypeSufIndex := strings.Index(ossFile.FileOldName,".")
 
 		fileType := ossFile.FileOldName[fileTypeSufIndex:]
 
 		ossFile.FileType = fileType
+	}
 
 	//generate only file name
 	ossFile.FileNewName = uuid.NewV5(uuid.NewV4(),ossFile.FileOldName).String() + ossFile.FileType

oss_operation.go

@@ -15,11 +15,12 @@ type AliOssClient struct {
 //params:  ossDir string  `oss dir [要推送到的oss目录]`  example: test/20201121/
 //params:  file interface `upload file resource [文件资源]`
 //return string  `oss file accessible uri [可访问地址]`
-func (client *AliOssClient) Put(ossDir string, file interface{}) string {
+func (client *AliOssClient) Put(ossDir string, file interface{},fileType string) string {
 	//file to []byte
 	//文件转字节流
 	uploadFile := &OssFile{
 		File: file,
+		FileType: fileType,
 	}
 
 	ossFile,err := uploadFile.FileTypeTransForm()

oss_test.go

@@ -1,7 +1,9 @@
 package go_aliyun_oss
 
 import (
+	"encoding/base64"
 	"fmt"
+	"io/ioutil"
 	"testing"
 )
 
@@ -15,11 +17,35 @@ func TestPut(t *testing.T)  {
 
 	client := ossConfig.CreateOssConnect()
 
-	uri := client.Put("logo/","./File/3HaqWaOzJWD86DDvZD9Pmn9VUEOBOBbuWackGOXb (2).jpeg")
+	uri := client.Put("logo/","./File/3HaqWaOzJWD86DDvZD9Pmn9VUEOBOBbuWackGOXb (2).jpeg",".png")
 
 	fmt.Println(uri)
 }
 
+func TestPutBase64(t *testing.T)  {
+	ossConfig := &AliOssConfig{
+		EndPoint: "",
+		AccessKeyId: "",
+		AccessKeySecret: "",
+		BucketName: "",
+	}
+
+	client := ossConfig.CreateOssConnect()
+
+	// 读取base
+	file,_ := ioutil.ReadFile("./File/3HaqWaOzJWD86DDvZD9Pmn9VUEOBOBbuWackGOXb (2).jpeg")
+
+
+	fileStr := base64.StdEncoding.EncodeToString(file)
+
+	bat,_ := base64.StdEncoding.DecodeString(fileStr)
+
+
+	uri := client.Put("logo/",bat,".png")
+	//
+	fmt.Println(uri)
+}
+
 func TestExists(t *testing.T)  {
 	ossConfig := &AliOssConfig{
 		EndPoint: "oss-cn-shenzhen.aliyuncs.com",