common.go 2.07 KB
Newer Older
haoyanbin's avatar
1  
haoyanbin committed
1 2
package utils

haoyanbin's avatar
1  
haoyanbin committed
3
import (
haoyanbin's avatar
1  
haoyanbin committed
4
	"bytes"
haoyanbin's avatar
haoyanbin committed
5 6
	"crypto/md5"
	"encoding/hex"
haoyanbin's avatar
1  
haoyanbin committed
7
	"fmt"
haoyanbin's avatar
1  
haoyanbin committed
8 9 10 11 12 13
	"github.com/aliyun/aliyun-oss-go-sdk/oss"
	uuid "github.com/satori/go.uuid"
	extConfig "go-admin/config"
	"io"
	"mime/multipart"
	"path/filepath"
haoyanbin's avatar
1  
haoyanbin committed
14
	"strconv"
haoyanbin's avatar
1  
haoyanbin committed
15
	"time"
haoyanbin's avatar
1  
haoyanbin committed
16
)
haoyanbin's avatar
1  
haoyanbin committed
17 18 19 20 21 22 23

func StringToInt(data string) int {
	reply, _ := strconv.Atoi(data)
	return reply
}

func GetAvg(data string, count int64) string {
haoyanbin's avatar
1  
haoyanbin committed
24 25 26
	floatData, _ := strconv.ParseFloat(data, 64)
	res := fmt.Sprintf("%.2f", floatData/float64(count))
	return res
haoyanbin's avatar
1  
haoyanbin committed
27
}
haoyanbin's avatar
haoyanbin committed
28 29 30 31 32

func StringToFloat(data string) float64 {
	reply, _ := strconv.ParseFloat(data, 64)
	return reply
}
haoyanbin's avatar
1  
haoyanbin committed
33

haoyanbin's avatar
haoyanbin committed
34 35 36 37 38
func StringToTime(data string) time.Time {
	timeData, _ := time.Parse("2006-01-02 15:04:05", data)
	return timeData
}

haoyanbin's avatar
1  
haoyanbin committed
39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61
//上传文件
func UploadFile(f multipart.File, h *multipart.FileHeader) (string, error) {

	fullFileName := ""

	bucket, err := GetOssBucket()
	if err != nil {
		return fullFileName, err
	}

	uuidValue := uuid.NewV4()
	guid := uuidValue.String()
	buf := bytes.NewBuffer(nil)
	_, err = io.Copy(buf, f)

	if err != nil {
		return fullFileName, err
	}

	//上传到阿里云oss
	yunfiletmp := fmt.Sprintf("%s/%s", extConfig.ExtConfig.Upload.Dir, GetDate())
	fullFileName = fmt.Sprintf("%s/%s%s", yunfiletmp, guid, filepath.Ext(h.Filename))
	//fmt.Println(fullFileName)
haoyanbin's avatar
haoyanbin committed
62 63
	option := oss.ContentType("image/jpg")
	if err = bucket.PutObject(fullFileName, buf, option); err != nil {
haoyanbin's avatar
1  
haoyanbin committed
64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93
		return fullFileName, err
	}
	return fullFileName, nil
}

func GetOssBucket() (*oss.Bucket, error) {
	var Buckets *oss.Bucket

	client, err := oss.New(extConfig.ExtConfig.Aliyunoss.Endpoint, extConfig.ExtConfig.Aliyunoss.Accessid, extConfig.ExtConfig.Aliyunoss.Accesskey)

	if err != nil {
		return nil, err
	}

	// 获取存储空间。
	Buckets, err = client.Bucket(extConfig.ExtConfig.Aliyunoss.Bucket)

	if err != nil {
		return nil, err
	}

	return Buckets, nil
}

func GetDate() string {
	year := time.Now().Format("2006")
	month := time.Now().Format("01")
	day := time.Now().Format("02")
	return year + month + day
}
haoyanbin's avatar
haoyanbin committed
94 95 96 97 98 99

func Md5Pwd(data string) string {
	h := md5.New()
	h.Write([]byte(data))
	return hex.EncodeToString(h.Sum(nil))
}