prdUploader/pkg/fileserver/local/local.go

package fileLocal

import (
	"bytes"
	"encoding/json"
	"epur-pay/pkg/logger"
	"fmt"
	"io"
	"mime/multipart"
	"net/http"
	"reflect"
	"time"
)

type Local struct {
	AccessKeyId     string
	AccessKeySecret string
	BucketName      string
	EndPoint        string
	Access          string
	RegionId        string
}

type ReturnUpload struct {
	Code    int64  `json:"code"`
	Message string `json:"message"`
	Data    struct {
		FileUrl string `json:"fileUrl"`
	} `json:"data"`
}

func New(accessKeyId, accessKeySecret, bucketName, endPoint string, access string, regionId string) (*Local, error) {
	instance := &Local{
		AccessKeyId:     accessKeyId,
		AccessKeySecret: accessKeySecret,
		BucketName:      bucketName,
		EndPoint:        endPoint,
		Access:          access,
		RegionId:        regionId,
	}
	return instance, nil
}

func (this *Local) ReadUrl(filePath string) string {
	return this.url(filePath)
}

func (this *Local) url(fileName string) string {
	return fmt.Sprintf("http://%s/%s", this.Access, fileName)
}

func (this *Local) Put(fileName string, contentType string, data interface{}) string {
	var ossData io.Reader
	typeOf := reflect.TypeOf(data).Kind()
	switch typeOf {
	case reflect.Slice:
		ossData = bytes.NewBuffer(data.([]byte))
	case reflect.Struct:
		ossData = data.(io.Reader)
	case reflect.Ptr:
		a1 := reflect.ValueOf(data).Interface()
		d := a1.([]uint8)
		ossData = bytes.NewBuffer(d)
	default:
		panic("数据不合法--!")
	}

	var buffer bytes.Buffer
	writer := multipart.NewWriter(&buffer)
	writer.WriteField("path", "test/file")
	// 创建一个新的表单文件字段，并将文件内容的io.Reader传递给它
	fileField, err := writer.CreateFormFile("file", fileName)
	if err != nil {
		fmt.Println("创建表单文件字段时出错：", err)
		return ""
	}
	// 将文件内容的io.Reader复制到表单文件字段中
	_, err = io.Copy(fileField, ossData)
	if err != nil {
		fmt.Println("复制文件内容时出错：", err)
		return ""
	}
	// 关闭multipart写入器以完成请求
	writer.Close()

	request, err := http.NewRequest(http.MethodPost, fmt.Sprintf("http://%s", this.EndPoint), &buffer)
	if err != nil {
		logger.AccessLogger.Errorf("---> local upload request fail. %s", err.Error())
		return ""
	}
	request.Header.Set("Content-Type", writer.FormDataContentType())

	client := http.Client{Timeout: 5 * time.Second}
	response, errs := client.Do(request)
	if errs != nil {
		logger.AccessLogger.Errorf("---> local upload fail. %s", errs.Error())
		return ""
	}

	if response.StatusCode == 200 {
		result := ReturnUpload{}
		bodyBuffer := new(bytes.Buffer)
		_, err = bodyBuffer.ReadFrom(response.Body)

		logger.AccessLogger.Errorf("----> %s", bodyBuffer.String())
		if err != nil {
			logger.AccessLogger.Errorf("---> local upload fail. %s", errs.Error())
			return ""
		}

		if err1 := json.Unmarshal([]byte(bodyBuffer.String()), &result); err1 != nil {
			panic(result.Message)
		}

		if result.Code == 200 {
			return result.Data.FileUrl
		}
	}
	defer response.Body.Close()

	return ""
}

// 获取直传token
func (this *Local) Token(key string, contentType string) (map[string]interface{}, error) {
	token := make(map[string]interface{})
	token["host"] = fmt.Sprintf("http://%s", this.EndPoint)
	token["directory"] = "pic"

	return token, nil
}