gocos/gocos.go
xiaoqidun 379e65dafc
All checks were successful
continuous-integration/drone/push Build is passing
feat(时区设置): 默认北京时区,支持TZ环境变量
2020-12-26 10:52:32 +08:00

125 lines
2.5 KiB
Go

package main
import (
"context"
"errors"
"github.com/tencentyun/cos-go-sdk-v5"
"io/fs"
"log"
"net/http"
"net/url"
"os"
"path/filepath"
"strings"
"time"
_ "time/tzdata"
)
const (
SecretID = "secret_id"
SecretKey = "secret_key"
BucketURL = "bucket_url"
Source = "source"
Target = "target"
StripPrefix = "strip_prefix"
PathSeparator = "/"
)
func ErrExit(err error) {
log.Println(err.Error())
os.Exit(1)
}
func GetConfig(key string) string {
key = "PLUGIN_" + strings.ToUpper(key)
return os.Getenv(key)
}
func VarIsEmpty(a ...interface{}) bool {
for _, v := range a {
switch v := v.(type) {
case string:
if "" == v {
return true
}
case []byte:
if 0 == len(v) {
return true
}
case []string:
if 0 == len(v) {
return true
}
}
}
return false
}
func init() {
tz := os.Getenv("TZ")
if tz == "" {
tz = "Asia/Shanghai"
}
time.Local, _ = time.LoadLocation(tz)
}
func main() {
var (
err error
secretID = GetConfig(SecretID)
secretKey = GetConfig(SecretKey)
bucketURL = GetConfig(BucketURL)
source = GetConfig(Source)
target = GetConfig(Target)
stripPrefix = GetConfig(StripPrefix)
)
if VarIsEmpty(secretID, secretKey, bucketURL, source, target) {
ErrExit(errors.New("input error"))
}
sourceFiles := make([]string, 0, 0)
if err = filepath.WalkDir(source, func(path string, d fs.DirEntry, err error) error {
if !d.IsDir() {
sourceFiles = append(sourceFiles, path)
}
return nil
}); err != nil {
ErrExit(err)
}
sourceLen := len(sourceFiles)
if sourceLen == 0 {
return
}
if !strings.HasSuffix(target, PathSeparator) {
target += PathSeparator
}
u, err := url.Parse(bucketURL)
if err != nil {
ErrExit(err)
}
cosClient := cos.NewClient(
&cos.BaseURL{
BucketURL: u,
},
&http.Client{
Transport: &cos.AuthorizationTransport{
SecretID: secretID,
SecretKey: secretKey,
},
})
for i := 0; i < sourceLen; i++ {
local := sourceFiles[i]
remote := strings.TrimPrefix(local, stripPrefix)
if strings.HasPrefix(remote, PathSeparator) {
remote = strings.TrimPrefix(remote, PathSeparator)
}
remote = target + remote
if _, err = cosClient.Object.PutFromFile(context.Background(), remote, local, &cos.ObjectPutOptions{
ACLHeaderOptions: &cos.ACLHeaderOptions{},
ObjectPutHeaderOptions: &cos.ObjectPutHeaderOptions{},
}); err != nil {
ErrExit(err)
}
log.Printf("source:%s target:%s\r\n", local, remote)
}
}