Compare commits
14 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
02edbdae8d | ||
|
|
5e75094c16 | ||
|
|
140317e2df | ||
|
|
8e2301eb34 | ||
|
|
c673da89cf | ||
|
|
005cf11b2a | ||
|
|
8195d633a1 | ||
|
|
4741a0bd93 | ||
|
|
f22c9df605 | ||
|
|
82cffdfe29 | ||
|
|
9da1a3bf25 | ||
|
|
2cb005e8ed | ||
|
|
374fdac477 | ||
|
|
595ababfde |
@@ -1,7 +1,7 @@
|
|||||||
package config
|
package config
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/ayflying/utility_go"
|
"github.com/ayflying/utility_go/pkg"
|
||||||
"github.com/gogf/gf/v2/util/gutil"
|
"github.com/gogf/gf/v2/util/gutil"
|
||||||
"sync"
|
"sync"
|
||||||
)
|
)
|
||||||
@@ -23,7 +23,7 @@ func (c *{mod}Mod) Load(_cfg ...string) {
|
|||||||
c.lock.Lock()
|
c.lock.Lock()
|
||||||
defer c.lock.Unlock()
|
defer c.lock.Unlock()
|
||||||
c.cfgArr = make([]*{cfg}Cfg, 0)
|
c.cfgArr = make([]*{cfg}Cfg, 0)
|
||||||
data, err := utility_go.Config.GetFile("{file}")
|
data, err := pkg.Config().GetFile("{file}")
|
||||||
err = data.Scan(&c.cfgArr)
|
err = data.Scan(&c.cfgArr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
|
|||||||
@@ -27,26 +27,26 @@ type Cfg struct {
|
|||||||
|
|
||||||
// Deprecated : pkg.Config().GetDbFile(name)
|
// Deprecated : pkg.Config().GetDbFile(name)
|
||||||
func (c *Cfg) GetDbFile(name string) (res *g.Var, err error) {
|
func (c *Cfg) GetDbFile(name string) (res *g.Var, err error) {
|
||||||
pkg.Config().GetDbFile(name)
|
res, err = pkg.Config().GetDbFile(name)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Deprecated : pkg.Config().GetFile(name, obj...)
|
// Deprecated : pkg.Config().GetFile(name, obj...)
|
||||||
func (c *Cfg) GetFile(filename string, obj ...Load) (jsonObj *gjson.Json, err error) {
|
func (c *Cfg) GetFile(filename string, obj ...Load) (jsonObj *gjson.Json, err error) {
|
||||||
pkg.Config().GetFile(filename)
|
jsonObj, err = pkg.Config().GetFile(filename)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// getUrlFile 获取远程配置
|
// getUrlFile 获取远程配置
|
||||||
// Deprecated : pkg.Config().GetUrlFile(name)
|
// Deprecated : pkg.Config().GetUrlFile(name)
|
||||||
func (c *Cfg) GetUrlFile(name string) (jsonObj *gjson.Json, err error) {
|
func (c *Cfg) GetUrlFile(name string) (jsonObj *gjson.Json, err error) {
|
||||||
pkg.Config().GetUrlFile(name)
|
jsonObj, err = pkg.Config().GetUrlFile(name)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Deprecated : pkg.Config().GetApollo(name, obj)
|
// Deprecated : pkg.Config().GetApollo(name, obj)
|
||||||
func (c *Cfg) GetApollo(name string, obj Load) (jsonObj *gjson.Json, err error) {
|
func (c *Cfg) GetApollo(name string, obj Load) (jsonObj *gjson.Json, err error) {
|
||||||
pkg.Config().GetApollo(name, obj)
|
jsonObj, err = pkg.Config().GetApollo(name, obj)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
4
go.mod
4
go.mod
@@ -4,7 +4,7 @@ go 1.24.0
|
|||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/apolloconfig/agollo/v4 v4.4.0
|
github.com/apolloconfig/agollo/v4 v4.4.0
|
||||||
github.com/ayflying/excel2json v1.1.7
|
github.com/ayflying/excel2json v1.1.9
|
||||||
github.com/elastic/go-elasticsearch/v8 v8.18.0
|
github.com/elastic/go-elasticsearch/v8 v8.18.0
|
||||||
github.com/go-pay/crypto v0.0.1
|
github.com/go-pay/crypto v0.0.1
|
||||||
github.com/go-pay/gopay v1.5.109
|
github.com/go-pay/gopay v1.5.109
|
||||||
@@ -12,7 +12,6 @@ require (
|
|||||||
github.com/goccy/go-json v0.10.5
|
github.com/goccy/go-json v0.10.5
|
||||||
github.com/gogf/gf/contrib/config/apollo/v2 v2.9.0
|
github.com/gogf/gf/contrib/config/apollo/v2 v2.9.0
|
||||||
github.com/gogf/gf/v2 v2.9.0
|
github.com/gogf/gf/v2 v2.9.0
|
||||||
github.com/gorilla/websocket v1.5.3
|
|
||||||
github.com/lionsoul2014/ip2region/binding/golang v0.0.0-20241220152942-06eb5c6e8230
|
github.com/lionsoul2014/ip2region/binding/golang v0.0.0-20241220152942-06eb5c6e8230
|
||||||
github.com/minio/minio-go/v7 v7.0.91
|
github.com/minio/minio-go/v7 v7.0.91
|
||||||
github.com/prometheus/client_golang v1.22.0
|
github.com/prometheus/client_golang v1.22.0
|
||||||
@@ -45,6 +44,7 @@ require (
|
|||||||
github.com/golang/protobuf v1.5.4 // indirect
|
github.com/golang/protobuf v1.5.4 // indirect
|
||||||
github.com/google/uuid v1.6.0 // indirect
|
github.com/google/uuid v1.6.0 // indirect
|
||||||
github.com/googleapis/gax-go/v2 v2.0.5 // indirect
|
github.com/googleapis/gax-go/v2 v2.0.5 // indirect
|
||||||
|
github.com/gorilla/websocket v1.5.3 // indirect
|
||||||
github.com/grokify/html-strip-tags-go v0.1.0 // indirect
|
github.com/grokify/html-strip-tags-go v0.1.0 // indirect
|
||||||
github.com/hashicorp/hcl v1.0.0 // indirect
|
github.com/hashicorp/hcl v1.0.0 // indirect
|
||||||
github.com/klauspost/compress v1.18.0 // indirect
|
github.com/klauspost/compress v1.18.0 // indirect
|
||||||
|
|||||||
4
go.sum
4
go.sum
@@ -50,8 +50,8 @@ github.com/apolloconfig/agollo/v4 v4.4.0/go.mod h1:6WjI68IzqMk/Y6ghMtrj5AX6Uewo2
|
|||||||
github.com/armon/circbuf v0.0.0-20150827004946-bbbad097214e/go.mod h1:3U/XgcO3hCbHZ8TKRvWD2dDTCfh9M9ya+I9JpbB7O8o=
|
github.com/armon/circbuf v0.0.0-20150827004946-bbbad097214e/go.mod h1:3U/XgcO3hCbHZ8TKRvWD2dDTCfh9M9ya+I9JpbB7O8o=
|
||||||
github.com/armon/go-metrics v0.0.0-20180917152333-f0300d1749da/go.mod h1:Q73ZrmVTwzkszR9V5SSuryQ31EELlFMUz1kKyl939pY=
|
github.com/armon/go-metrics v0.0.0-20180917152333-f0300d1749da/go.mod h1:Q73ZrmVTwzkszR9V5SSuryQ31EELlFMUz1kKyl939pY=
|
||||||
github.com/armon/go-radix v0.0.0-20180808171621-7fddfc383310/go.mod h1:ufUuZ+zHj4x4TnLV4JWEpy2hxWSpsRywHrMgIH9cCH8=
|
github.com/armon/go-radix v0.0.0-20180808171621-7fddfc383310/go.mod h1:ufUuZ+zHj4x4TnLV4JWEpy2hxWSpsRywHrMgIH9cCH8=
|
||||||
github.com/ayflying/excel2json v1.1.7 h1:gp4Ls6hNXd1EuT2MEiDQ6yPLaSKIFWkGyaNajMTc59c=
|
github.com/ayflying/excel2json v1.1.9 h1:AKaUFc/FNQXV/53/VSgp3J+2d4PMWhVpTDRj/yx3VOs=
|
||||||
github.com/ayflying/excel2json v1.1.7/go.mod h1:GjXYP2cDzeTgQxssQ3ba8QRa9JsScAeJMwCrZ7QTlQo=
|
github.com/ayflying/excel2json v1.1.9/go.mod h1:GjXYP2cDzeTgQxssQ3ba8QRa9JsScAeJMwCrZ7QTlQo=
|
||||||
github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
|
github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
|
||||||
github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
|
github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
|
||||||
github.com/bgentry/speakeasy v0.1.0/go.mod h1:+zsyZBPWlz7T6j88CTgSN5bM796AkVf0kBD4zp0CCIs=
|
github.com/bgentry/speakeasy v0.1.0/go.mod h1:+zsyZBPWlz7T6j88CTgSN5bM796AkVf0kBD4zp0CCIs=
|
||||||
|
|||||||
@@ -136,6 +136,10 @@ func (s *sSystemCron) AddCronV2(typ v1.CronType, _func func(context.Context) err
|
|||||||
// @receiver s
|
// @receiver s
|
||||||
// @return err
|
// @return err
|
||||||
func (s *sSystemCron) StartCron() (err error) {
|
func (s *sSystemCron) StartCron() (err error) {
|
||||||
|
//如果没有数据库配置,跳过计划任务执行
|
||||||
|
if g.Cfg().MustGet(gctx.New(), "database") == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
//预防重复启动
|
//预防重复启动
|
||||||
if startTime != nil {
|
if startTime != nil {
|
||||||
return
|
return
|
||||||
|
|||||||
5
modules/websocket/go.mod
Normal file
5
modules/websocket/go.mod
Normal file
@@ -0,0 +1,5 @@
|
|||||||
|
module github.com/ayflying/utility_go/modules/websocket
|
||||||
|
|
||||||
|
go 1.24.0
|
||||||
|
|
||||||
|
require github.com/ayflying/utility_go v1.2.26
|
||||||
2
modules/websocket/go.sum
Normal file
2
modules/websocket/go.sum
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
github.com/ayflying/utility_go v1.2.26/go.mod h1:y6qwRUmYk73cH/dTRSzj2mnaQdtrAfFNH4wRFewxlIc=
|
||||||
|
github.com/gorilla/websocket v1.5.3/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
|
||||||
101
package/pay/taptap/sign.go
Normal file
101
package/pay/taptap/sign.go
Normal file
@@ -0,0 +1,101 @@
|
|||||||
|
package taptap
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"crypto/hmac"
|
||||||
|
"crypto/sha256"
|
||||||
|
"encoding/base64"
|
||||||
|
"fmt"
|
||||||
|
"github.com/gogf/gf/v2/os/gtime"
|
||||||
|
"github.com/gogf/gf/v2/util/grand"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"sort"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type pTapTap struct {
|
||||||
|
Secret string `json:"secret" dc:"秘钥"`
|
||||||
|
OrderId string `json:"order_id" dc:"订单唯一 ID"`
|
||||||
|
ClientId string `json:"client_id" dc:"应用的 Client ID"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func New(orderId string) *pTapTap {
|
||||||
|
return &pTapTap{
|
||||||
|
Secret: "5AFEWnadBA0NgJK2mxeBLQEde0qyIefxLSc4XKHsx9AwkQRhxzkQ9DixsOkK6gcV",
|
||||||
|
ClientId: "mox88lbz43edfukdgk",
|
||||||
|
OrderId: orderId,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *pTapTap) Sign(url string, body []byte) (sign string, ts int64, nonce string, err error) {
|
||||||
|
//nolint:gosec
|
||||||
|
secret := p.Secret
|
||||||
|
//body := gjson.MustEncode(g.Map{})
|
||||||
|
//body := []byte(`{"event_type":"charge.succeeded","order":{"order_id":"1790288650833465345","purchase_token":"rT2Et9p0cfzq4fwjrTsGSacq0jQExFDqf5gTy1alp+Y=","client_id":"o6nD4iNavjQj75zPQk","open_id":"4+Axcl2RFgXbt6MZwdh++w==","user_region":"US","goods_open_id":"com.goods.open_id","goods_name":"TestGoodsName","status":"charge.succeeded","amount":"19000000000","currency":"USD","create_time":"1716168000","pay_time":"1716168000","extra":"1111111111111111111"}}`)
|
||||||
|
//url := "https://example.com/my-service/v1/my-method"
|
||||||
|
ts = gtime.Now().Unix()
|
||||||
|
nonce = grand.S(5)
|
||||||
|
method := "POST"
|
||||||
|
header := http.Header{
|
||||||
|
"Content-Type": {"Content-Type: application/json; charset=utf-8"},
|
||||||
|
"X-Tap-Ts": {strconv.FormatInt(ts, 10)},
|
||||||
|
"X-Tap-Nonce": {nonce},
|
||||||
|
}
|
||||||
|
ctx := context.Background()
|
||||||
|
req, err := http.NewRequestWithContext(ctx, method, url, bytes.NewBuffer(body))
|
||||||
|
req.Header = header
|
||||||
|
sign, err = Sign(req, secret)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
req.Header.Set("X-Tap-Sign", sign)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sign signs the request.
|
||||||
|
func Sign(req *http.Request, secret string) (string, error) {
|
||||||
|
methodPart := req.Method
|
||||||
|
urlPathAndQueryPart := req.URL.RequestURI()
|
||||||
|
headersPart, err := getHeadersPart(req.Header)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
bodyPart, err := io.ReadAll(req.Body)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
signParts := methodPart + "\n" + urlPathAndQueryPart + "\n" + headersPart + "\n" + string(bodyPart) + "\n"
|
||||||
|
fmt.Println(signParts)
|
||||||
|
h := hmac.New(sha256.New, []byte(secret))
|
||||||
|
h.Write([]byte(signParts))
|
||||||
|
rawSign := h.Sum(nil)
|
||||||
|
sign := base64.StdEncoding.EncodeToString(rawSign)
|
||||||
|
return sign, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// getHeadersPart returns the headers part of the request.
|
||||||
|
func getHeadersPart(header http.Header) (string, error) {
|
||||||
|
var headerKeys []string
|
||||||
|
for k, v := range header {
|
||||||
|
k = strings.ToLower(k)
|
||||||
|
if !strings.HasPrefix(k, "x-tap-") {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if k == "x-tap-sign" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if len(v) > 1 {
|
||||||
|
return "", fmt.Errorf("invalid header, %q has multiple values", k)
|
||||||
|
}
|
||||||
|
headerKeys = append(headerKeys, k)
|
||||||
|
}
|
||||||
|
sort.Strings(headerKeys)
|
||||||
|
headers := make([]string, 0, len(headerKeys))
|
||||||
|
for _, k := range headerKeys {
|
||||||
|
headers = append(headers, fmt.Sprintf("%s:%s", k, header.Get(k)))
|
||||||
|
}
|
||||||
|
return strings.Join(headers, "\n"), nil
|
||||||
|
}
|
||||||
59
package/pay/taptap/taptap.go
Normal file
59
package/pay/taptap/taptap.go
Normal file
@@ -0,0 +1,59 @@
|
|||||||
|
package taptap
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
|
"github.com/gogf/gf/v2/net/gclient"
|
||||||
|
"github.com/gogf/gf/v2/os/gctx"
|
||||||
|
"strconv"
|
||||||
|
)
|
||||||
|
|
||||||
|
type GetPayData struct {
|
||||||
|
Data struct {
|
||||||
|
Order struct {
|
||||||
|
} `json:"order"`
|
||||||
|
} `json:"data"`
|
||||||
|
Success bool `json:"success"`
|
||||||
|
}
|
||||||
|
|
||||||
|
//查询订单信息
|
||||||
|
func (p *pTapTap) Info(orderId string, clientId string, token []byte) (res string, err error) {
|
||||||
|
url := fmt.Sprintf("https://cloud-payment.tapapis.cn/order/v1/info?client_id=%v&order_id=%v", orderId, clientId)
|
||||||
|
res, err = p.get(url, token)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
//验证并核销订单
|
||||||
|
func (p *pTapTap) Verify(orderId string, clientId string, token []byte) (res string, err error) {
|
||||||
|
url := fmt.Sprintf("https://cloud-payment.tapapis.cn/order/v1/verify?client_id=%v", clientId)
|
||||||
|
res, err = p.get(url, token)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *pTapTap) get(url string, token []byte, _data ...any) (res string, err error) {
|
||||||
|
sign, ts, nonce, err := p.Sign(url, token)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var _get *gclient.Response
|
||||||
|
if len(_data) == 0 {
|
||||||
|
_get, err = g.Client().Header(map[string]string{
|
||||||
|
"X-Tap-Sign": sign,
|
||||||
|
"X-Tap-Nonce": nonce,
|
||||||
|
"X-Tap-Ts": strconv.FormatInt(ts, 10),
|
||||||
|
}).Get(gctx.New(), url)
|
||||||
|
|
||||||
|
} else {
|
||||||
|
_get, err = g.Client().Header(map[string]string{
|
||||||
|
"X-Tap-Sign": sign,
|
||||||
|
"X-Tap-Nonce": nonce,
|
||||||
|
"X-Tap-Ts": strconv.FormatInt(ts, 10),
|
||||||
|
}).Post(gctx.New(), url, _data[0])
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
res = _get.ReadAllString()
|
||||||
|
return
|
||||||
|
}
|
||||||
40
package/pay/taptap/webhook.go
Normal file
40
package/pay/taptap/webhook.go
Normal file
@@ -0,0 +1,40 @@
|
|||||||
|
package taptap
|
||||||
|
|
||||||
|
import "github.com/gogf/gf/v2/encoding/gjson"
|
||||||
|
|
||||||
|
type WebhookData struct {
|
||||||
|
Order *Order `json:"order"`
|
||||||
|
EventType string `json:"event_type"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// Order 订单信息结构体
|
||||||
|
type Order struct {
|
||||||
|
OrderID string `json:"order_id" dc:"订单唯一ID"`
|
||||||
|
PurchaseToken string `json:"purchase_token" dc:"用于订单核销的token"`
|
||||||
|
ClientID string `json:"client_id" dc:"应用的Client ID"`
|
||||||
|
OpenID string `json:"open_id" dc:"用户的开放平台ID"`
|
||||||
|
UserRegion string `json:"user_region" dc:"用户地区"`
|
||||||
|
GoodsOpenID string `json:"goods_open_id" dc:"商品唯一ID"`
|
||||||
|
GoodsName string `json:"goods_name" dc:"商品名称"`
|
||||||
|
Status string `json:"status" dc:"订单状态"`
|
||||||
|
Amount string `json:"amount" dc:"金额(本币金额x1,000,000)"`
|
||||||
|
Currency string `json:"currency" dc:"币种"`
|
||||||
|
CreateTime string `json:"create_time" dc:"创建时间"`
|
||||||
|
PayTime string `json:"pay_time" dc:"支付时间"`
|
||||||
|
Extra string `json:"extra" dc:"商户自定义数据,如角色信息等,长度不超过255 UTF-8字符"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *pTapTap) Webhook(body []byte) (res string, err error) {
|
||||||
|
var data *WebhookData
|
||||||
|
gjson.DecodeTo(body, &data)
|
||||||
|
|
||||||
|
switch data.EventType {
|
||||||
|
case "charge.succeeded": //充值成功
|
||||||
|
//todo 处理订单信息
|
||||||
|
|
||||||
|
case "refund.succeeded": //退款成功
|
||||||
|
case "refund.failed": //退款失败
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
@@ -3,121 +3,270 @@ package drive
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"github.com/gogf/gf/v2/container/gvar"
|
"github.com/gogf/gf/v2/container/gvar"
|
||||||
|
"github.com/gogf/gf/v2/crypto/gmd5"
|
||||||
|
"github.com/gogf/gf/v2/encoding/gjson"
|
||||||
"github.com/gogf/gf/v2/os/gcache"
|
"github.com/gogf/gf/v2/os/gcache"
|
||||||
|
"github.com/gogf/gf/v2/os/gctx"
|
||||||
"github.com/gogf/gf/v2/os/gfile"
|
"github.com/gogf/gf/v2/os/gfile"
|
||||||
|
"github.com/gogf/gf/v2/os/gtime"
|
||||||
"github.com/gogf/gf/v2/util/gconv"
|
"github.com/gogf/gf/v2/util/gconv"
|
||||||
"path"
|
"path"
|
||||||
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
fileIndex = "index.txt"
|
||||||
|
)
|
||||||
|
|
||||||
|
type FileIndex struct {
|
||||||
|
File string `json:"file"`
|
||||||
|
Duration time.Duration `json:"duration"`
|
||||||
|
}
|
||||||
|
type FileData struct {
|
||||||
|
Data interface{} `json:"data"`
|
||||||
|
Time int64 `json:"time"`
|
||||||
|
}
|
||||||
|
|
||||||
type AdapterFile struct {
|
type AdapterFile struct {
|
||||||
FilePath string
|
FilePath string
|
||||||
|
Lock sync.Mutex
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a AdapterFile) Set(ctx context.Context, key interface{}, value interface{}, duration time.Duration) error {
|
func (a *AdapterFile) Key2Name(key interface{}) string {
|
||||||
arr := strings.Split(":", gconv.String(key))
|
md5Str, _ := gmd5.Encrypt(key)
|
||||||
fileName := path.Join(arr...)
|
fileName := path.Join(md5Str[0:2], md5Str[2:4], md5Str[4:6], md5Str[16:])
|
||||||
return gfile.PutBytes(fileName, gconv.Bytes(value))
|
fileNameAll := path.Join(a.FilePath, fileName)
|
||||||
|
|
||||||
|
return fileNameAll
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a AdapterFile) SetMap(ctx context.Context, data map[interface{}]interface{}, duration time.Duration) error {
|
func (a *AdapterFile) AddIndex(FileName interface{}, duration time.Duration) {
|
||||||
|
var isEdit bool
|
||||||
|
|
||||||
|
var setTime int64
|
||||||
|
if duration == 0 {
|
||||||
|
setTime = 0
|
||||||
|
} else {
|
||||||
|
setTime = gtime.Now().Add(duration).Unix()
|
||||||
|
}
|
||||||
|
saveArr := []string{
|
||||||
|
gconv.String(FileName),
|
||||||
|
strconv.FormatInt(setTime, 10),
|
||||||
|
}
|
||||||
|
saveStr := strings.Join(saveArr, "|")
|
||||||
|
|
||||||
|
gfile.ReadLines(fileIndex, func(text string) (err error) {
|
||||||
|
arr := strings.Split(text, "|")
|
||||||
|
if arr[0] == FileName {
|
||||||
|
isEdit = true
|
||||||
|
gfile.ReplaceFile(text, saveStr, fileIndex)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return
|
||||||
|
})
|
||||||
|
|
||||||
|
if isEdit {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
gfile.PutContentsAppend(fileIndex, saveStr+"\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AdapterFile) DelIndex(FileName interface{}) {
|
||||||
|
//var save bool
|
||||||
|
gfile.ReadLines(fileIndex, func(text string) (err error) {
|
||||||
|
arr := strings.Split(text, "|")
|
||||||
|
if arr[0] == FileName {
|
||||||
|
//save = true
|
||||||
|
err = gfile.ReplaceFile(text+"\n", "", fileIndex)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
})
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AdapterFile) Set(ctx context.Context, key interface{}, value interface{}, duration time.Duration) (err error) {
|
||||||
|
fileNameAll := a.Key2Name(key)
|
||||||
|
var send = &FileData{
|
||||||
|
Data: value,
|
||||||
|
Time: gtime.Now().Add(duration).Unix(),
|
||||||
|
}
|
||||||
|
err = gfile.PutBytes(fileNameAll, gjson.MustEncode(send))
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
a.AddIndex(key, duration)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AdapterFile) SetMap(ctx context.Context, data map[interface{}]interface{}, duration time.Duration) (err error) {
|
||||||
|
for k, v := range data {
|
||||||
|
//fileNameAll := a.Key2Name(k)
|
||||||
|
//var send = &FileData{
|
||||||
|
// Data: v,
|
||||||
|
// Time: gtime.Now().Add(duration).Unix(),
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
//err = gfile.PutBytes(fileNameAll, gconv.Bytes(send))
|
||||||
|
a.Set(ctx, k, v, duration)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AdapterFile) SetIfNotExist(ctx context.Context, key interface{}, value interface{}, duration time.Duration) (ok bool, err error) {
|
||||||
|
err = a.Set(ctx, key, value, duration)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AdapterFile) SetIfNotExistFunc(ctx context.Context, key interface{}, f gcache.Func, duration time.Duration) (ok bool, err error) {
|
||||||
//TODO implement me
|
//TODO implement me
|
||||||
panic("implement me")
|
panic("implement me")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a AdapterFile) SetIfNotExist(ctx context.Context, key interface{}, value interface{}, duration time.Duration) (ok bool, err error) {
|
func (a *AdapterFile) SetIfNotExistFuncLock(ctx context.Context, key interface{}, f gcache.Func, duration time.Duration) (ok bool, err error) {
|
||||||
//TODO implement me
|
//TODO implement me
|
||||||
panic("implement me")
|
panic("implement me")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a AdapterFile) SetIfNotExistFunc(ctx context.Context, key interface{}, f gcache.Func, duration time.Duration) (ok bool, err error) {
|
func (a *AdapterFile) Get(ctx context.Context, key interface{}) (res *gvar.Var, err error) {
|
||||||
|
var data *FileData
|
||||||
|
name := a.Key2Name(key)
|
||||||
|
if !gfile.IsFile(name) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
file := gfile.GetBytes(name)
|
||||||
|
gjson.DecodeTo(file, &data)
|
||||||
|
if data.Time < time.Now().Unix() {
|
||||||
|
a.Remove(ctx, key)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
res = gvar.New(data.Data)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AdapterFile) GetOrSet(ctx context.Context, key interface{}, value interface{}, duration time.Duration) (result *gvar.Var, err error) {
|
||||||
|
a.Set(ctx, key, value, duration)
|
||||||
|
result, err = a.Get(ctx, key)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AdapterFile) GetOrSetFunc(ctx context.Context, key interface{}, f gcache.Func, duration time.Duration) (result *gvar.Var, err error) {
|
||||||
|
value := f
|
||||||
|
a.Set(ctx, key, value, duration)
|
||||||
|
result, err = a.Get(ctx, key)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AdapterFile) GetOrSetFuncLock(ctx context.Context, key interface{}, f gcache.Func, duration time.Duration) (result *gvar.Var, err error) {
|
||||||
|
a.Lock.Lock()
|
||||||
|
defer a.Lock.Unlock()
|
||||||
|
result, err = a.GetOrSetFunc(ctx, key, f, duration)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AdapterFile) Contains(ctx context.Context, key interface{}) (bool, error) {
|
||||||
|
return gfile.IsFile(a.FilePath), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AdapterFile) Size(ctx context.Context) (size int, err error) {
|
||||||
|
size = int(gfile.Size(a.FilePath))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AdapterFile) Data(ctx context.Context) (data map[interface{}]interface{}, err error) {
|
||||||
//TODO implement me
|
//TODO implement me
|
||||||
panic("implement me")
|
panic("implement me")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a AdapterFile) SetIfNotExistFuncLock(ctx context.Context, key interface{}, f gcache.Func, duration time.Duration) (ok bool, err error) {
|
func (a *AdapterFile) Keys(ctx context.Context) (keys []interface{}, err error) {
|
||||||
|
keys = make([]interface{}, 0)
|
||||||
|
gfile.ReadLines(fileIndex, func(text string) (err error) {
|
||||||
|
arr := strings.Split(text, "|")
|
||||||
|
keys = append(keys, arr[0])
|
||||||
|
return
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AdapterFile) Values(ctx context.Context) (values []interface{}, err error) {
|
||||||
//TODO implement me
|
//TODO implement me
|
||||||
panic("implement me")
|
panic("implement me")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a AdapterFile) Get(ctx context.Context, key interface{}) (*gvar.Var, error) {
|
func (a *AdapterFile) Update(ctx context.Context, key interface{}, value interface{}) (oldValue *gvar.Var, exist bool, err error) {
|
||||||
|
fileNameAll := a.Key2Name(key)
|
||||||
|
getFile := gfile.GetBytes(fileNameAll)
|
||||||
|
var data *FileData
|
||||||
|
gconv.Scan(getFile, &data)
|
||||||
|
oldValue.Set(data.Data)
|
||||||
|
|
||||||
|
var send = &FileData{
|
||||||
|
Data: value,
|
||||||
|
Time: data.Time,
|
||||||
|
}
|
||||||
|
|
||||||
|
err = gfile.PutBytes(fileNameAll, gconv.Bytes(send))
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
a.AddIndex(key, time.Duration(data.Time-gtime.Now().Unix()))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *AdapterFile) UpdateExpire(ctx context.Context, key interface{}, duration time.Duration) (oldDuration time.Duration, err error) {
|
||||||
//TODO implement me
|
//TODO implement me
|
||||||
panic("implement me")
|
panic("implement me")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a AdapterFile) GetOrSet(ctx context.Context, key interface{}, value interface{}, duration time.Duration) (result *gvar.Var, err error) {
|
func (a *AdapterFile) GetExpire(ctx context.Context, key interface{}) (time.Duration, error) {
|
||||||
//TODO implement me
|
//TODO implement me
|
||||||
panic("implement me")
|
panic("implement me")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a AdapterFile) GetOrSetFunc(ctx context.Context, key interface{}, f gcache.Func, duration time.Duration) (result *gvar.Var, err error) {
|
func (a *AdapterFile) Remove(ctx context.Context, keys ...interface{}) (lastValue *gvar.Var, err error) {
|
||||||
//TODO implement me
|
for _, v := range keys {
|
||||||
panic("implement me")
|
fileNameAll := a.Key2Name(v)
|
||||||
|
lastValue, err = a.Get(ctx, fileNameAll)
|
||||||
|
err = gfile.RemoveFile(fileNameAll)
|
||||||
|
//删除索引文件
|
||||||
|
a.DelIndex(v)
|
||||||
|
}
|
||||||
|
return nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a AdapterFile) GetOrSetFuncLock(ctx context.Context, key interface{}, f gcache.Func, duration time.Duration) (result *gvar.Var, err error) {
|
func (a *AdapterFile) Clear(ctx context.Context) error {
|
||||||
//TODO implement me
|
return gfile.RemoveAll(a.FilePath)
|
||||||
panic("implement me")
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a AdapterFile) Contains(ctx context.Context, key interface{}) (bool, error) {
|
func (a *AdapterFile) Close(ctx context.Context) error {
|
||||||
//TODO implement me
|
return nil
|
||||||
panic("implement me")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a AdapterFile) Size(ctx context.Context) (size int, err error) {
|
|
||||||
//TODO implement me
|
|
||||||
panic("implement me")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a AdapterFile) Data(ctx context.Context) (data map[interface{}]interface{}, err error) {
|
|
||||||
//TODO implement me
|
|
||||||
panic("implement me")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a AdapterFile) Keys(ctx context.Context) (keys []interface{}, err error) {
|
|
||||||
//TODO implement me
|
|
||||||
panic("implement me")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a AdapterFile) Values(ctx context.Context) (values []interface{}, err error) {
|
|
||||||
//TODO implement me
|
|
||||||
panic("implement me")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a AdapterFile) Update(ctx context.Context, key interface{}, value interface{}) (oldValue *gvar.Var, exist bool, err error) {
|
|
||||||
//TODO implement me
|
|
||||||
panic("implement me")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a AdapterFile) UpdateExpire(ctx context.Context, key interface{}, duration time.Duration) (oldDuration time.Duration, err error) {
|
|
||||||
//TODO implement me
|
|
||||||
panic("implement me")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a AdapterFile) GetExpire(ctx context.Context, key interface{}) (time.Duration, error) {
|
|
||||||
//TODO implement me
|
|
||||||
panic("implement me")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a AdapterFile) Remove(ctx context.Context, keys ...interface{}) (lastValue *gvar.Var, err error) {
|
|
||||||
//TODO implement me
|
|
||||||
panic("implement me")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a AdapterFile) Clear(ctx context.Context) error {
|
|
||||||
//TODO implement me
|
|
||||||
panic("implement me")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a AdapterFile) Close(ctx context.Context) error {
|
|
||||||
//TODO implement me
|
|
||||||
panic("implement me")
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewAdapterFile(filePath string) gcache.Adapter {
|
func NewAdapterFile(filePath string) gcache.Adapter {
|
||||||
return &AdapterFile{
|
fileIndex = path.Join(filePath, fileIndex)
|
||||||
|
|
||||||
|
//清空过期文件目录
|
||||||
|
delKeys := make([]interface{}, 0)
|
||||||
|
gfile.ReadLines(fileIndex, func(text string) error {
|
||||||
|
arr := strings.Split(text, "|")
|
||||||
|
if gtime.Now().Unix() > gconv.Int64(arr[1]) {
|
||||||
|
delKeys = append(delKeys, arr[0])
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
|
||||||
|
obj := &AdapterFile{
|
||||||
FilePath: filePath,
|
FilePath: filePath,
|
||||||
}
|
}
|
||||||
|
if len(delKeys) > 0 {
|
||||||
|
obj.Remove(gctx.New(), delKeys...)
|
||||||
|
}
|
||||||
|
|
||||||
|
return obj
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ import (
|
|||||||
"github.com/gogf/gf/v2/container/gvar"
|
"github.com/gogf/gf/v2/container/gvar"
|
||||||
"github.com/gogf/gf/v2/encoding/gjson"
|
"github.com/gogf/gf/v2/encoding/gjson"
|
||||||
"github.com/gogf/gf/v2/frame/g"
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
|
"github.com/gogf/gf/v2/os/gctx"
|
||||||
"github.com/gogf/gf/v2/os/gfile"
|
"github.com/gogf/gf/v2/os/gfile"
|
||||||
"github.com/gogf/gf/v2/os/gres"
|
"github.com/gogf/gf/v2/os/gres"
|
||||||
"github.com/gogf/gf/v2/text/gstr"
|
"github.com/gogf/gf/v2/text/gstr"
|
||||||
@@ -86,14 +87,22 @@ func (c *Cfg) GetFile(filename string, _pathStr ...string) (jsonObj *gjson.Json,
|
|||||||
bytes = gres.GetContent(filePath) // 从打包资源中获取内容
|
bytes = gres.GetContent(filePath) // 从打包资源中获取内容
|
||||||
}
|
}
|
||||||
|
|
||||||
|
for range 5 {
|
||||||
//如果还是没有读取到配置,从当前目录返回上级读取
|
//如果还是没有读取到配置,从当前目录返回上级读取
|
||||||
if bytes == nil {
|
if bytes == nil {
|
||||||
// 拼接完整的文件路径
|
// 上级拼接完整的文件路径
|
||||||
filePath = "../../" + pathStr + filename + ".json"
|
filePath = "../" + filePath
|
||||||
if gfile.IsFile(filePath) {
|
if gfile.IsFile(filePath) {
|
||||||
bytes = gfile.GetBytes(filePath) // 读取物理文件内容
|
bytes = gfile.GetBytes(filePath) // 读取物理文件内容
|
||||||
|
//找到配置了,跳过
|
||||||
|
break
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
if bytes == nil {
|
||||||
|
g.Log().Errorf(gctx.New(), "未读取到配置文件:%v", filePath)
|
||||||
|
}
|
||||||
|
|
||||||
// 解析 JSON 内容并返回结果
|
// 解析 JSON 内容并返回结果
|
||||||
jsonObj, err = gjson.DecodeToJson(bytes)
|
jsonObj, err = gjson.DecodeToJson(bytes)
|
||||||
return
|
return
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ import (
|
|||||||
"github.com/ayflying/utility_go/pkg/notice"
|
"github.com/ayflying/utility_go/pkg/notice"
|
||||||
"github.com/ayflying/utility_go/pkg/rank"
|
"github.com/ayflying/utility_go/pkg/rank"
|
||||||
"github.com/ayflying/utility_go/pkg/s3"
|
"github.com/ayflying/utility_go/pkg/s3"
|
||||||
"github.com/ayflying/utility_go/pkg/websocket"
|
//"github.com/ayflying/utility_go/pkg/websocket"
|
||||||
"github.com/gogf/gf/v2/os/gcache"
|
"github.com/gogf/gf/v2/os/gcache"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -32,9 +32,9 @@ func Rank() *rank.Mod {
|
|||||||
return rank.New()
|
return rank.New()
|
||||||
}
|
}
|
||||||
|
|
||||||
func Websocket() *websocket.SocketV1 {
|
//func Websocket() *websocket.SocketV1 {
|
||||||
return websocket.NewV1()
|
// return websocket.NewV1()
|
||||||
}
|
//}
|
||||||
|
|
||||||
func Config() *config.Cfg {
|
func Config() *config.Cfg {
|
||||||
return config.NewV1()
|
return config.NewV1()
|
||||||
|
|||||||
23
pkg/s3/s3.go
23
pkg/s3/s3.go
@@ -31,6 +31,7 @@ type DataType struct {
|
|||||||
Url string `json:"url"` // S3 服务的访问 URL
|
Url string `json:"url"` // S3 服务的访问 URL
|
||||||
BucketName string `json:"bucket_name"` // 默认存储桶名称
|
BucketName string `json:"bucket_name"` // 默认存储桶名称
|
||||||
BucketNameCdn string `json:"bucket_name_cdn"` // CDN 存储桶名称
|
BucketNameCdn string `json:"bucket_name_cdn"` // CDN 存储桶名称
|
||||||
|
Provider string `json:"provider"` // S3 服务的提供方
|
||||||
}
|
}
|
||||||
|
|
||||||
// Mod 定义了 S3 模块的结构体,包含一个 S3 客户端实例和配置信息
|
// Mod 定义了 S3 模块的结构体,包含一个 S3 客户端实例和配置信息
|
||||||
@@ -64,6 +65,7 @@ func New(_name ...string) *Mod {
|
|||||||
&minio.Options{
|
&minio.Options{
|
||||||
Creds: credentials.NewStaticV4(cfg.AccessKey, cfg.SecretKey, ""),
|
Creds: credentials.NewStaticV4(cfg.AccessKey, cfg.SecretKey, ""),
|
||||||
Secure: cfg.Ssl,
|
Secure: cfg.Ssl,
|
||||||
|
//BucketLookup: minio.BucketLookupPath,
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -133,11 +135,13 @@ func (s *Mod) ListBuckets() []minio.BucketInfo {
|
|||||||
func (s *Mod) PutObject(f io.Reader, name string, bucketName string, _size ...int64) (res minio.UploadInfo, err error) {
|
func (s *Mod) PutObject(f io.Reader, name string, bucketName string, _size ...int64) (res minio.UploadInfo, err error) {
|
||||||
// 初始化文件大小为 -1,表示将读取文件至结束
|
// 初始化文件大小为 -1,表示将读取文件至结束
|
||||||
var size = int64(-1)
|
var size = int64(-1)
|
||||||
if len(_size) > 0 {
|
//if len(_size) > 0 {
|
||||||
size = _size[0]
|
// size = _size[0]
|
||||||
}
|
//}
|
||||||
// 调用 S3 客户端上传文件,设置内容类型为 "application/octet-stream"
|
// 调用 S3 客户端上传文件,设置内容类型为 "application/octet-stream"
|
||||||
res, err = s.client.PutObject(ctx, bucketName, name, f, size, minio.PutObjectOptions{ContentType: "application/octet-stream"})
|
res, err = s.client.PutObject(ctx, bucketName, name, f, size, minio.PutObjectOptions{
|
||||||
|
//ContentType: "application/octet-stream",
|
||||||
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
// 记录上传错误日志
|
// 记录上传错误日志
|
||||||
g.Log().Error(ctx, err)
|
g.Log().Error(ctx, err)
|
||||||
@@ -148,6 +152,7 @@ func (s *Mod) PutObject(f io.Reader, name string, bucketName string, _size ...in
|
|||||||
// RemoveObject 从指定存储桶中删除指定名称的文件
|
// RemoveObject 从指定存储桶中删除指定名称的文件
|
||||||
func (s *Mod) RemoveObject(name string, bucketName string) (err error) {
|
func (s *Mod) RemoveObject(name string, bucketName string) (err error) {
|
||||||
opts := minio.RemoveObjectOptions{
|
opts := minio.RemoveObjectOptions{
|
||||||
|
ForceDelete: true,
|
||||||
//GovernanceBypass: true,
|
//GovernanceBypass: true,
|
||||||
//VersionID: "myversionid",
|
//VersionID: "myversionid",
|
||||||
}
|
}
|
||||||
@@ -187,9 +192,15 @@ func (s *Mod) GetUrl(filePath string, defaultFile ...string) (url string) {
|
|||||||
filePath = defaultFile[0]
|
filePath = defaultFile[0]
|
||||||
}
|
}
|
||||||
|
|
||||||
if s.cfg.Ssl {
|
//switch s.cfg.Provider {
|
||||||
|
//case "qiniu":
|
||||||
|
// url = get + path.Join(bucketName, filePath)
|
||||||
|
//default:
|
||||||
|
// url = get + filePath
|
||||||
|
//}
|
||||||
url = get + filePath
|
url = get + filePath
|
||||||
} else {
|
|
||||||
|
if !s.cfg.Ssl {
|
||||||
url = get + path.Join(bucketName, filePath)
|
url = get + path.Join(bucketName, filePath)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user