openbilibili/app/admin/main/credit/dao/upload/dao.go

package upload

import (
	"context"
	"crypto/hmac"
	"crypto/sha1"
	"encoding/base64"
	"fmt"
	"hash"
	"io"
	"net/http"
	"strconv"

	"go-common/app/admin/main/credit/conf"
	"go-common/library/log"
	httpx "go-common/library/net/http/blademaster"
)

const (
	_uploadURL = "/bfs/%s/%s"
	_template  = "%s\n%s\n%s\n%d\n"
	_method    = "PUT"
	_bucket    = "test" //blocked
)

// Dao .
type Dao struct {
	key    string
	secret string
	host   string
	client *httpx.Client
}

// New .
func New(c *conf.Config) (d *Dao) {
	d = &Dao{
		key:    c.BFS.Key,
		secret: c.BFS.Secret,
		host:   c.Host.Bfs,
		client: httpx.NewClient(c.HTTPClient),
	}
	return
}

// Upload upload picture or log file to bfs
func (d *Dao) Upload(c context.Context, fileName, fileType string, expire int64, body io.Reader) (location string, err error) {
	var (
		url  string
		req  *http.Request
		resp *http.Response
		code int
	)
	client := &http.Client{}
	url = fmt.Sprintf(d.host+_uploadURL, _bucket, fileName)
	if req, err = http.NewRequest(_method, url, body); err != nil {
		log.Error("http.NewRequest() Upload(%v) error(%v)", url, err)
		return
	}
	authorization := authorize(d.key, d.secret, _method, _bucket, fileName, expire)
	req.Header.Set("Host", d.host)
	req.Header.Add("Date", fmt.Sprint(expire))
	req.Header.Add("Authorization", authorization)
	req.Header.Add("Content-Type", fileType)
	resp, err = client.Do(req)
	if err != nil {
		log.Error("resp.Do(%s) error(%v)", url, err)
		return
	}
	defer resp.Body.Close()
	if resp.StatusCode != http.StatusOK {
		err = fmt.Errorf("status code error:%v", resp.StatusCode)
		return
	}
	code, err = strconv.Atoi(resp.Header.Get("code"))
	if err != nil || code != 200 {
		err = fmt.Errorf("response code error:%v", code)
		return
	}
	location = resp.Header.Get("Location")
	return
}

// authorize returns authorization for upload file to bfs
func authorize(key, secret, method, bucket, file string, expire int64) (authorization string) {
	var (
		content   string
		mac       hash.Hash
		signature string
	)
	content = fmt.Sprintf(_template, method, bucket, file, expire)
	mac = hmac.New(sha1.New, []byte(secret))
	mac.Write([]byte(content))
	signature = base64.StdEncoding.EncodeToString(mac.Sum(nil))
	authorization = fmt.Sprintf("%s:%s:%d", key, signature, expire)
	return
}