Compare commits
34 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
dd8c05b344 | ||
|
|
e781e132ed | ||
|
|
862a6c8410 | ||
|
|
42535d0023 | ||
|
|
dd999cacf9 | ||
|
|
30d30bb8c6 | ||
|
|
f1c22dc9e6 | ||
| 14cf759ce1 | |||
| 19a19c1ff1 | |||
| 183d6d8b10 | |||
| d0cad61028 | |||
| efb34e0c5b | |||
|
|
8190e9f6b7 | ||
|
|
27435b57b7 | ||
|
|
0628882533 | ||
|
|
f68655eee6 | ||
|
|
d8491f0aba | ||
|
|
999f1f6a84 | ||
|
|
039acea0af | ||
|
|
d54de73e11 | ||
|
|
61d69159ea | ||
|
|
fa659d87ae | ||
|
|
ccc50a7dd0 | ||
|
|
d1a7ba8119 | ||
|
|
7c19a66cb5 | ||
|
|
90b72129a5 | ||
|
|
58bea0c09a | ||
|
|
50fe34e1c1 | ||
|
|
25c00d5072 | ||
|
|
ee32c8b83d | ||
|
|
12a193fdee | ||
|
|
ca55880beb | ||
|
|
ee9e399a81 | ||
|
|
ed9b0dd248 |
@@ -124,11 +124,11 @@ var (
|
|||||||
FileUrl: url[v.S3],
|
FileUrl: url[v.S3],
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
Proxy := g.Cfg().MustGet(ctx, "update_proxy", "http://192.168.50.170:10808").String()
|
||||||
g.Log().Debugf(ctx, "切换代理进行上传:err=%v", err)
|
g.Log().Debugf(ctx, "切换代理进行上传:err=%v", err)
|
||||||
get, err = client.Proxy("http://192.168.50.114:10808").
|
get, err = client.Proxy(Proxy).Post(ctx, address+"/callback/update", &UpdateReq{
|
||||||
Post(ctx, address+"/callback/update", &UpdateReq{
|
FileUrl: url[v.S3],
|
||||||
FileUrl: url[v.S3],
|
})
|
||||||
})
|
|
||||||
}
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
g.Log().Error(ctx, err)
|
g.Log().Error(ctx, err)
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ package boot
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
|
||||||
v1 "github.com/ayflying/utility_go/api/system/v1"
|
v1 "github.com/ayflying/utility_go/api/system/v1"
|
||||||
"github.com/ayflying/utility_go/service"
|
"github.com/ayflying/utility_go/service"
|
||||||
"github.com/gogf/gf/v2/os/gctx"
|
"github.com/gogf/gf/v2/os/gctx"
|
||||||
@@ -13,14 +14,15 @@ var (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func Boot() (err error) {
|
func Boot() (err error) {
|
||||||
err = service.SystemCron().StartCron()
|
// 启动计划任务定时器,预防debug工具激活计划任务造成重复执行,此处不执行计划任务
|
||||||
|
//err = service.SystemCron().StartCron()
|
||||||
|
|
||||||
//用户活动持久化
|
//用户活动持久化每小时执行一次
|
||||||
service.SystemCron().AddCronV2(v1.CronType_DAILY, func(ctx context.Context) error {
|
service.SystemCron().AddCronV2(v1.CronType_HOUR, func(ctx context.Context) error {
|
||||||
err = service.GameKv().SavesV1()
|
err = service.GameKv().SavesV1()
|
||||||
err = service.GameAct().Saves(ctx)
|
err = service.GameAct().Saves(ctx)
|
||||||
return err
|
return err
|
||||||
})
|
}, true)
|
||||||
|
|
||||||
//初始化自启动方法
|
//初始化自启动方法
|
||||||
for _, v := range _func {
|
for _, v := range _func {
|
||||||
|
|||||||
@@ -3,6 +3,10 @@ package gameAct
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/ayflying/utility_go/internal/model/do"
|
"github.com/ayflying/utility_go/internal/model/do"
|
||||||
"github.com/ayflying/utility_go/internal/model/entity"
|
"github.com/ayflying/utility_go/internal/model/entity"
|
||||||
"github.com/ayflying/utility_go/pkg"
|
"github.com/ayflying/utility_go/pkg"
|
||||||
@@ -13,15 +17,13 @@ import (
|
|||||||
"github.com/gogf/gf/v2/os/gctx"
|
"github.com/gogf/gf/v2/os/gctx"
|
||||||
"github.com/gogf/gf/v2/os/gtime"
|
"github.com/gogf/gf/v2/os/gtime"
|
||||||
"github.com/gogf/gf/v2/util/gconv"
|
"github.com/gogf/gf/v2/util/gconv"
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
ctx = gctx.New()
|
ctx = gctx.New()
|
||||||
Name = "game_act"
|
Name = "game_act"
|
||||||
ActList = gset.New(true)
|
ActList = gset.New(true)
|
||||||
|
RunTimeMax *gtime.Time
|
||||||
)
|
)
|
||||||
|
|
||||||
type sGameAct struct {
|
type sGameAct struct {
|
||||||
@@ -108,16 +110,24 @@ func (s *sGameAct) Set(uid int64, actId int, data interface{}) (err error) {
|
|||||||
|
|
||||||
func (s *sGameAct) Saves(ctx context.Context) (err error) {
|
func (s *sGameAct) Saves(ctx context.Context) (err error) {
|
||||||
getCache, _ := pkg.Cache("redis").Get(nil, "cron:game_act")
|
getCache, _ := pkg.Cache("redis").Get(nil, "cron:game_act")
|
||||||
|
g.Log().Debug(ctx, "开始执行游戏act数据保存了")
|
||||||
//如果没有执行过,设置时间戳
|
//如果没有执行过,设置时间戳
|
||||||
if getCache.Int64() > 0 {
|
if getCache.Int64() > 0 {
|
||||||
return
|
return
|
||||||
} else {
|
} else {
|
||||||
pkg.Cache("redis").Set(nil, "cron:game_act", gtime.Now().Unix(), time.Hour)
|
pkg.Cache("redis").Set(nil, "cron:game_act", gtime.Now().Unix(), time.Hour)
|
||||||
}
|
}
|
||||||
|
// 最大允许执行时间
|
||||||
|
RunTimeMax = gtime.Now().Add(time.Minute * 30)
|
||||||
//遍历执行
|
//遍历执行
|
||||||
ActList.Iterator(func(i interface{}) bool {
|
ActList.Iterator(func(i interface{}) bool {
|
||||||
err = s.Save(ctx, i.(int))
|
//在时间内允许执行
|
||||||
|
if gtime.Now().Before(RunTimeMax) {
|
||||||
|
g.Log().Debug(ctx, "开始执行游戏act数据保存: act%v", i)
|
||||||
|
err = s.Save(ctx, i.(int))
|
||||||
|
} else {
|
||||||
|
g.Log().Errorf(ctx, "游戏act数据保存超时: act=%v", i)
|
||||||
|
}
|
||||||
return true
|
return true
|
||||||
})
|
})
|
||||||
return
|
return
|
||||||
@@ -134,7 +144,13 @@ func (s *sGameAct) Save(ctx context.Context, actId int) (err error) {
|
|||||||
|
|
||||||
//循环获取缓存数据
|
//循环获取缓存数据
|
||||||
err = tools.Redis.RedisScanV2(cacheKey, func(keys []string) (err error) {
|
err = tools.Redis.RedisScanV2(cacheKey, func(keys []string) (err error) {
|
||||||
var add []interface{}
|
//判断是否超时
|
||||||
|
if gtime.Now().After(RunTimeMax) {
|
||||||
|
g.Log().Debug(ctx, "执行超时了,停止执行!")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var add = make([]*entity.GameAct, 0)
|
||||||
|
var update = make([]*entity.GameAct, 0)
|
||||||
var delKey []string
|
var delKey []string
|
||||||
for _, cacheKey = range keys {
|
for _, cacheKey = range keys {
|
||||||
result := strings.Split(cacheKey, ":")
|
result := strings.Split(cacheKey, ":")
|
||||||
@@ -176,40 +192,57 @@ func (s *sGameAct) Save(ctx context.Context, actId int) (err error) {
|
|||||||
}
|
}
|
||||||
actionData := cacheGet.String()
|
actionData := cacheGet.String()
|
||||||
if data == nil {
|
if data == nil {
|
||||||
//data =
|
add = append(add, &entity.GameAct{
|
||||||
add = append(add, &do.GameAct{
|
|
||||||
ActId: actId,
|
ActId: actId,
|
||||||
Uid: uid,
|
Uid: uid,
|
||||||
Action: actionData,
|
Action: actionData,
|
||||||
})
|
})
|
||||||
} else {
|
} else {
|
||||||
//覆盖数据
|
//覆盖数据
|
||||||
|
data.ActId = actId
|
||||||
|
data.Uid = uid
|
||||||
data.Action = actionData
|
data.Action = actionData
|
||||||
add = append(add, data)
|
update = append(update, data)
|
||||||
}
|
}
|
||||||
//最后删除key
|
//最后删除key
|
||||||
delKey = append(delKey, cacheKey)
|
delKey = append(delKey, cacheKey)
|
||||||
}
|
}
|
||||||
|
|
||||||
//批量写入数据库
|
//批量写入数据库
|
||||||
if len(add) > 0 {
|
if len(delKey) > 0 {
|
||||||
dbRes, err2 := g.Model(Name).Batch(30).Data(add).Save()
|
for _, v := range update {
|
||||||
add = make([]interface{}, 0)
|
v.UpdatedAt = gtime.Now()
|
||||||
if err2 != nil {
|
_, err2 := g.Model(Name).Where(do.GameAct{
|
||||||
g.Log().Error(ctx, err2)
|
Uid: v.Uid,
|
||||||
return
|
ActId: v.ActId,
|
||||||
}
|
}).Data(v).Update()
|
||||||
|
|
||||||
for _, v := range delKey {
|
|
||||||
_, err2 = g.Redis().Del(ctx, v)
|
|
||||||
if err2 != nil {
|
if err2 != nil {
|
||||||
g.Log().Error(ctx, err2)
|
g.Log().Error(ctx, err2)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
update = make([]*entity.GameAct, 0)
|
||||||
|
var count int64
|
||||||
|
|
||||||
|
if len(add) > 0 {
|
||||||
|
dbRes, err2 := g.Model(Name).Batch(50).Data(add).Save()
|
||||||
|
add = make([]*entity.GameAct, 0)
|
||||||
|
err = err2
|
||||||
|
if err != nil {
|
||||||
|
g.Log().Error(ctx, err2)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
count, _ = dbRes.RowsAffected()
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, v := range delKey {
|
||||||
|
_, err = g.Redis().Del(ctx, v)
|
||||||
|
if err != nil {
|
||||||
|
g.Log().Error(ctx, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
delKey = make([]string, 0)
|
delKey = make([]string, 0)
|
||||||
|
|
||||||
count, _ := dbRes.RowsAffected()
|
|
||||||
g.Log().Debugf(ctx, "当前 %v 写入数据库: %v 条", actId, count)
|
g.Log().Debugf(ctx, "当前 %v 写入数据库: %v 条", actId, count)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -2,21 +2,23 @@ package gameKv
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/ayflying/utility_go/pkg"
|
"github.com/ayflying/utility_go/pkg"
|
||||||
"github.com/ayflying/utility_go/service"
|
"github.com/ayflying/utility_go/service"
|
||||||
"github.com/ayflying/utility_go/tools"
|
"github.com/ayflying/utility_go/tools"
|
||||||
"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/gctx"
|
||||||
"github.com/gogf/gf/v2/os/gtime"
|
"github.com/gogf/gf/v2/os/gtime"
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
"sync"
|
|
||||||
"time"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
ctx = gctx.New()
|
ctx = gctx.New()
|
||||||
Name = "game_kv"
|
Name = "game_kv"
|
||||||
|
RunTimeMax *gtime.Time
|
||||||
)
|
)
|
||||||
|
|
||||||
type sGameKv struct {
|
type sGameKv struct {
|
||||||
@@ -37,6 +39,10 @@ func init() {
|
|||||||
// @receiver s: sGameKv的实例。
|
// @receiver s: sGameKv的实例。
|
||||||
// @return err: 错误信息,如果操作成功,则为nil。
|
// @return err: 错误信息,如果操作成功,则为nil。
|
||||||
func (s *sGameKv) SavesV1() (err error) {
|
func (s *sGameKv) SavesV1() (err error) {
|
||||||
|
// 最大允许执行时间
|
||||||
|
RunTimeMax = gtime.Now().Add(time.Minute * 30)
|
||||||
|
g.Log().Debug(ctx, "开始执行游戏kv数据保存")
|
||||||
|
|
||||||
getCache, err := pkg.Cache("redis").Get(nil, "cron:game_kv")
|
getCache, err := pkg.Cache("redis").Get(nil, "cron:game_kv")
|
||||||
//如果没有执行过,设置时间戳
|
//如果没有执行过,设置时间戳
|
||||||
if getCache.Int64() > 0 {
|
if getCache.Int64() > 0 {
|
||||||
@@ -48,6 +54,12 @@ func (s *sGameKv) SavesV1() (err error) {
|
|||||||
// 从Redis列表中获取所有用户KV索引的键
|
// 从Redis列表中获取所有用户KV索引的键
|
||||||
//keys, err := utils.RedisScan("user:kv:*")
|
//keys, err := utils.RedisScan("user:kv:*")
|
||||||
err = tools.Redis.RedisScanV2("user:kv:*", func(keys []string) (err error) {
|
err = tools.Redis.RedisScanV2("user:kv:*", func(keys []string) (err error) {
|
||||||
|
//判断是否超时
|
||||||
|
if gtime.Now().After(RunTimeMax) {
|
||||||
|
g.Log().Error(ctx, "执行超时了,停止执行!")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
// 定义用于存储用户数据的结构体
|
// 定义用于存储用户数据的结构体
|
||||||
type ListData struct {
|
type ListData struct {
|
||||||
Uid int64 `json:"uid"`
|
Uid int64 `json:"uid"`
|
||||||
@@ -107,8 +119,7 @@ func (s *sGameKv) SavesV1() (err error) {
|
|||||||
for _, v := range delKey {
|
for _, v := range delKey {
|
||||||
_, err2 = g.Redis().Del(ctx, v)
|
_, err2 = g.Redis().Del(ctx, v)
|
||||||
if err2 != nil {
|
if err2 != nil {
|
||||||
g.Log().Errorf(ctx, "删除存档错误:%v,err=%v", v, err2)
|
g.Log().Errorf(ctx, "删除存档失败:%v,err=%v", v, err2)
|
||||||
return
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -122,17 +133,5 @@ func (s *sGameKv) SavesV1() (err error) {
|
|||||||
return
|
return
|
||||||
})
|
})
|
||||||
|
|
||||||
//if err != nil {
|
|
||||||
// return err
|
|
||||||
//}
|
|
||||||
////跳过
|
|
||||||
//if len(keys) == 0 {
|
|
||||||
// return
|
|
||||||
//}
|
|
||||||
////一次最多处理10w条
|
|
||||||
//if len(keys) > 10000 {
|
|
||||||
// keys = keys[:10000]
|
|
||||||
//}
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,13 +1,14 @@
|
|||||||
package ip2region
|
package ip2region
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"net"
|
||||||
|
"strings"
|
||||||
|
|
||||||
"github.com/ayflying/utility_go/service"
|
"github.com/ayflying/utility_go/service"
|
||||||
"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/gctx"
|
||||||
"github.com/gogf/gf/v2/os/gfile"
|
"github.com/gogf/gf/v2/os/gfile"
|
||||||
"github.com/lionsoul2014/ip2region/binding/golang/xdb"
|
"github.com/lionsoul2014/ip2region/binding/golang/xdb"
|
||||||
"net"
|
|
||||||
"strings"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -15,6 +16,8 @@ var (
|
|||||||
wait = false
|
wait = false
|
||||||
)
|
)
|
||||||
|
|
||||||
|
const IpDbPath = "runtime/library/ip2region.xdb"
|
||||||
|
|
||||||
type sIp2region struct {
|
type sIp2region struct {
|
||||||
searcher *xdb.Searcher
|
searcher *xdb.Searcher
|
||||||
}
|
}
|
||||||
@@ -35,16 +38,17 @@ func init() {
|
|||||||
// Load 加载到内存中
|
// Load 加载到内存中
|
||||||
//
|
//
|
||||||
// @Description: 加载ip2region数据库到内存中。
|
// @Description: 加载ip2region数据库到内存中。
|
||||||
|
|
||||||
// @receiver s *sIp2region: sIp2region的实例。
|
// @receiver s *sIp2region: sIp2region的实例。
|
||||||
func (s *sIp2region) Load() {
|
func (s *sIp2region) Load() {
|
||||||
var err error
|
var err error
|
||||||
|
|
||||||
var dbPath = "runtime/library/ip2region.xdb"
|
//var dbPath = "runtime/library/ip2region.xdb"
|
||||||
var url = "https://github.com/ayflying/resource/raw/refs/heads/main/attachment/ip2region.xdb"
|
var url = "https://github.com/ayflying/resource/raw/refs/heads/master/attachment/ip2region.xdb"
|
||||||
if wait {
|
if wait {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if gfile.IsEmpty(dbPath) {
|
if gfile.IsEmpty(IpDbPath) {
|
||||||
wait = true
|
wait = true
|
||||||
defer func() {
|
defer func() {
|
||||||
wait = false
|
wait = false
|
||||||
@@ -55,9 +59,9 @@ func (s *sIp2region) Load() {
|
|||||||
if err2 != nil {
|
if err2 != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
err = gfile.PutBytes(dbPath, putData.ReadAll())
|
err = gfile.PutBytes(IpDbPath, putData.ReadAll())
|
||||||
}
|
}
|
||||||
cBuff := gfile.GetBytes(dbPath)
|
cBuff := gfile.GetBytes(IpDbPath)
|
||||||
/*
|
/*
|
||||||
var cBuff []byte
|
var cBuff []byte
|
||||||
if gres.Contains(dbPath) {
|
if gres.Contains(dbPath) {
|
||||||
@@ -78,7 +82,9 @@ func (s *sIp2region) Load() {
|
|||||||
|
|
||||||
func (s *sIp2region) GetIp(ip string) (res []string) {
|
func (s *sIp2region) GetIp(ip string) (res []string) {
|
||||||
//初始化加载
|
//初始化加载
|
||||||
s.Load()
|
if s.searcher == nil {
|
||||||
|
s.Load()
|
||||||
|
}
|
||||||
|
|
||||||
res = make([]string, 5)
|
res = make([]string, 5)
|
||||||
if s.searcher == nil {
|
if s.searcher == nil {
|
||||||
|
|||||||
@@ -2,6 +2,9 @@ package systemCron
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/ayflying/utility_go/api/system/v1"
|
"github.com/ayflying/utility_go/api/system/v1"
|
||||||
"github.com/ayflying/utility_go/service"
|
"github.com/ayflying/utility_go/service"
|
||||||
"github.com/gogf/gf/v2/frame/g"
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
@@ -9,8 +12,6 @@ import (
|
|||||||
"github.com/gogf/gf/v2/os/gctx"
|
"github.com/gogf/gf/v2/os/gctx"
|
||||||
"github.com/gogf/gf/v2/os/gtime"
|
"github.com/gogf/gf/v2/os/gtime"
|
||||||
"github.com/gogf/gf/v2/os/gtimer"
|
"github.com/gogf/gf/v2/os/gtimer"
|
||||||
"sync"
|
|
||||||
"time"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -80,7 +81,8 @@ func (s *sSystemCron) AddCron(typ v1.CronType, _func func() error) {
|
|||||||
var _func2 = func(ctx context.Context) error {
|
var _func2 = func(ctx context.Context) error {
|
||||||
return _func()
|
return _func()
|
||||||
}
|
}
|
||||||
s.AddCronV2(typ, _func2)
|
// 老版本计划任务全都是主服务器唯一执行
|
||||||
|
s.AddCronV2(typ, _func2, true)
|
||||||
}
|
}
|
||||||
|
|
||||||
// AddCronV2 添加一个定时任务到相应的调度列表中。
|
// AddCronV2 添加一个定时任务到相应的调度列表中。
|
||||||
@@ -89,7 +91,16 @@ func (s *sSystemCron) AddCron(typ v1.CronType, _func func() error) {
|
|||||||
// @receiver s: sSystemCron的实例,代表一个调度系统。
|
// @receiver s: sSystemCron的实例,代表一个调度系统。
|
||||||
// @param typ: 任务的类型,决定该任务将被添加到哪个列表中。对应不同的时间间隔。
|
// @param typ: 任务的类型,决定该任务将被添加到哪个列表中。对应不同的时间间隔。
|
||||||
// @param _func: 要添加的任务函数,该函数执行时应该返回一个error。
|
// @param _func: 要添加的任务函数,该函数执行时应该返回一个error。
|
||||||
func (s *sSystemCron) AddCronV2(typ v1.CronType, _func func(context.Context) error) {
|
// @param _onlyMain: 是否只在主服务器上执行一次,true 唯一执行,false 全局执行不判断唯一
|
||||||
|
func (s *sSystemCron) AddCronV2(typ v1.CronType, _func func(context.Context) error, _onlyMain ...bool) {
|
||||||
|
//如果传过来的任务是需要主服务器执行一次
|
||||||
|
if len(_onlyMain) > 0 && _onlyMain[0] {
|
||||||
|
//判断当前是否为主服务器
|
||||||
|
if !g.Cfg().MustGet(gctx.New(), "game.cron_main").Bool() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
//加锁
|
//加锁
|
||||||
s.Lock.Lock()
|
s.Lock.Lock()
|
||||||
defer s.Lock.Unlock()
|
defer s.Lock.Unlock()
|
||||||
|
|||||||
399
package/gamelog/sdk.go
Normal file
399
package/gamelog/sdk.go
Normal file
@@ -0,0 +1,399 @@
|
|||||||
|
package gamelog
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"compress/gzip"
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
|
"github.com/gogf/gf/v2/net/gclient"
|
||||||
|
"github.com/gogf/gf/v2/os/gfile"
|
||||||
|
"github.com/gogf/gf/v2/os/gtime"
|
||||||
|
)
|
||||||
|
|
||||||
|
type sendBody struct {
|
||||||
|
Pid string `json:"pid"`
|
||||||
|
Data [][]any `json:"data"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// todo 游戏日志对象
|
||||||
|
type GameLog struct {
|
||||||
|
Uid string // 唯一uid
|
||||||
|
Event string // 事件名
|
||||||
|
Property map[string]any // 事件属性
|
||||||
|
EventTimems int64 // 时间戳毫秒级别
|
||||||
|
EventTimeLoc string // 带时区的本地时间字符串
|
||||||
|
}
|
||||||
|
|
||||||
|
type SDKConfig struct {
|
||||||
|
// 配置变量
|
||||||
|
Pid string // 项目id
|
||||||
|
BaseUrl string // 日志服务器地址
|
||||||
|
ReportSk string // 上报解密key
|
||||||
|
FlushInterval int // 刷新间隔
|
||||||
|
DiskBakPath string // 磁盘备份路径
|
||||||
|
RetryN int // 每N次重试
|
||||||
|
ChanSize int // 信道大小, 默认1000
|
||||||
|
|
||||||
|
reportN int
|
||||||
|
}
|
||||||
|
|
||||||
|
type SDK struct {
|
||||||
|
// 控制变量
|
||||||
|
wg sync.WaitGroup
|
||||||
|
shutdown chan struct{}
|
||||||
|
mu sync.Mutex
|
||||||
|
sdkConfig *SDKConfig
|
||||||
|
bufferChan chan GameLog // 日志队列
|
||||||
|
buffer []GameLog // 日志队列
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
ctx = context.Background()
|
||||||
|
gamelogClient *gclient.Client
|
||||||
|
|
||||||
|
// location map
|
||||||
|
// locationMap map[string]*time.Location = map[string]*time.Location{}
|
||||||
|
locationMap sync.Map // 声明一个线程安全的Map
|
||||||
|
|
||||||
|
)
|
||||||
|
|
||||||
|
func getLocationMapValue(key string) *time.Location {
|
||||||
|
// 1. 先尝试读
|
||||||
|
value, loaded := locationMap.Load(key)
|
||||||
|
if loaded {
|
||||||
|
return value.(*time.Location) // 如果已经存在,直接返回
|
||||||
|
}
|
||||||
|
// 2. 不存在,就初始化一个该key对应的**固定的**新值
|
||||||
|
location, err := time.LoadLocation(key)
|
||||||
|
if err != nil {
|
||||||
|
g.Log().Warningf(ctx, "[GameLog]load location error, try use local timezone: %v", err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
// 3. 核心:原子性地存储,如果key已存在则返回已存在的值
|
||||||
|
actualValue, loaded := locationMap.LoadOrStore(key, location)
|
||||||
|
if loaded {
|
||||||
|
// 如果loaded为true,说明其他goroutine抢先存了
|
||||||
|
// 我们可以丢弃刚创建的newValue(如果有需要的话),返回已存在的actualValue
|
||||||
|
return actualValue.(*time.Location)
|
||||||
|
}
|
||||||
|
// 如果loaded为false,说明是我们存成功的,返回我们刚创建的newValue
|
||||||
|
return actualValue.(*time.Location)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sdk *SDK) varinit() error {
|
||||||
|
sdk.sdkConfig = &SDKConfig{}
|
||||||
|
|
||||||
|
_pid, err := g.Config().Get(ctx, "angergs.bisdk.pid")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
sdk.sdkConfig.Pid = _pid.String()
|
||||||
|
|
||||||
|
_baseUrl, err := g.Config().Get(ctx, "angergs.bisdk.recodeServerBaseUrl")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
sdk.sdkConfig.BaseUrl = _baseUrl.String()
|
||||||
|
|
||||||
|
_sk, err := g.Config().Get(ctx, "angergs.bisdk.reportSk")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
sdk.sdkConfig.ReportSk = _sk.String()
|
||||||
|
|
||||||
|
_flushInterval, err := g.Config().Get(ctx, "angergs.bisdk.flushInterval")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
sdk.sdkConfig.FlushInterval = _flushInterval.Int()
|
||||||
|
|
||||||
|
_diskBakPath, err := g.Config().Get(ctx, "angergs.bisdk.diskBakPath")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
sdk.sdkConfig.DiskBakPath = _diskBakPath.String()
|
||||||
|
|
||||||
|
_retryN, err := g.Config().Get(ctx, "angergs.bisdk.retryN")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
sdk.sdkConfig.RetryN = _retryN.Int()
|
||||||
|
|
||||||
|
_chanSize, err := g.Config().Get(ctx, "angergs.bisdk.chanSize")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
sdk.sdkConfig.ChanSize = _chanSize.Int()
|
||||||
|
|
||||||
|
g.Log().Infof(ctx, "[GameLog]client init success, config: %v", sdk.sdkConfig)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sdk *SDK) checkConfig() error {
|
||||||
|
config := sdk.sdkConfig
|
||||||
|
if config.Pid == "" {
|
||||||
|
return fmt.Errorf("pid is empty")
|
||||||
|
}
|
||||||
|
if config.BaseUrl == "" {
|
||||||
|
return fmt.Errorf("baseUrl is empty")
|
||||||
|
}
|
||||||
|
if config.ReportSk == "" {
|
||||||
|
return fmt.Errorf("reportSk is empty")
|
||||||
|
}
|
||||||
|
if config.FlushInterval <= 0 {
|
||||||
|
return fmt.Errorf("flushInterval is invalid")
|
||||||
|
}
|
||||||
|
if config.DiskBakPath == "" {
|
||||||
|
return fmt.Errorf("diskBakPath is empty")
|
||||||
|
}
|
||||||
|
if config.RetryN == 0 {
|
||||||
|
config.RetryN = 10
|
||||||
|
}
|
||||||
|
if config.ChanSize == 0 {
|
||||||
|
config.ChanSize = 1000
|
||||||
|
}
|
||||||
|
config.DiskBakPath = strings.TrimSuffix(config.DiskBakPath, "/")
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func INIT(config *SDKConfig) (*SDK, error) {
|
||||||
|
// 加载并检查配置
|
||||||
|
sdk := &SDK{}
|
||||||
|
if config != nil {
|
||||||
|
sdk.sdkConfig = config
|
||||||
|
} else if err := sdk.varinit(); err != nil { // 可以读goframe的配置
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := sdk.checkConfig(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
gamelogClient = g.Client()
|
||||||
|
|
||||||
|
// 初始化队列
|
||||||
|
sdk.shutdown = make(chan struct{})
|
||||||
|
sdk.bufferChan = make(chan GameLog, 1000)
|
||||||
|
sdk.buffer = make([]GameLog, 0, 100)
|
||||||
|
// 加载失败日志
|
||||||
|
failLogs, err := sdk.loadFailLogs4disk()
|
||||||
|
if err != nil {
|
||||||
|
g.Log().Errorf(ctx, "[GameLog]load fail logs error: %v", err)
|
||||||
|
} else if len(failLogs) > 0 {
|
||||||
|
sdk.buffer = append(sdk.buffer, failLogs...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 开启协程进行日志发送
|
||||||
|
sdk.wg = sync.WaitGroup{}
|
||||||
|
sdk.wg.Add(1)
|
||||||
|
go func() {
|
||||||
|
defer sdk.wg.Done()
|
||||||
|
ticker := time.NewTicker(time.Duration(sdk.sdkConfig.FlushInterval) * time.Second)
|
||||||
|
defer ticker.Stop()
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-sdk.shutdown:
|
||||||
|
// 关闭时, 上传一次并备份失败数据
|
||||||
|
g.Log().Infof(ctx, "[GameLog]begin shutdown and flush last")
|
||||||
|
sdk.flush()
|
||||||
|
return
|
||||||
|
case log := <-sdk.bufferChan:
|
||||||
|
sdk.buffer = append(sdk.buffer, log)
|
||||||
|
case <-ticker.C:
|
||||||
|
sdk.flush()
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
return sdk, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 从磁盘加载失败日志
|
||||||
|
func (sdk *SDK) loadFailLogs4disk() (logs []GameLog, err error) {
|
||||||
|
if !gfile.Exists(sdk.sdkConfig.DiskBakPath) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// 遍历diskBakPath下所有failBufferxxx.bak.log文件, 读取到log中
|
||||||
|
files, err := gfile.ScanDir(sdk.sdkConfig.DiskBakPath, "failBuffer*.bak.log")
|
||||||
|
logs = []GameLog{}
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// 读取每个备份文件
|
||||||
|
for _, fp := range files {
|
||||||
|
// 每一行都是一次失败的记录
|
||||||
|
gfile.ReadLines(fp, func(line string) error {
|
||||||
|
_logs := []GameLog{}
|
||||||
|
err := json.Unmarshal([]byte(line), &_logs)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
// 合并到总日志列表
|
||||||
|
logs = append(logs, _logs...)
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
g.Log().Infof(ctx, "[GameLog]load %d faillogs from %s", len(logs), fp)
|
||||||
|
gfile.Remove(fp)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// 备份失败日志追加到磁盘
|
||||||
|
func (sdk *SDK) bakFailLogs2disk(failLogs []GameLog) {
|
||||||
|
bakPath := fmt.Sprintf("%s/failBuffer%s.bak.log", sdk.sdkConfig.DiskBakPath, gtime.Now().Format("YmdH"))
|
||||||
|
content, err := json.Marshal(failLogs)
|
||||||
|
if err != nil {
|
||||||
|
g.Log().Errorf(ctx, "[GameLog]marshal fail logs error: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
gfile.PutContentsAppend(bakPath, string(content)+"\n")
|
||||||
|
g.Log().Infof(ctx, "[GameLog]backup fail buffer to %s", bakPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 优雅关闭
|
||||||
|
func (sdk *SDK) Shutdown() {
|
||||||
|
close(sdk.shutdown)
|
||||||
|
sdk.wg.Wait()
|
||||||
|
}
|
||||||
|
|
||||||
|
// 日志时间格式
|
||||||
|
const datetimeFmt = time.DateOnly + " " + time.TimeOnly
|
||||||
|
|
||||||
|
// 记录日志
|
||||||
|
func (sdk *SDK) Log(uid, event string, property map[string]any, timezone string) {
|
||||||
|
loc := time.Local
|
||||||
|
if _loc := getLocationMapValue(timezone); _loc != nil {
|
||||||
|
loc = _loc
|
||||||
|
}
|
||||||
|
log := GameLog{
|
||||||
|
Uid: uid,
|
||||||
|
Event: event,
|
||||||
|
Property: property,
|
||||||
|
EventTimems: gtime.Now().TimestampMilli(),
|
||||||
|
EventTimeLoc: gtime.Now().In(loc).Format(datetimeFmt),
|
||||||
|
}
|
||||||
|
// 线程安全
|
||||||
|
sdk.bufferChan <- log
|
||||||
|
}
|
||||||
|
|
||||||
|
// 按服务器时区记录日志
|
||||||
|
func (sdk *SDK) LogLtz(uid, event string, property map[string]any) {
|
||||||
|
sdk.Log(uid, event, property, time.Local.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
// 这个方法只会在内部协程调用
|
||||||
|
func (sdk *SDK) flush() {
|
||||||
|
sdk.mu.Lock()
|
||||||
|
defer sdk.mu.Unlock()
|
||||||
|
if len(sdk.buffer) == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
batch := make([]GameLog, len(sdk.buffer))
|
||||||
|
copy(batch, sdk.buffer)
|
||||||
|
sdk.buffer = sdk.buffer[:0]
|
||||||
|
|
||||||
|
// 第N次的时候加载失败数据进行尝试
|
||||||
|
if sdk.sdkConfig.reportN != 0 && sdk.sdkConfig.reportN%sdk.sdkConfig.RetryN == 0 {
|
||||||
|
faillogs, err := sdk.loadFailLogs4disk()
|
||||||
|
if err != nil {
|
||||||
|
g.Log().Errorf(ctx, "[GameLog]load fail logs error: %v", err)
|
||||||
|
}
|
||||||
|
// 如果有失败日志则加入到批量数组中
|
||||||
|
if len(faillogs) > 0 {
|
||||||
|
batch = append(batch, faillogs...)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
sdk.send(batch)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 发送消息
|
||||||
|
func (sdk *SDK) send(logs []GameLog) {
|
||||||
|
waitSecond := time.Duration(sdk.sdkConfig.FlushInterval/4) * time.Second
|
||||||
|
timeoutCtx, cancel := context.WithTimeout(context.Background(), waitSecond)
|
||||||
|
defer cancel()
|
||||||
|
data := make([][]any, 0, len(logs))
|
||||||
|
// logs 拆分成二维数组
|
||||||
|
for _, log := range logs {
|
||||||
|
propertyJson, err := json.Marshal(log.Property)
|
||||||
|
if err != nil {
|
||||||
|
g.Log().Errorf(ctx, "[GameLog]skip log parse, marshal property error: %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
data = append(data, []any{
|
||||||
|
log.Uid,
|
||||||
|
log.Event,
|
||||||
|
string(propertyJson),
|
||||||
|
log.EventTimems,
|
||||||
|
log.EventTimeLoc,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
// json化
|
||||||
|
sbody := sendBody{
|
||||||
|
Pid: sdk.sdkConfig.Pid,
|
||||||
|
Data: data,
|
||||||
|
}
|
||||||
|
jsonBody, err := json.Marshal(sbody)
|
||||||
|
if err != nil {
|
||||||
|
g.Log().Errorf(ctx, "[GameLog]marshal send body error: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// giz压缩
|
||||||
|
gzBody := bytes.NewBuffer([]byte{})
|
||||||
|
gz := gzip.NewWriter(gzBody)
|
||||||
|
gz.Write(jsonBody)
|
||||||
|
gz.Close()
|
||||||
|
|
||||||
|
// XOR 加密
|
||||||
|
xorBody := bytesXOR(gzBody.Bytes(), []byte(sdk.sdkConfig.ReportSk))
|
||||||
|
|
||||||
|
sdk.sdkConfig.reportN += 1
|
||||||
|
res, err := gamelogClient.Post(timeoutCtx, sdk.sdkConfig.BaseUrl+"/report/event", xorBody)
|
||||||
|
// 失败重新加入缓冲区
|
||||||
|
if err != nil {
|
||||||
|
sdk.bakFailLogs2disk(logs)
|
||||||
|
g.Log().Warningf(ctx, "[GameLog]send log error, bak to fail buffer(%d): %v", len(logs), err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer func() {
|
||||||
|
cerr := res.Close()
|
||||||
|
if cerr != nil {
|
||||||
|
g.Log().Errorf(ctx, "[GameLog]close response error: %v", cerr)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
httpcode := res.StatusCode
|
||||||
|
resBody := res.ReadAllString()
|
||||||
|
// 收集器拦截, 重新加入缓冲区
|
||||||
|
if httpcode != http.StatusOK {
|
||||||
|
sdk.bakFailLogs2disk(logs)
|
||||||
|
g.Log().Warningf(ctx, "[GameLog]send log error, bak to fail buffer(%d): %v", len(logs), resBody)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 混淆
|
||||||
|
func bytesXOR(data []byte, key []byte) []byte {
|
||||||
|
obfuscated := make([]byte, len(data))
|
||||||
|
keyLen := len(key)
|
||||||
|
if keyLen == 0 {
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := range data {
|
||||||
|
obfuscated[i] = data[i] ^ key[i%keyLen]
|
||||||
|
}
|
||||||
|
return obfuscated
|
||||||
|
|
||||||
|
// // 使用示例
|
||||||
|
// key := []byte{0x12, 0x34, 0x56, 0x78}
|
||||||
|
// obfuscated := multiXorObfuscate(original, key)
|
||||||
|
// deobfuscated := multiXorObfuscate(obfuscated, key) // 解密
|
||||||
|
}
|
||||||
52
package/gamelog/test/gamelog_test.go
Normal file
52
package/gamelog/test/gamelog_test.go
Normal file
@@ -0,0 +1,52 @@
|
|||||||
|
package test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/ayflying/utility_go/package/gamelog"
|
||||||
|
"github.com/gogf/gf/v2/test/gtest"
|
||||||
|
"github.com/gogf/gf/v2/util/grand"
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestGamelog(t *testing.T) {
|
||||||
|
glsdk, err := gamelog.INIT(&gamelog.SDKConfig{
|
||||||
|
// 必填
|
||||||
|
Pid: "test5", // 项目ID
|
||||||
|
BaseUrl: "http://47.76.178.47:10101", // 香港测试服上报地址
|
||||||
|
// BaseUrl: "http://127.0.0.1:10101", // 本次测试上报地址
|
||||||
|
ReportSk: "sngame2025", // xor混淆key
|
||||||
|
FlushInterval: 5, // 上报间隔
|
||||||
|
DiskBakPath: "gamelog", // 本地磁盘备份, 用于意外情况下临时保存日志, 请确保该目录持久化(容器内要挂载). 每次启动时或每N次上报时加载到失败队列
|
||||||
|
// 可填
|
||||||
|
RetryN: 2, // 默认每10次, 上传检查一次磁盘的失败数据
|
||||||
|
ChanSize: 500, // 默认1000, 信道size
|
||||||
|
})
|
||||||
|
|
||||||
|
// 随机测试事件和属性
|
||||||
|
events := []string{"e1", "e2", "e3", "e4"}
|
||||||
|
pms := []map[string]any{
|
||||||
|
{"a": "1"},
|
||||||
|
{"a": "2"},
|
||||||
|
{"a": "3"},
|
||||||
|
{"a": "4"},
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
gtest.C(t, func(t *gtest.T) {
|
||||||
|
go func() {
|
||||||
|
for {
|
||||||
|
uuidval, _ := uuid.NewUUID()
|
||||||
|
randUid := strings.ReplaceAll(uuidval.String(), "-", "")
|
||||||
|
glsdk.LogLtz(randUid, events[grand.Intn(len(events))], pms[grand.Intn(len(pms))])
|
||||||
|
time.Sleep(time.Millisecond * 100)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
time.Sleep(time.Second * 14)
|
||||||
|
// 模拟等待信号后优雅关闭
|
||||||
|
glsdk.Shutdown()
|
||||||
|
})
|
||||||
|
}
|
||||||
32
package/pay/chongchong/chongchong.go
Normal file
32
package/pay/chongchong/chongchong.go
Normal file
@@ -0,0 +1,32 @@
|
|||||||
|
package chongchong
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/ayflying/utility_go/package/pay/common"
|
||||||
|
"github.com/gogf/gf/v2/crypto/gmd5"
|
||||||
|
)
|
||||||
|
|
||||||
|
//验单
|
||||||
|
func (p *Pay) Verify(req *CallbackData, sign string) (isOk bool, err error) {
|
||||||
|
//req := g.RequestFromCtx(ctx).Request
|
||||||
|
//data, err := common.ParseNotifyToBodyMap(req)
|
||||||
|
|
||||||
|
var data = map[string]interface{}{
|
||||||
|
"orderPrice": req.OrderPrice,
|
||||||
|
"packageId": req.PackageId,
|
||||||
|
"partnerTransactionNo": req.PartnerTransactionNo,
|
||||||
|
"productId": req.ProductId,
|
||||||
|
"statusCode": req.StatusCode,
|
||||||
|
"transactionNo": req.TransactionNo,
|
||||||
|
}
|
||||||
|
|
||||||
|
dataStr, err := common.BuildSignStr(data)
|
||||||
|
|
||||||
|
var SingStr = fmt.Sprintf("%v&%v", dataStr, p.ApiKey)
|
||||||
|
sign2, err := gmd5.EncryptString(SingStr)
|
||||||
|
|
||||||
|
if sign == sign2 {
|
||||||
|
isOk = true
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
25
package/pay/chongchong/model.go
Normal file
25
package/pay/chongchong/model.go
Normal file
@@ -0,0 +1,25 @@
|
|||||||
|
package chongchong
|
||||||
|
|
||||||
|
type Pay struct {
|
||||||
|
ApiKey string `json:"api_key"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func New(pay *Pay) *Pay {
|
||||||
|
return &Pay{
|
||||||
|
ApiKey: pay.ApiKey,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// CallbackData 用于处理回调数据的结构体
|
||||||
|
type CallbackData struct {
|
||||||
|
TransactionNo string `json:"transactionNo" dc:"平台交易单号,唯一标识一笔交易"`
|
||||||
|
PartnerTransactionNo string `json:"partnerTransactionNo" dc:"合作方交易单号,由合作方生成"`
|
||||||
|
StatusCode string `json:"statusCode" dc:"交易状态码,SUCCESS表示成功,FAIL表示失败"`
|
||||||
|
ProductId int `json:"productId" dc:"产品ID,对应后台配置的商品"`
|
||||||
|
OrderPrice float64 `json:"orderPrice" dc:"订单金额,单位为元"`
|
||||||
|
PackageId int `json:"packageId" dc:"套餐ID,可选字段,部分商品有套餐区分"`
|
||||||
|
ProductName string `json:"productName" dc:"产品名称,展示用"`
|
||||||
|
ExtParam string `json:"extParam" dc:"扩展参数,回调时原样返回"`
|
||||||
|
UserId int `json:"userId" dc:"用户ID,标识购买者"`
|
||||||
|
Sign string `json:"sign" dc:"签名,用于验证请求合法性"`
|
||||||
|
}
|
||||||
105
package/pay/common/func.go
Normal file
105
package/pay/common/func.go
Normal file
@@ -0,0 +1,105 @@
|
|||||||
|
package common
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
|
"github.com/gogf/gf/v2/util/gconv"
|
||||||
|
"net/http"
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// FormatPublicKey 将原始公钥字符串格式化为标准PEM格式的公钥
|
||||||
|
// 功能:为原始公钥添加PEM头部和尾部,并按64字符长度拆分换行,符合PKCS#8标准格式要求
|
||||||
|
// 参数 publicKey: 原始未格式化的公钥字符串(通常为Base64编码且无换行)
|
||||||
|
// 返回值: 格式化后的PEM格式公钥字符串
|
||||||
|
func FormatPublicKey(publicKey string) (pKey string) {
|
||||||
|
var buffer strings.Builder
|
||||||
|
// 写入PEM格式头部
|
||||||
|
buffer.WriteString("-----BEGIN PUBLIC KEY-----\n")
|
||||||
|
|
||||||
|
// 定义每行公钥的标准长度(PEM格式要求64字符/行)
|
||||||
|
rawLen := 64
|
||||||
|
keyLen := len(publicKey)
|
||||||
|
// 计算需要拆分的总行数(向上取整)
|
||||||
|
raws := keyLen / rawLen
|
||||||
|
temp := keyLen % rawLen
|
||||||
|
if temp > 0 {
|
||||||
|
raws++ // 若有余数则增加一行
|
||||||
|
}
|
||||||
|
|
||||||
|
// 按行拆分并写入公钥内容
|
||||||
|
start := 0
|
||||||
|
end := start + rawLen
|
||||||
|
for i := 0; i < raws; i++ {
|
||||||
|
if i == raws-1 {
|
||||||
|
// 最后一行取剩余所有字符(处理不足64字符的情况)
|
||||||
|
buffer.WriteString(publicKey[start:])
|
||||||
|
} else {
|
||||||
|
// 非最后行取固定64字符
|
||||||
|
buffer.WriteString(publicKey[start:end])
|
||||||
|
}
|
||||||
|
buffer.WriteByte('\n') // 每行结束添加换行符
|
||||||
|
start += rawLen
|
||||||
|
end = start + rawLen
|
||||||
|
}
|
||||||
|
|
||||||
|
// 写入PEM格式尾部
|
||||||
|
buffer.WriteString("-----END PUBLIC KEY-----\n")
|
||||||
|
pKey = buffer.String()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// ParseNotifyToBodyMap 将HTTP请求中的表单数据解析为键值对映射
|
||||||
|
// 功能:解析请求表单数据,提取单值字段并转换为map[string]interface{}格式
|
||||||
|
// 参数 req: 包含表单数据的HTTP请求对象
|
||||||
|
// 返回值: 解析后的键值对映射(bm)和可能的错误(err)
|
||||||
|
func ParseNotifyToBodyMap(req *http.Request) (bm map[string]interface{}, err error) {
|
||||||
|
// 解析请求表单数据,若失败则返回错误
|
||||||
|
if err = req.ParseForm(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
// 获取解析后的表单数据(key为字段名,value为字符串切片形式的字段值)
|
||||||
|
var form map[string][]string = req.Form
|
||||||
|
// 初始化结果映射,预分配容量(表单字段数+1,预留扩展空间)
|
||||||
|
bm = make(map[string]interface{}, len(form)+1)
|
||||||
|
// 遍历表单字段,仅保留单值字段(忽略多值字段)
|
||||||
|
for k, v := range form {
|
||||||
|
if len(v) == 1 {
|
||||||
|
bm[k] = v[0]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// BuildSignStr 根据传入的g.Map构建签名字符串
|
||||||
|
// 规则:对所有非空值的键进行字母排序后,按"key=value&"格式拼接,最后去除末尾的"&"
|
||||||
|
// 参数 bm: 包含键值对的g.Map
|
||||||
|
// 返回值: 构建好的签名字符串和可能的错误
|
||||||
|
func BuildSignStr(bm g.Map) (string, error) {
|
||||||
|
var (
|
||||||
|
buf strings.Builder
|
||||||
|
keyList []string
|
||||||
|
)
|
||||||
|
// 收集所有键名
|
||||||
|
for k := range bm {
|
||||||
|
keyList = append(keyList, k)
|
||||||
|
}
|
||||||
|
// 对键名进行字母排序
|
||||||
|
sort.Strings(keyList)
|
||||||
|
// 遍历排序后的键,拼接非空值的键值对
|
||||||
|
for _, k := range keyList {
|
||||||
|
if v := bm[k]; v != "" {
|
||||||
|
buf.WriteString(k)
|
||||||
|
buf.WriteByte('=')
|
||||||
|
buf.WriteString(gconv.String(v))
|
||||||
|
buf.WriteByte('&')
|
||||||
|
// 去除末尾多余的'&'字符
|
||||||
|
// 检查是否有有效的键值对被拼接
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if buf.Len() <= 0 {
|
||||||
|
return "", errors.New("length is error")
|
||||||
|
}
|
||||||
|
return buf.String()[:buf.Len()-1], nil
|
||||||
|
}
|
||||||
5
package/pay/honor/const.go
Normal file
5
package/pay/honor/const.go
Normal file
@@ -0,0 +1,5 @@
|
|||||||
|
package honor
|
||||||
|
|
||||||
|
const (
|
||||||
|
Host = "https://iap-api.cloud.honor.com"
|
||||||
|
)
|
||||||
70
package/pay/honor/honor.go
Normal file
70
package/pay/honor/honor.go
Normal file
@@ -0,0 +1,70 @@
|
|||||||
|
package honor
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"crypto"
|
||||||
|
"crypto/rsa"
|
||||||
|
"crypto/sha256"
|
||||||
|
"crypto/x509"
|
||||||
|
"encoding/base64"
|
||||||
|
"encoding/pem"
|
||||||
|
"errors"
|
||||||
|
"github.com/ayflying/utility_go/package/pay/common"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Pay struct {
|
||||||
|
PubKey string `json:"pubKey"`
|
||||||
|
AppId string `json:"appId"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func New(pay *Pay) *Pay {
|
||||||
|
return &Pay{
|
||||||
|
AppId: pay.AppId,
|
||||||
|
PubKey: pay.PubKey,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// VerifyRSASignature 验证RSA数字签名
|
||||||
|
// data: 原始数据字节
|
||||||
|
// sign: 签名的Base64编码字符串
|
||||||
|
// pubKey: PEM格式的公钥字符串
|
||||||
|
// 返回验证结果和可能的错误
|
||||||
|
func (p *Pay) VerifyRSASignature(ctx context.Context, data []byte, sign string) (bool, error) {
|
||||||
|
//req := g.RequestFromCtx(ctx).Request
|
||||||
|
//post, err := common.ParseNotifyToBodyMap(req)
|
||||||
|
//var data = gjson.MustEncode(post)
|
||||||
|
|
||||||
|
// 解码Base64格式的签名
|
||||||
|
signBytes, err := base64.StdEncoding.DecodeString(sign)
|
||||||
|
if err != nil {
|
||||||
|
return false, errors.New("签名解码失败: " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
pubkey := common.FormatPublicKey(p.PubKey)
|
||||||
|
// 解析PEM格式的公钥
|
||||||
|
block, _ := pem.Decode([]byte(pubkey))
|
||||||
|
if block == nil {
|
||||||
|
return false, errors.New("无效的PEM格式公钥")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 解析公钥
|
||||||
|
publicKey, err := x509.ParsePKIXPublicKey(block.Bytes)
|
||||||
|
if err != nil {
|
||||||
|
return false, errors.New("公钥解析失败: " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
// 类型断言为公钥
|
||||||
|
rsaPubKey, ok := publicKey.(*rsa.PublicKey)
|
||||||
|
if !ok {
|
||||||
|
return false, errors.New("不是有效的RSA公钥")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 计算数据的SHA-256哈希
|
||||||
|
hasher := sha256.New()
|
||||||
|
hasher.Write(data)
|
||||||
|
hash := hasher.Sum(nil)
|
||||||
|
|
||||||
|
// 验证签名
|
||||||
|
err = rsa.VerifyPKCS1v15(rsaPubKey, crypto.SHA256, hash, signBytes)
|
||||||
|
return err == nil, err
|
||||||
|
}
|
||||||
45
package/pay/honor/model.go
Normal file
45
package/pay/honor/model.go
Normal file
@@ -0,0 +1,45 @@
|
|||||||
|
package honor
|
||||||
|
|
||||||
|
type PayCallbackHeader struct {
|
||||||
|
Charset string `json:"charset" dc:"字符集,当前只支持utf-8。"`
|
||||||
|
SignType string `json:"signType" dc:"签名算法类型, 当前只支持RSA"`
|
||||||
|
Sign string `json:"sign" dc:"notificationMessage的签名,已废弃,请用signature。"`
|
||||||
|
Signature string `json:"signature" dc:"对data的签名。"`
|
||||||
|
AppId string `json:"appId" dc:"应用ID"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type PayCallback struct {
|
||||||
|
Env string `json:"env" dc:"发送通知的环境,sandbox为沙盒测试环境,非sandbox为正式环境"`
|
||||||
|
EventType string `json:"eventType" dc:"事件类型,如付款成功、退款失败等"`
|
||||||
|
EventCode int `json:"eventCode" dc:"事件类型对应的code值"`
|
||||||
|
Version string `json:"version" dc:"iap版本"`
|
||||||
|
EventTime string `json:"eventTime" dc:"通知时间"`
|
||||||
|
Data PayCallbackData `json:"data" dc:"通知内容notificationMessage的json字符串"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type PayCallbackData struct {
|
||||||
|
AppId string `json:"appId" dc:"应用ID"`
|
||||||
|
OrderId string `json:"orderId" dc:"订单ID"`
|
||||||
|
BizOrderNo string `json:"bizOrderNo,omitempty" dc:"max-length:64#业务订单号"`
|
||||||
|
ProductType int `json:"productType" dc:"商品类型0:消耗型,1:非消耗型,2:订阅型"`
|
||||||
|
ProductId string `json:"productId" dc:"商品ID"`
|
||||||
|
ProductName string `json:"productName" dc:"商品名称"`
|
||||||
|
PurchaseTime int64 `json:"purchaseTime" dc:"购买时间UTC时间戳(毫秒)"`
|
||||||
|
PurchaseState int `json:"purchaseState" dc:"订单状态 0:已购买 1:已退款 2:付款失败 3:退款失败 4:未支付 5:退款中"`
|
||||||
|
ConsumptionState int `json:"consumptionState" dc:"消耗状态 0:未消耗 1:已消耗"`
|
||||||
|
PurchaseToken string `json:"purchaseToken" dc:"购买令牌"`
|
||||||
|
Currency string `json:"currency" dc:"币种"`
|
||||||
|
Price string `json:"price" dc:"商品价格"`
|
||||||
|
PayMoney string `json:"payMoney" dc:"实际支付金额"`
|
||||||
|
DeveloperPayload string `json:"developerPayload,omitempty" dc:"max-length:1024#商户信息"`
|
||||||
|
OriOrder string `json:"oriOrder" dc:"原订单信息"`
|
||||||
|
SandboxFlag int `json:"sandboxFlag" dc:"沙盒标识"`
|
||||||
|
AgreementNo string `json:"agreementNo,omitempty" dc:"订阅合约号"`
|
||||||
|
ExecuteTime string `json:"executeTime,omitempty" dc:"下次扣费时间(订阅)"`
|
||||||
|
SecondChargeTime int64 `json:"secondChargeTime,omitempty" dc:"第二次扣费时间(订阅升级)"`
|
||||||
|
OldProductId string `json:"oldProductId,omitempty" dc:"老商品ID(订阅升级)"`
|
||||||
|
SubStartTime string `json:"subStartTime,omitempty" dc:"订阅开始时间"`
|
||||||
|
SubEndTime string `json:"subEndTime,omitempty" dc:"订阅结束时间"`
|
||||||
|
OriginalPrice string `json:"originalPrice" dc:"原始价格"`
|
||||||
|
CancelTime string `json:"cancelTime,omitempty" dc:"订阅取消时间"`
|
||||||
|
}
|
||||||
26
package/pay/honor/notification.go
Normal file
26
package/pay/honor/notification.go
Normal file
@@ -0,0 +1,26 @@
|
|||||||
|
package honor
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
|
"github.com/gogf/gf/v2/os/gctx"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (p *Pay) Notification(r *http.Request) {
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// ConsumeProduct 商品消耗
|
||||||
|
func (p *Pay) ConsumeProduct(purchaseToken string) (err error) {
|
||||||
|
url := Host + "/iap/server/consumeProduct"
|
||||||
|
_, err = g.Client().ContentJson().Post(gctx.New(), url, g.Map{
|
||||||
|
"purchaseToken": purchaseToken,
|
||||||
|
"developerChallenge": "",
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
25
package/pay/huawei/consts.go
Normal file
25
package/pay/huawei/consts.go
Normal file
@@ -0,0 +1,25 @@
|
|||||||
|
package huawei
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
TokenUrl = "https://oauth-login.cloud.huawei.com/oauth2/v3/token"
|
||||||
|
)
|
||||||
|
|
||||||
|
func getOrderUrl(accountFlag int) string {
|
||||||
|
if accountFlag == 1 {
|
||||||
|
// site for telecom carrier
|
||||||
|
//return "https://orders-at-dre.iap.dbankcloud.com"
|
||||||
|
return "https://orders-drcn.iap.cloud.huawei.com.cn"
|
||||||
|
} else {
|
||||||
|
// TODO: replace the (ip:port) to the real one
|
||||||
|
return "http://exampleserver/_mockserver_"
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// default http client with 5 seconds timeout
|
||||||
|
var RequestHttpClient = http.Client{Timeout: time.Second * 5}
|
||||||
166
package/pay/huawei/huawei.go
Normal file
166
package/pay/huawei/huawei.go
Normal file
@@ -0,0 +1,166 @@
|
|||||||
|
package huawei
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"crypto"
|
||||||
|
"crypto/rsa"
|
||||||
|
"crypto/sha256"
|
||||||
|
"crypto/x509"
|
||||||
|
"encoding/base64"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"github.com/gogf/gf/v2/encoding/gjson"
|
||||||
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
|
"github.com/gogf/gf/v2/os/gctx"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Pay struct {
|
||||||
|
ClientSecret string `json:"client_secret"`
|
||||||
|
ClientId string `json:"client_id"`
|
||||||
|
//TokenUrl string `json:"token_url"`
|
||||||
|
ApplicationPublicKey string `json:"application_public_key"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func New(cfg *Pay) *Pay {
|
||||||
|
return cfg
|
||||||
|
}
|
||||||
|
|
||||||
|
// ConfirmPurchase 发货后确认购买接口(华为支付)消耗商品
|
||||||
|
// 功能:通知华为支付平台当前订单已完成发货,触发支付完成流程(需在商品实际发货后调用)
|
||||||
|
// 参数说明:
|
||||||
|
// purchaseToken: 华为支付返回的购买令牌(唯一标识一笔具体的购买交易,由客户端支付成功后返回)
|
||||||
|
// productId: 应用内商品的唯一标识(需与客户端发起支付时使用的productId一致)
|
||||||
|
// accountFlag: 账户标识(用于区分不同账户体系/环境,如0-普通用户、1-企业用户,具体值由业务定义)
|
||||||
|
func (p *Pay) ConfirmPurchase(purchaseToken, productId string, accountFlag int) {
|
||||||
|
// 构造请求体参数(包含购买令牌和产品ID)
|
||||||
|
bodyMap := map[string]string{
|
||||||
|
"purchaseToken": purchaseToken, // 华为支付返回的购买凭证
|
||||||
|
"productId": productId, // 对应应用内商品的唯一标识
|
||||||
|
}
|
||||||
|
url := getOrderUrl(accountFlag) + "/applications/v2/purchases/confirm"
|
||||||
|
bodyBytes, err := p.SendRequest(url, bodyMap)
|
||||||
|
if err != nil {
|
||||||
|
// 请求失败时记录错误日志(实际业务中建议增加重试或异常处理逻辑)
|
||||||
|
log.Printf("err is %s", err)
|
||||||
|
}
|
||||||
|
// 打印响应结果(实际业务中需替换为具体处理逻辑,如更新订单状态、校验响应数据等)
|
||||||
|
// TODO: 建议根据华为支付文档解析响应数据(如检查responseCode是否为0表示成功)
|
||||||
|
log.Printf("%s", bodyBytes)
|
||||||
|
}
|
||||||
|
|
||||||
|
// VerifyToken 验证回调订单
|
||||||
|
//您可以调用本接口向华为应用内支付服务器校验支付结果中的购买令牌,确认支付结果的准确性。
|
||||||
|
func (p *Pay) VerifyToken(purchaseToken, productId string, accountFlag int) (res *PurchaseTokenData, err error) {
|
||||||
|
bodyMap := map[string]string{"purchaseToken": purchaseToken, "productId": productId}
|
||||||
|
url := getOrderUrl(accountFlag) + "/applications/purchases/tokens/verify"
|
||||||
|
bodyBytes, err := p.SendRequest(url, bodyMap)
|
||||||
|
if err != nil {
|
||||||
|
g.Log().Error(gctx.New(), "err is %s", err)
|
||||||
|
}
|
||||||
|
var data *VerifyTokenRes
|
||||||
|
err = gjson.DecodeTo(bodyBytes, &data)
|
||||||
|
err = gjson.DecodeTo(data.PurchaseTokenData, &res)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Pay) SendRequest(url string, bodyMap map[string]string) (string, error) {
|
||||||
|
authHeaderString, err := p.BuildAuthorization()
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
bodyString, err := json.Marshal(bodyMap)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
req, err := http.NewRequest("POST", url, bytes.NewReader(bodyString))
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
req.Header.Set("Content-Type", "application/json; charset=UTF-8")
|
||||||
|
req.Header.Set("Authorization", authHeaderString)
|
||||||
|
response, err := RequestHttpClient.Do(req)
|
||||||
|
defer response.Body.Close()
|
||||||
|
bodyBytes, err := io.ReadAll(response.Body)
|
||||||
|
|
||||||
|
//req, err := g.Client().Header(g.MapStrStr{
|
||||||
|
// "Content-Type": "application/json; charset=UTF-8",
|
||||||
|
// "Authorization": authHeaderString,
|
||||||
|
//}).Post(gctx.New(), url, bodyString)
|
||||||
|
//defer req.Close()
|
||||||
|
//var bodyBytes = req.ReadAll()
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return string(bodyBytes), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Pay) VerifyRsaSign(content string, sign string, publicKey string) error {
|
||||||
|
//publicKey = common.FormatPublicKey(publicKey)
|
||||||
|
publicKeyByte, err := base64.StdEncoding.DecodeString(publicKey)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
pub, err := x509.ParsePKIXPublicKey(publicKeyByte)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
hashed := sha256.Sum256([]byte(content))
|
||||||
|
signature, err := base64.StdEncoding.DecodeString(sign)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return rsa.VerifyPKCS1v15(pub.(*rsa.PublicKey), crypto.SHA256, hashed[:], signature)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Pay) GetAppAt() (string, error) {
|
||||||
|
//demoConfig := GetDefaultConfig()
|
||||||
|
urlValue := url.Values{
|
||||||
|
"grant_type": {"client_credentials"},
|
||||||
|
"client_secret": {p.ClientSecret},
|
||||||
|
"client_id": {p.ClientId},
|
||||||
|
}
|
||||||
|
resp, err := RequestHttpClient.PostForm(TokenUrl, urlValue)
|
||||||
|
defer resp.Body.Close()
|
||||||
|
bodyBytes, err := io.ReadAll(resp.Body)
|
||||||
|
|
||||||
|
//post := g.MapStrStr{
|
||||||
|
// "grant_type": "client_credentials",
|
||||||
|
// "client_secret": p.ClientSecret,
|
||||||
|
// "client_id": p.ClientId,
|
||||||
|
//}
|
||||||
|
//resp, err := g.Client().PostForm(gctx.New(), p.TokenUrl, post)
|
||||||
|
//if err != nil {
|
||||||
|
// return "", err
|
||||||
|
//}
|
||||||
|
//resp.Close()
|
||||||
|
//bodyBytes := resp.ReadAll()
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
var atResponse AtResponse
|
||||||
|
json.Unmarshal(bodyBytes, &atResponse)
|
||||||
|
if atResponse.AccessToken != "" {
|
||||||
|
return atResponse.AccessToken, nil
|
||||||
|
} else {
|
||||||
|
return "", errors.New("Get token fail, " + string(bodyBytes))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Pay) BuildAuthorization() (string, error) {
|
||||||
|
appAt, err := p.GetAppAt()
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
oriString := fmt.Sprintf("APPAT:%s", appAt)
|
||||||
|
var authString = base64.StdEncoding.EncodeToString([]byte(oriString))
|
||||||
|
var authHeaderString = fmt.Sprintf("Basic %s", authString)
|
||||||
|
return authHeaderString, nil
|
||||||
|
}
|
||||||
82
package/pay/huawei/model.go
Normal file
82
package/pay/huawei/model.go
Normal file
@@ -0,0 +1,82 @@
|
|||||||
|
package huawei
|
||||||
|
|
||||||
|
type CallbackType struct {
|
||||||
|
Version string `json:"version"`
|
||||||
|
NotifyTime int64 `json:"notifyTime"`
|
||||||
|
EventType string `json:"eventType"`
|
||||||
|
ApplicationId string `json:"applicationId"`
|
||||||
|
OrderNotification *OrderNotification `json:"orderNotification"`
|
||||||
|
SubNotification *SubNotification `json:"subNotification"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type OrderNotification struct {
|
||||||
|
Version string `json:"version" dc:"通知版本:v2"`
|
||||||
|
NotificationType int `json:"notificationType" dc:"通知事件的类型,取值如下:1:支付成功 2:退款成功"`
|
||||||
|
PurchaseToken string `json:"purchaseToken" dc:"待下发商品的购买Token"`
|
||||||
|
ProductId string `json:"productId" dc:"商品ID"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type SubNotification struct {
|
||||||
|
StatusUpdateNotification *StatusUpdateNotification `json:"statusUpdateNotification" dc:"通知消息"`
|
||||||
|
NotificationSignature string `json:"notificationSignature" dc:"对statusUpdateNotification字段的签名字符串,签名算法为signatureAlgorithm表示的签名算法。"`
|
||||||
|
Version string `json:"version" dc:"通知版本:v2"`
|
||||||
|
SignatureAlgorithm string `json:"signatureAlgorithm" dc:"签名算法。"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// StatusUpdateNotification 订阅状态更新通知
|
||||||
|
type StatusUpdateNotification struct {
|
||||||
|
Environment string `json:"environment" dc:"发送通知的环境。PROD:正式环境;Sandbox:沙盒测试"`
|
||||||
|
NotificationType int `json:"notificationType" dc:"通知事件的类型,具体定义需参考相关文档说明"`
|
||||||
|
SubscriptionID string `json:"subscriptionId" dc:"订阅ID"`
|
||||||
|
CancellationDate int64 `json:"cancellationDate" dc:"撤销订阅时间或退款时间,UTC时间戳,以毫秒为单位,仅在notificationType取值为CANCEL的场景下会传入"`
|
||||||
|
OrderID string `json:"orderId" dc:"订单ID,唯一标识一笔需要收费的收据,由华为应用内支付服务器在创建订单以及订阅型商品续费时生成。每一笔新的收据都会使用不同的orderId。通知类型为NEW_RENEWAL_PREF时不存在"`
|
||||||
|
LatestReceipt string `json:"latestReceipt" dc:"最近的一笔收据的token,仅在notificationType取值为INITIAL_BUY 、RENEWAL或INTERACTIVE_RENEWAL并且续期成功情况下传入"`
|
||||||
|
LatestReceiptInfo string `json:"latestReceiptInfo" dc:"最近的一笔收据,JSON字符串格式,包含的参数请参见InappPurchaseDetails,在notificationType取值为CANCEL时无值"`
|
||||||
|
LatestReceiptInfoSignature string `json:"latestReceiptInfoSignature" dc:"对latestReceiptInfo的签名字符串,签名算法为statusUpdateNotification中的signatureAlgorithm。服务器在收到签名字符串后,需要参见对返回结果验签使用IAP公钥对latestReceiptInfo的JSON字符串进行验签。公钥获取请参见查询支付服务信息"`
|
||||||
|
LatestExpiredReceipt string `json:"latestExpiredReceipt" dc:"最近的一笔过期收据的token"`
|
||||||
|
LatestExpiredReceiptInfo string `json:"latestExpiredReceiptInfo" dc:"最近的一笔过期收据,JSON字符串格式,在notificationType取值为RENEWAL或INTERACTIVE_RENEWAL时有值"`
|
||||||
|
LatestExpiredReceiptInfoSignature string `json:"latestExpiredReceiptInfoSignature" dc:"对latestExpiredReceiptInfo的签名字符串,签名算法为statusUpdateNotification中的signatureAlgorithm。服务器在收到签名字符串后,需要参见对返回结果验签使用IAP公钥对latestExpiredReceiptInfo的JSON字符串进行验签。公钥获取请参见查询支付服务信息"`
|
||||||
|
AutoRenewStatus int `json:"autoRenewStatus" dc:"续期状态。取值说明:1:当前周期到期后正常续期;0:用户已终止续期"`
|
||||||
|
RefundPayOrderId string `json:"refundPayOrderId" dc:"退款交易号,在notificationType取值为CANCEL时有值"`
|
||||||
|
ProductID string `json:"productId" dc:"订阅型商品ID"`
|
||||||
|
ApplicationID string `json:"applicationId" dc:"应用ID"`
|
||||||
|
ExpirationIntent int `json:"expirationIntent" dc:"超期原因,仅在notificationType为RENEWAL或INTERACTIVE_RENEWAL时并且续期失败情况下有值"`
|
||||||
|
PurchaseToken string `json:"purchaseToken" dc:"订阅token,与上述订阅ID字段subscriptionId对应。"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type AtResponse struct {
|
||||||
|
AccessToken string `json:"access_token"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type VerifyTokenRes struct {
|
||||||
|
ResponseCode string `json:"responseCode"`
|
||||||
|
PurchaseTokenData string `json:"purchaseTokenData"`
|
||||||
|
DataSignature string `json:"dataSignature"`
|
||||||
|
SignatureAlgorithm string `json:"signatureAlgorithm"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type PurchaseTokenData struct {
|
||||||
|
AutoRenewing bool `json:"autoRenewing" dc:"表示订阅是否自动续费"`
|
||||||
|
OrderId string `json:"orderId" dc:"订单ID,唯一标识一笔订单"`
|
||||||
|
PackageName string `json:"packageName" dc:"应用的包名"`
|
||||||
|
ApplicationId int `json:"applicationId" dc:"应用ID,以整数形式表示"`
|
||||||
|
ApplicationIdString string `json:"applicationIdString" dc:"应用ID的字符串形式"`
|
||||||
|
Kind int `json:"kind" dc:"购买类型的某种标识,具体含义可能取决于业务逻辑"`
|
||||||
|
ProductId string `json:"productId" dc:"商品ID,用于标识购买的商品"`
|
||||||
|
ProductName string `json:"productName" dc:"商品名称"`
|
||||||
|
PurchaseTime int64 `json:"purchaseTime" dc:"购买时间,可能是某种特定格式的时间表示"`
|
||||||
|
PurchaseTimeMillis int64 `json:"purchaseTimeMillis" dc:"购买时间,以毫秒为单位的时间戳"`
|
||||||
|
PurchaseState int `json:"purchaseState" dc:"购买状态,不同的整数值代表不同的状态,具体含义取决于业务逻辑"`
|
||||||
|
DeveloperPayload string `json:"developerPayload" dc:"开发者自定义负载数据"`
|
||||||
|
PurchaseToken string `json:"purchaseToken" dc:"购买令牌"`
|
||||||
|
ResponseCode string `json:"responseCode" dc:"响应代码,用于表示购买操作的响应结果"`
|
||||||
|
ConsumptionState int `json:"consumptionState" dc:"消费状态,不同的整数值代表不同的消费状态,具体含义取决于业务逻辑"`
|
||||||
|
Confirmed int `json:"confirmed" dc:"确认状态,不同的整数值代表不同的确认情况,具体含义取决于业务逻辑"`
|
||||||
|
PurchaseType int `json:"purchaseType" dc:"购买类型,不同的整数值代表不同的购买类型,具体含义取决于业务逻辑"`
|
||||||
|
Currency string `json:"currency" dc:"货币类型"`
|
||||||
|
Price int `json:"price" dc:"商品价格"`
|
||||||
|
Country string `json:"country" dc:"购买所在国家"`
|
||||||
|
PayOrderId string `json:"payOrderId" dc:"支付订单ID"`
|
||||||
|
PayType string `json:"payType" dc:"支付类型"`
|
||||||
|
SdkChannel string `json:"sdkChannel" dc:"SDK渠道"`
|
||||||
|
}
|
||||||
107
package/pay/huawei/notification.go
Normal file
107
package/pay/huawei/notification.go
Normal file
@@ -0,0 +1,107 @@
|
|||||||
|
package huawei
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Copyright 2020. Huawei Technologies Co., Ltd. All rights reserved.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
//import "encoding/json"
|
||||||
|
|
||||||
|
const (
|
||||||
|
INITIAL_BUY = 0
|
||||||
|
CANCEL = 1
|
||||||
|
RENEWAL = 2
|
||||||
|
INTERACTIVE_RENEWAL = 3
|
||||||
|
NEW_RENEWAL_PREF = 4
|
||||||
|
RENEWAL_STOPPED = 5
|
||||||
|
RENEWAL_RESTORED = 6
|
||||||
|
RENEWAL_RECURRING = 7
|
||||||
|
ON_HOLD = 9
|
||||||
|
PAUSED = 10
|
||||||
|
PAUSE_PLAN_CHANGED = 11
|
||||||
|
PRICE_CHANGE_CONFIRMED = 12
|
||||||
|
DEFERRED = 13
|
||||||
|
)
|
||||||
|
|
||||||
|
//type NotificationServer struct {
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
//var NotificationDemo = &NotificationServer{}
|
||||||
|
|
||||||
|
//type NotificationRequest struct {
|
||||||
|
// StatusUpdateNotification string `json:"statusUpdateNotification"`
|
||||||
|
// NotificationSignature string `json:"notifycationSignature"`
|
||||||
|
//}
|
||||||
|
|
||||||
|
//
|
||||||
|
//type NotificationResponse struct {
|
||||||
|
// ErrorCode string `json:"errorCode"`
|
||||||
|
// ErrorMsg string `json:"errorMsg"`
|
||||||
|
//}
|
||||||
|
|
||||||
|
//type StatusUpdateNotification struct {
|
||||||
|
// Environment string `json:"environment"`
|
||||||
|
// NotificationType int `json:"notificationType"`
|
||||||
|
// SubscriptionID string `json:"subscriptionId"`
|
||||||
|
// CancellationDate int64 `json:"cancellationDate"`
|
||||||
|
// OrderID string `json:"orderId"`
|
||||||
|
// LatestReceipt string `json:"latestReceipt"`
|
||||||
|
// LatestReceiptInfo string `json:"latestReceiptInfo"`
|
||||||
|
// LatestReceiptInfoSignature string `json:"latestReceiptInfoSignature"`
|
||||||
|
// LatestExpiredReceipt string `json:"latestExpiredReceipt"`
|
||||||
|
// LatestExpiredReceiptInfo string `json:"latestExpiredReceiptInfo"`
|
||||||
|
// LatestExpiredReceiptInfoSignature string `json:"latestExpiredReceiptInfoSignature"`
|
||||||
|
// AutoRenewStatus int `json:"autoRenewStatus"`
|
||||||
|
// RefundPayOrderId string `json:"refundPayOrderId"`
|
||||||
|
// ProductID string `json:"productId"`
|
||||||
|
// ApplicationID string `json:"applicationId"`
|
||||||
|
// ExpirationIntent int `json:"expirationIntent"`
|
||||||
|
//}
|
||||||
|
|
||||||
|
func (p *Pay) DealNotification(information string) (err error) {
|
||||||
|
//var request PayCallback
|
||||||
|
//err = json.Unmarshal([]byte(information), &request)
|
||||||
|
//if err != nil {
|
||||||
|
// return
|
||||||
|
//}
|
||||||
|
//err = p.VerifyRsaSign(request.StatusUpdateNotification, request.NotificationSignature, DefaultConfig.ApplicationPublicKey)
|
||||||
|
//if err != nil {
|
||||||
|
// return
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
//var info = request.StatusUpdateNotification
|
||||||
|
////json.Unmarshal([]byte(request.StatusUpdateNotification), &info)
|
||||||
|
//switch notificationType := info.NotificationType; notificationType {
|
||||||
|
//case INITIAL_BUY:
|
||||||
|
//case CANCEL:
|
||||||
|
//case RENEWAL:
|
||||||
|
//case INTERACTIVE_RENEWAL:
|
||||||
|
//case NEW_RENEWAL_PREF:
|
||||||
|
//case RENEWAL_STOPPED:
|
||||||
|
//case RENEWAL_RESTORED:
|
||||||
|
//case RENEWAL_RECURRING:
|
||||||
|
//case ON_HOLD:
|
||||||
|
//case PAUSED:
|
||||||
|
//case PAUSE_PLAN_CHANGED:
|
||||||
|
//case PRICE_CHANGE_CONFIRMED:
|
||||||
|
//case DEFERRED:
|
||||||
|
//default:
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
////response := NotificationResponse{ErrorCode: "0"}
|
||||||
|
////return &response, nil
|
||||||
|
//return
|
||||||
|
return
|
||||||
|
}
|
||||||
9
package/pay/oppo/const.go
Normal file
9
package/pay/oppo/const.go
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
package oppo
|
||||||
|
|
||||||
|
const (
|
||||||
|
LoginUrl = "https://iopen.game.oppomobile.com/sdkopen/user/fileIdInfo?fileId=%s&token=%s"
|
||||||
|
LocationShanghai = "Asia/Shanghai"
|
||||||
|
|
||||||
|
RSA = "RSA"
|
||||||
|
RSA2 = "RSA2"
|
||||||
|
)
|
||||||
97
package/pay/oppo/login.go
Normal file
97
package/pay/oppo/login.go
Normal file
@@ -0,0 +1,97 @@
|
|||||||
|
package oppo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"crypto/hmac"
|
||||||
|
"crypto/sha1"
|
||||||
|
"encoding/base64"
|
||||||
|
"github.com/gogf/gf/v2/encoding/gjson"
|
||||||
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
|
"github.com/gogf/gf/v2/os/gtime"
|
||||||
|
"github.com/gogf/gf/v2/util/grand"
|
||||||
|
"io"
|
||||||
|
"net/url"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
const host = "https://iopen.game.heytapmobi.com"
|
||||||
|
|
||||||
|
// oppo参数类型
|
||||||
|
type LoginType struct {
|
||||||
|
Token string `json:"token"`
|
||||||
|
Ssoid string `json:"ssoid"`
|
||||||
|
Channel int `json:"channel"`
|
||||||
|
AdId string `json:"adId"`
|
||||||
|
}
|
||||||
|
|
||||||
|
//登录回复
|
||||||
|
type LoginResType struct {
|
||||||
|
ResultCode string `json:"resultCode" dc:"响应码,成功为 200"`
|
||||||
|
ResultMsg string `json:"resultMsg" dc:"响应信息"`
|
||||||
|
LoginToken string `json:"loginToken" dc:"透传的token"`
|
||||||
|
Ssoid string `json:"ssoid" dc:"透传的ssoid"`
|
||||||
|
//AppKey string `json:"appKey" dc:"秘钥key,因隐私安全规范,该字段目前已不返回信息"`
|
||||||
|
UserName string `json:"userName" dc:"用户ssoid绑定的账户昵称"`
|
||||||
|
//Email string `json:"email" dc:"因隐私安全规范,该字段目前已不返回信息"`
|
||||||
|
//MobileNumber string `json:"mobileNumber" dc:"因隐私安全规范,该字段目前已不返回信息"`
|
||||||
|
//CreateTime string `json:"createTime" dc:"因隐私安全规范,该字段目前已不返回信息"`
|
||||||
|
UserStatus string `json:"userStatus" dc:"用户状态:NORMAL 表示正常"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *OppoType) FileIdInfo(ctx context.Context, oauthToken string, ssoid string) (res *LoginResType, err error) {
|
||||||
|
url := host + "/sdkopen/user/fileIdInfo"
|
||||||
|
header := p.GetHeader(oauthToken)
|
||||||
|
getHtml, err := g.Client().Header(header).Get(ctx, url, g.Map{
|
||||||
|
"token": oauthToken,
|
||||||
|
"fileId": ssoid,
|
||||||
|
})
|
||||||
|
getRes := getHtml.ReadAllString()
|
||||||
|
gjson.DecodeTo(getRes, &res)
|
||||||
|
//g.Log().Debugf(ctx, "当前登陆请求的:%v", res)
|
||||||
|
return
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *OppoType) GenParam(oauthToken, oauthTimestamp, oauthNonce string) string {
|
||||||
|
// 注意:拼接的顺序不能有改变,不然会导致联运方验签失败
|
||||||
|
params := []string{
|
||||||
|
"oauthConsumerKey=" + url.QueryEscape(p.AppKey),
|
||||||
|
"oauthToken=" + url.QueryEscape(oauthToken),
|
||||||
|
"oauthSignatureMethod=" + url.QueryEscape("HMAC-SHA1"),
|
||||||
|
"oauthTimestamp=" + url.QueryEscape(oauthTimestamp),
|
||||||
|
"oauthNonce=" + url.QueryEscape(oauthNonce),
|
||||||
|
"oauthVersion=" + url.QueryEscape("1.0"),
|
||||||
|
}
|
||||||
|
return strings.Join(params, "&") + "&"
|
||||||
|
}
|
||||||
|
|
||||||
|
// 生成签名
|
||||||
|
func (p *OppoType) GenOauthSignature(param string) string {
|
||||||
|
oauthSignatureKey := p.AppSecret + "&"
|
||||||
|
mac := hmac.New(sha1.New, []byte(oauthSignatureKey))
|
||||||
|
io.WriteString(mac, param)
|
||||||
|
signature := base64.StdEncoding.EncodeToString(mac.Sum(nil))
|
||||||
|
return url.QueryEscape(signature)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *OppoType) GetHeader(oauthToken string) (headers map[string]string) {
|
||||||
|
|
||||||
|
// 没有做过 urlEncode 的 token,由游戏客户端调用 OPPO SDK 直接获取
|
||||||
|
//oauthToken := "TICKET_Ajnxxxxx"
|
||||||
|
oauthTimestamp := gtime.Now().TimestampStr()
|
||||||
|
oauthNonce := grand.S(5)
|
||||||
|
|
||||||
|
// 生成请求头参数和签名
|
||||||
|
param := p.GenParam(oauthToken, oauthTimestamp, oauthNonce)
|
||||||
|
oauthSignature := p.GenOauthSignature(param)
|
||||||
|
|
||||||
|
// 封装请求头
|
||||||
|
headers = map[string]string{
|
||||||
|
"param": param,
|
||||||
|
"oauthSignature": oauthSignature,
|
||||||
|
}
|
||||||
|
|
||||||
|
//fmt.Println("游戏服务端登录鉴权请求头为:", headers)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
13
package/pay/oppo/model.go
Normal file
13
package/pay/oppo/model.go
Normal file
@@ -0,0 +1,13 @@
|
|||||||
|
package oppo
|
||||||
|
|
||||||
|
// OPPO支付回调参数结构体
|
||||||
|
type PayCallback struct {
|
||||||
|
NotifyId string `json:"notifyId" dc:"回调通知单号,以GC开头,必填,示例:GC20230314657000"`
|
||||||
|
PartnerOrder string `json:"partnerOrder" dc:"开发者订单号,必填,示例:123456"`
|
||||||
|
ProductName string `json:"productName" dc:"商品名称,必填,示例:10元宝"`
|
||||||
|
ProductDesc string `json:"productDesc" dc:"商品描述,必填,示例:10元宝等于1元"`
|
||||||
|
Price int64 `json:"price" dc:"商品价格,单位为分,需要游戏服务端做验证,必填,示例:100"`
|
||||||
|
Count int `json:"count" dc:"商品数量(一般为1),必填,示例:1"`
|
||||||
|
Attach string `json:"attach" dc:"请求支付时上传的附加参数,可能为空,选填"`
|
||||||
|
Sign string `json:"sign" dc:"OPPO服务端签名,需要游戏服务端做验证,必填"`
|
||||||
|
}
|
||||||
32
package/pay/oppo/notify.go
Normal file
32
package/pay/oppo/notify.go
Normal file
@@ -0,0 +1,32 @@
|
|||||||
|
package oppo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (p *OppoType) ParseNotifyToBodyMap(req *http.Request) (bm map[string]interface{}, err error) {
|
||||||
|
if err = req.ParseForm(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
var form map[string][]string = req.Form
|
||||||
|
bm = make(map[string]interface{}, len(form)+1)
|
||||||
|
for k, v := range form {
|
||||||
|
if len(v) == 1 {
|
||||||
|
bm[k] = v[0]
|
||||||
|
//bm.Set(k, v[0])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *OppoType) ParseNotifyByURLValues(value url.Values) (bm map[string]interface{}, err error) {
|
||||||
|
bm = make(map[string]interface{}, len(value)+1)
|
||||||
|
for k, v := range value {
|
||||||
|
if len(v) == 1 {
|
||||||
|
bm[k] = v[0]
|
||||||
|
//bm.Set(k, v[0])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
40
package/pay/oppo/oppo.go
Normal file
40
package/pay/oppo/oppo.go
Normal file
@@ -0,0 +1,40 @@
|
|||||||
|
package oppo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
|
)
|
||||||
|
|
||||||
|
// 跟充值平台通信的加密key
|
||||||
|
//const PUBLIC_KEY = `dfsdfs`
|
||||||
|
|
||||||
|
type OppoType struct {
|
||||||
|
AppId string `json:"app_id"`
|
||||||
|
AppKey string `json:"app_key"`
|
||||||
|
AppSecret string `json:"app_secret"`
|
||||||
|
PublicKey string `json:"public_key"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func New(cfg *OppoType) *OppoType {
|
||||||
|
|
||||||
|
return &OppoType{
|
||||||
|
AppKey: cfg.AppKey,
|
||||||
|
AppSecret: cfg.AppSecret,
|
||||||
|
PublicKey: cfg.PublicKey,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *OppoType) Verify(ctx context.Context) (err error) {
|
||||||
|
// OPPO公钥. 在官方给的 demo 中. 无需修改,改了就验证不过
|
||||||
|
oppoPublicKey := "MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCmreYIkPwVovKR8rLHWlFVw7YDfm9uQOJKL89Smt6ypXGVdrAKKl0wNYc3/jecAoPi2ylChfa2iRu5gunJyNmpWZzlCNRIau55fxGW0XEu553IiprOZcaw5OuYGlf60ga8QT6qToP0/dpiL/ZbmNUO9kUhosIjEu22uFgR+5cYyQIDAQAB"
|
||||||
|
//oppoPublicKey := p.PublicKey
|
||||||
|
// 解析请求参数
|
||||||
|
bodyMap, err := p.ParseNotifyToBodyMap(g.RequestFromCtx(ctx).Request)
|
||||||
|
if err != nil {
|
||||||
|
// 解析失败, 处理错误逻辑
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
err = p.VerifySign(oppoPublicKey, bodyMap)
|
||||||
|
return
|
||||||
|
}
|
||||||
80
package/pay/oppo/sign.go
Normal file
80
package/pay/oppo/sign.go
Normal file
@@ -0,0 +1,80 @@
|
|||||||
|
package oppo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto"
|
||||||
|
"crypto/hmac"
|
||||||
|
"crypto/rsa"
|
||||||
|
"crypto/sha1"
|
||||||
|
"crypto/x509"
|
||||||
|
"encoding/base64"
|
||||||
|
"encoding/pem"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"github.com/ayflying/utility_go/package/pay/common"
|
||||||
|
"github.com/gogf/gf/v2/util/gconv"
|
||||||
|
"hash"
|
||||||
|
"math/rand"
|
||||||
|
"net/url"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (p *OppoType) GenLoginBaseStr(bm map[string]interface{}, appKey, appSecret string) (string, string) {
|
||||||
|
baseStr := fmt.Sprintf("oauthConsumerKey=%s&oauthToken=%s&oauthSignatureMethod=HMAC-SHA1&oauthTimestamp=%d&oauthNonce=%d&oauthVersion=1.0&",
|
||||||
|
appKey, url.QueryEscape(gconv.String(bm["token"])), time.Now().Unix(), rand.Int31n(100000000))
|
||||||
|
|
||||||
|
var h hash.Hash
|
||||||
|
h = hmac.New(sha1.New, []byte(appSecret+"&"))
|
||||||
|
h.Write([]byte(baseStr))
|
||||||
|
|
||||||
|
sign := url.QueryEscape(base64.StdEncoding.EncodeToString(h.Sum(nil)))
|
||||||
|
return baseStr, sign
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *OppoType) VerifySign(oppoPayPublicKey string, bm map[string]interface{}) (err error) {
|
||||||
|
if oppoPayPublicKey == "" || bm == nil {
|
||||||
|
return errors.New("oppoPayPublicKey or bm is nil")
|
||||||
|
}
|
||||||
|
|
||||||
|
bodySign := bm["sign"].(string)
|
||||||
|
bodySignType := RSA
|
||||||
|
signData := fmt.Sprintf("notifyId=%s&partnerOrder=%s&productName=%s&productDesc=%s&price=%s&count=%s&attach=%s",
|
||||||
|
bm["notifyId"], bm["partnerOrder"], bm["productName"],
|
||||||
|
bm["productDesc"], bm["price"], bm["count"], bm["attach"])
|
||||||
|
pKey := common.FormatPublicKey(oppoPayPublicKey)
|
||||||
|
if err = p.verifySign(signData, bodySign, bodySignType, pKey); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *OppoType) verifySign(signData, sign, signType, oppoPayPublicKey string) (err error) {
|
||||||
|
var (
|
||||||
|
h hash.Hash
|
||||||
|
hashs crypto.Hash
|
||||||
|
block *pem.Block
|
||||||
|
pubKey interface{}
|
||||||
|
publicKey *rsa.PublicKey
|
||||||
|
ok bool
|
||||||
|
)
|
||||||
|
signBytes, _ := base64.StdEncoding.DecodeString(sign)
|
||||||
|
if block, _ = pem.Decode([]byte(oppoPayPublicKey)); block == nil {
|
||||||
|
return errors.New("OPPO公钥Decode错误")
|
||||||
|
}
|
||||||
|
if pubKey, err = x509.ParsePKIXPublicKey(block.Bytes); err != nil {
|
||||||
|
return fmt.Errorf("x509.ParsePKIXPublicKey:%w", err)
|
||||||
|
}
|
||||||
|
if publicKey, ok = pubKey.(*rsa.PublicKey); !ok {
|
||||||
|
return errors.New("OPPO公钥转换错误")
|
||||||
|
}
|
||||||
|
switch signType {
|
||||||
|
case RSA:
|
||||||
|
hashs = crypto.SHA1
|
||||||
|
case RSA2:
|
||||||
|
hashs = crypto.SHA256
|
||||||
|
default:
|
||||||
|
hashs = crypto.SHA256
|
||||||
|
}
|
||||||
|
h = hashs.New()
|
||||||
|
h.Write([]byte(signData))
|
||||||
|
return rsa.VerifyPKCS1v15(publicKey, hashs, h.Sum(nil), signBytes)
|
||||||
|
}
|
||||||
@@ -1,60 +1,62 @@
|
|||||||
package taptap
|
package taptap
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
|
||||||
"context"
|
|
||||||
"crypto/hmac"
|
"crypto/hmac"
|
||||||
"crypto/sha256"
|
"crypto/sha256"
|
||||||
"encoding/base64"
|
"encoding/base64"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/gogf/gf/v2/os/gtime"
|
|
||||||
"github.com/gogf/gf/v2/util/grand"
|
|
||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"sort"
|
"sort"
|
||||||
"strconv"
|
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
type pTapTap struct {
|
type pTapTap struct {
|
||||||
Secret string `json:"secret" dc:"秘钥"`
|
Secret string `json:"secret" dc:"秘钥"`
|
||||||
OrderId string `json:"order_id" dc:"订单唯一 ID"`
|
//OrderId string `json:"order_id" dc:"订单唯一 ID"`
|
||||||
ClientId string `json:"client_id" dc:"应用的 Client ID"`
|
ClientId string `json:"client_id" dc:"应用的 Client ID"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(orderId string) *pTapTap {
|
func New(clientId string, secret string) *pTapTap {
|
||||||
return &pTapTap{
|
return &pTapTap{
|
||||||
Secret: "5AFEWnadBA0NgJK2mxeBLQEde0qyIefxLSc4XKHsx9AwkQRhxzkQ9DixsOkK6gcV",
|
Secret: secret,
|
||||||
ClientId: "mox88lbz43edfukdgk",
|
ClientId: clientId,
|
||||||
OrderId: orderId,
|
//OrderId: orderId,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (p *pTapTap) Sign(url string, body []byte) (sign string, ts int64, nonce string, err error) {
|
// Sign signs the request.
|
||||||
//nolint:gosec
|
func (p *pTapTap) Sign(req *http.Request, secret string) (string, error) {
|
||||||
secret := p.Secret
|
//获取请求参数
|
||||||
//body := gjson.MustEncode(g.Map{})
|
//req := g.RequestFromCtx(ctx).Request
|
||||||
//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"}}`)
|
return Sign(req, secret)
|
||||||
//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
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
//func (p *pTapTap) SignOld(ctx context.Context, method, url string, token string, data any) (sign string, ts int64, nonce string, err error) {
|
||||||
|
// //secret := p.Secret
|
||||||
|
// //ts = gtime.Now().Unix()
|
||||||
|
// //nonce = grand.S(5)
|
||||||
|
// //header := http.Header{
|
||||||
|
// // "Content-Type": {"Content-Type: application/json; charset=utf-8"},
|
||||||
|
// // "X-Tap-Ts": {strconv.FormatInt(ts, 10)},
|
||||||
|
// // "X-Tap-Nonce": {nonce},
|
||||||
|
// //}
|
||||||
|
// //if method == "POST" {
|
||||||
|
// // header.Set("Content-Type", "application/json; charset=utf-8")
|
||||||
|
// //}
|
||||||
|
// ////ctx := context.Background()
|
||||||
|
// //request := g.RequestFromCtx(ctx).Request
|
||||||
|
// //body, _ := json.Marshal(data)
|
||||||
|
// ////req, err := http.NewRequestWithContext(ctx, method, url, strings.NewReader(string(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.
|
// Sign signs the request.
|
||||||
func Sign(req *http.Request, secret string) (string, error) {
|
func Sign(req *http.Request, secret string) (string, error) {
|
||||||
methodPart := req.Method
|
methodPart := req.Method
|
||||||
|
|||||||
@@ -1,59 +1,129 @@
|
|||||||
package taptap
|
package taptap
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"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/net/gclient"
|
"github.com/gogf/gf/v2/net/gclient"
|
||||||
"github.com/gogf/gf/v2/os/gctx"
|
"github.com/gogf/gf/v2/os/gctx"
|
||||||
"strconv"
|
"github.com/gogf/gf/v2/os/gtime"
|
||||||
|
"github.com/gogf/gf/v2/util/grand"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
)
|
)
|
||||||
|
|
||||||
type GetPayData struct {
|
type GetPayData struct {
|
||||||
Data struct {
|
Data struct {
|
||||||
Order struct {
|
Order struct {
|
||||||
|
Order
|
||||||
} `json:"order"`
|
} `json:"order"`
|
||||||
} `json:"data"`
|
} `json:"data"`
|
||||||
Success bool `json:"success"`
|
Success bool `json:"success"`
|
||||||
}
|
}
|
||||||
|
|
||||||
//查询订单信息
|
type Order struct {
|
||||||
func (p *pTapTap) Info(orderId string, clientId string, token []byte) (res string, err error) {
|
OrderID string `json:"order_id"` // 订单唯一 ID
|
||||||
url := fmt.Sprintf("https://cloud-payment.tapapis.cn/order/v1/info?client_id=%v&order_id=%v", orderId, clientId)
|
PurchaseToken string `json:"purchase_token"` // 用于订单核销的 token
|
||||||
res, err = p.get(url, token)
|
ClientID string `json:"client_id"` // 应用的 Client ID
|
||||||
return
|
OpenID string `json:"open_id"` // 用户的开放平台 ID
|
||||||
|
UserRegion string `json:"user_region"` // 用户地区
|
||||||
|
GoodsOpenID string `json:"goods_open_id"` // 商品唯一 ID
|
||||||
|
GoodsName string `json:"goods_name"` // 商品名称
|
||||||
|
Status PaymentStatus `json:"status"` // 订单状态
|
||||||
|
Amount string `json:"amount"` // 金额(本币金额 x 1,000,000)
|
||||||
|
Currency string `json:"currency"` // 币种
|
||||||
|
CreateTime string `json:"create_time"` // 创建时间
|
||||||
|
PayTime string `json:"pay_time"` // 支付时间
|
||||||
|
Extra string `json:"extra"` // 商户自定义数据,如角色信息等,长度不超过 255 UTF-8 字符
|
||||||
}
|
}
|
||||||
|
type PaymentStatus string
|
||||||
|
|
||||||
//验证并核销订单
|
const (
|
||||||
func (p *pTapTap) Verify(orderId string, clientId string, token []byte) (res string, err error) {
|
ChargePending PaymentStatus = "charge.pending" // 待支付
|
||||||
url := fmt.Sprintf("https://cloud-payment.tapapis.cn/order/v1/verify?client_id=%v", clientId)
|
ChargeSucceeded PaymentStatus = "charge.succeeded" //支付成功
|
||||||
res, err = p.get(url, token)
|
ChargeConfirmed PaymentStatus = "charge.confirmed" //已核销
|
||||||
return
|
ChargeOverdue PaymentStatus = "charge.overdue" //支付超时关闭
|
||||||
}
|
RefundPending PaymentStatus = "refund.pending" //退款中
|
||||||
|
RefundSucceeded PaymentStatus = "refund.succeeded" //退款成功
|
||||||
|
RefundFailed PaymentStatus = "refund.failed" //退款失败
|
||||||
|
RefundRejected PaymentStatus = "refund.rejected" //退款被拒绝
|
||||||
|
)
|
||||||
|
|
||||||
func (p *pTapTap) get(url string, token []byte, _data ...any) (res string, err error) {
|
// 查询订单信息
|
||||||
sign, ts, nonce, err := p.Sign(url, token)
|
func (p *pTapTap) Info(ctx context.Context, order string) (getPayData *GetPayData, err error) {
|
||||||
|
url := fmt.Sprintf("https://cloud-payment.tapapis.cn/order/v1/info?client_id=%v&order_id=%v", p.ClientId, order)
|
||||||
|
getPayData, err = p.get(ctx, url)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 验证并核销订单
|
||||||
|
func (p *pTapTap) Verify(ctx context.Context, req any) (getPayData *GetPayData, err error) {
|
||||||
|
url := fmt.Sprintf("https://cloud-payment.tapapis.cn/order/v1/verify?client_id=%v", p.ClientId)
|
||||||
|
getPayData, err = p.get(ctx, url, req)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *pTapTap) get(ctx context.Context, url string, _data ...any) (getPayData *GetPayData, err error) {
|
||||||
|
|
||||||
var _get *gclient.Response
|
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)
|
|
||||||
|
|
||||||
|
var header = map[string]string{
|
||||||
|
"Content-Type": "Content-Type: application/json; charset=utf-8",
|
||||||
|
"X-Tap-Nonce": grand.S(6),
|
||||||
|
"X-Tap-Ts": gtime.Now().TimestampStr(),
|
||||||
|
}
|
||||||
|
ctx2 := context.Background()
|
||||||
|
var method = "GET"
|
||||||
|
if len(_data) > 0 {
|
||||||
|
method = "POST"
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
//temp := []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"}}`)
|
||||||
|
var body io.Reader
|
||||||
|
if len(_data) > 0 {
|
||||||
|
body = bytes.NewBuffer(gjson.MustEncode(_data[0]))
|
||||||
} else {
|
} else {
|
||||||
_get, err = g.Client().Header(map[string]string{
|
body = bytes.NewBuffer([]byte{})
|
||||||
"X-Tap-Sign": sign,
|
}
|
||||||
"X-Tap-Nonce": nonce,
|
req, _ := http.NewRequestWithContext(ctx2, method, url, body)
|
||||||
"X-Tap-Ts": strconv.FormatInt(ts, 10),
|
for k, v := range header {
|
||||||
}).Post(gctx.New(), url, _data[0])
|
req.Header.Set(k, v)
|
||||||
|
}
|
||||||
|
sign, err2 := p.Sign(req, p.Secret)
|
||||||
|
if err2 != nil {
|
||||||
|
err = err2
|
||||||
|
return
|
||||||
|
}
|
||||||
|
req.Header.Set("X-Tap-Sign", sign)
|
||||||
|
header["X-Tap-Sign"] = sign
|
||||||
|
if len(_data) == 0 {
|
||||||
|
_get, err = g.Client().Header(header).ContentJson().Get(gctx.New(), url)
|
||||||
|
} else {
|
||||||
|
_get, err = g.Client().Header(header).ContentJson().Post(gctx.New(), url, _data[0])
|
||||||
}
|
}
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
res = _get.ReadAllString()
|
getPayData = &GetPayData{}
|
||||||
|
resData := _get.ReadAll()
|
||||||
|
g.Dump(resData)
|
||||||
|
if err = json.Unmarshal(resData, &getPayData); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !getPayData.Success {
|
||||||
|
err = errors.New(string(resData))
|
||||||
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,40 +0,0 @@
|
|||||||
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
|
|
||||||
}
|
|
||||||
15
package/pay/vivo/client.go
Normal file
15
package/pay/vivo/client.go
Normal file
@@ -0,0 +1,15 @@
|
|||||||
|
package vivo
|
||||||
|
|
||||||
|
type Pay struct {
|
||||||
|
AppId string
|
||||||
|
AppKey string
|
||||||
|
//AppSecret string
|
||||||
|
}
|
||||||
|
|
||||||
|
func New(cfg *Pay) (client *Pay) {
|
||||||
|
return &Pay{
|
||||||
|
AppId: cfg.AppId,
|
||||||
|
AppKey: cfg.AppKey,
|
||||||
|
//AppSecret: cfg.AppSecret,
|
||||||
|
}
|
||||||
|
}
|
||||||
9
package/pay/vivo/const.go
Normal file
9
package/pay/vivo/const.go
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
package vivo
|
||||||
|
|
||||||
|
const (
|
||||||
|
AuthTokenUrl = "https://joint-account.vivo.com.cn/cp/user/auth"
|
||||||
|
LocationShanghai = "Asia/Shanghai"
|
||||||
|
|
||||||
|
RSA = "RSA"
|
||||||
|
RSA2 = "RSA2"
|
||||||
|
)
|
||||||
34
package/pay/vivo/model.go
Normal file
34
package/pay/vivo/model.go
Normal file
@@ -0,0 +1,34 @@
|
|||||||
|
package vivo
|
||||||
|
|
||||||
|
type TokenAuthResponse struct {
|
||||||
|
ReturnCode int `json:"retcode"`
|
||||||
|
Data *TokenAuthResponseData `json:"data,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type TokenAuthResponseData struct {
|
||||||
|
Success bool `json:"success,omitempty"`
|
||||||
|
OpenId string `json:"openid,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type LoginType struct {
|
||||||
|
Token string `json:"token"`
|
||||||
|
Ssoid string `json:"ssoid"`
|
||||||
|
Channel int `json:"channel"`
|
||||||
|
AdId string `json:"adId"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type PayCallback struct {
|
||||||
|
AppId string `json:"appId"`
|
||||||
|
CpId string `json:"cpId"`
|
||||||
|
CpOrderNumber string `json:"cpOrderNumber"`
|
||||||
|
ExtInfo string `json:"extInfo"`
|
||||||
|
OrderAmount string `json:"orderAmount"`
|
||||||
|
OrderNumber string `json:"orderNumber"`
|
||||||
|
PayTime string `json:"payTime"`
|
||||||
|
RespCode string `json:"respCode"`
|
||||||
|
RespMsg string `json:"respMsg"`
|
||||||
|
SignMethod string `json:"signMethod"`
|
||||||
|
Signature string `json:"signature"`
|
||||||
|
TradeStatus string `json:"tradeStatus"`
|
||||||
|
TradeType string `json:"tradeType"`
|
||||||
|
}
|
||||||
3
package/pay/vivo/payment_api.go
Normal file
3
package/pay/vivo/payment_api.go
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
package vivo
|
||||||
|
|
||||||
|
|
||||||
51
package/pay/vivo/sign.go
Normal file
51
package/pay/vivo/sign.go
Normal file
@@ -0,0 +1,51 @@
|
|||||||
|
package vivo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"github.com/ayflying/utility_go/package/pay/common"
|
||||||
|
"github.com/gogf/gf/v2/crypto/gmd5"
|
||||||
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
|
"github.com/gogf/gf/v2/util/gconv"
|
||||||
|
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (p *Pay) VerifySign(ctx context.Context, key string) bool {
|
||||||
|
bm, _ := common.ParseNotifyToBodyMap(g.RequestFromCtx(ctx).Request)
|
||||||
|
signature := bm["signature"]
|
||||||
|
delete(bm, "signature")
|
||||||
|
delete(bm, "signMethod")
|
||||||
|
sign := p.sign(bm, key)
|
||||||
|
return signature == sign
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Pay) sign(bm g.Map, key string) string {
|
||||||
|
s, _ := p.buildSignStr(bm)
|
||||||
|
s += "&" + gmd5.MustEncrypt(key)
|
||||||
|
return gmd5.MustEncrypt(s)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Pay) buildSignStr(bm g.Map) (string, error) {
|
||||||
|
var (
|
||||||
|
buf strings.Builder
|
||||||
|
keyList []string
|
||||||
|
)
|
||||||
|
for k := range bm {
|
||||||
|
keyList = append(keyList, k)
|
||||||
|
}
|
||||||
|
sort.Strings(keyList)
|
||||||
|
for _, k := range keyList {
|
||||||
|
if v := bm[k]; v != "" {
|
||||||
|
buf.WriteString(k)
|
||||||
|
buf.WriteByte('=')
|
||||||
|
buf.WriteString(gconv.String(v))
|
||||||
|
buf.WriteByte('&')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if buf.Len() <= 0 {
|
||||||
|
return "", errors.New("length is error")
|
||||||
|
}
|
||||||
|
return buf.String()[:buf.Len()-1], nil
|
||||||
|
}
|
||||||
49
package/pay/vivo/sign_test.go
Normal file
49
package/pay/vivo/sign_test.go
Normal file
@@ -0,0 +1,49 @@
|
|||||||
|
package vivo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestVerifySign(t *testing.T) {
|
||||||
|
type args struct {
|
||||||
|
bm g.Map
|
||||||
|
key string
|
||||||
|
}
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
args args
|
||||||
|
want bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "testCase-01",
|
||||||
|
args: args{
|
||||||
|
bm: map[string]interface{}{
|
||||||
|
"appId": "111",
|
||||||
|
"cpId": "11",
|
||||||
|
"cpOrderNumber": "111",
|
||||||
|
"extInfo": "扩展参数",
|
||||||
|
"orderAmount": "1",
|
||||||
|
"orderNumber": "11",
|
||||||
|
"payTime": "20210610213219",
|
||||||
|
"respCode": "200",
|
||||||
|
"respMsg": "交易成功",
|
||||||
|
"signMethod": "MD5",
|
||||||
|
"signature": "111",
|
||||||
|
"tradeStatus": "0000",
|
||||||
|
"tradeType": "01",
|
||||||
|
"uid": "111",
|
||||||
|
},
|
||||||
|
key: "1111",
|
||||||
|
},
|
||||||
|
want: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
if got := New(nil).VerifySign(tt.args.bm, tt.args.key); got != tt.want {
|
||||||
|
t.Errorf("VerifySign() = %v, want %v", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
52
package/pay/vivo/user_api.go
Normal file
52
package/pay/vivo/user_api.go
Normal file
@@ -0,0 +1,52 @@
|
|||||||
|
package vivo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
|
"github.com/gogf/gf/v2/os/gctx"
|
||||||
|
"net/url"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (p *Pay) AuthToken(bm g.Map) (rsp *TokenAuthResponse, err error) {
|
||||||
|
if _, ok := bm["opentoken"]; !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
//err = bm.CheckEmptyError("opentoken")
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
bs, err := p.doAuthToken(bm)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
rsp = new(TokenAuthResponse)
|
||||||
|
if err = json.Unmarshal(bs, rsp); err != nil {
|
||||||
|
return nil, fmt.Errorf("json.Unmarshal(%s):%w", string(bs), err)
|
||||||
|
}
|
||||||
|
return rsp, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Pay) doAuthToken(bm g.Map) (bs []byte, err error) {
|
||||||
|
param := p.FormatURLParam(bm)
|
||||||
|
//httpClient := xhttp.NewClient()
|
||||||
|
//res, bs, errs := httpClient.Type(xhttp.TypeFormData).Post(AuthTokenUrl).SendString(param).EndBytes()
|
||||||
|
res, err := g.Client().Post(gctx.New(), AuthTokenUrl, param)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if res.StatusCode != 200 {
|
||||||
|
return nil, fmt.Errorf("HTTP Request Error, StatusCode = %d", res.StatusCode)
|
||||||
|
}
|
||||||
|
return res.ReadAll(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 格式化请求URL参数
|
||||||
|
func (p *Pay) FormatURLParam(body g.Map) (urlParam string) {
|
||||||
|
v := url.Values{}
|
||||||
|
for key, value := range body {
|
||||||
|
v.Add(key, value.(string))
|
||||||
|
}
|
||||||
|
return v.Encode()
|
||||||
|
}
|
||||||
70
package/pay/vivo/user_api_test.go
Normal file
70
package/pay/vivo/user_api_test.go
Normal file
@@ -0,0 +1,70 @@
|
|||||||
|
package vivo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
|
"reflect"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestAuthToken(t *testing.T) {
|
||||||
|
type args struct {
|
||||||
|
bm g.Map
|
||||||
|
}
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
args args
|
||||||
|
wantRsp *TokenAuthResponse
|
||||||
|
wantErr bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "testCase-01",
|
||||||
|
args: args{
|
||||||
|
bm: map[string]interface{}{
|
||||||
|
"opentoken": "_STV1_797e3324f7e3f1a3_797e3324f7e3f1a3_8db97942_Awykia3hpb90kcu3l",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wantRsp: nil,
|
||||||
|
wantErr: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
gotRsp, err := New(nil).AuthToken(tt.args.bm)
|
||||||
|
if (err != nil) != tt.wantErr {
|
||||||
|
t.Errorf("AuthToken() error = %v, wantErr %v", err, tt.wantErr)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !reflect.DeepEqual(gotRsp, tt.wantRsp) {
|
||||||
|
if gotRsp != nil {
|
||||||
|
marshal, _ := json.Marshal(gotRsp)
|
||||||
|
println(string(marshal))
|
||||||
|
}
|
||||||
|
t.Errorf("AuthToken() gotRsp = %v, want %v", gotRsp, tt.wantRsp)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestFormatURLParam(t *testing.T) {
|
||||||
|
type args struct {
|
||||||
|
body g.Map
|
||||||
|
}
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
args args
|
||||||
|
wantUrlParam string
|
||||||
|
}{
|
||||||
|
{name: "testCase-01", args: args{body: map[string]interface{}{
|
||||||
|
"opentoken": "_STV1_797e3324f7e3f1a3_797e3324f7e3f1a3_8db97942_Abbccayhpb90kvd3m",
|
||||||
|
"123": "123",
|
||||||
|
}}},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
if gotUrlParam := New(nil).FormatURLParam(tt.args.body); gotUrlParam != tt.wantUrlParam {
|
||||||
|
t.Errorf("FormatURLParam() = %v, want %v", gotUrlParam, tt.wantUrlParam)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
63
package/pay/xiaomi/helper.go
Normal file
63
package/pay/xiaomi/helper.go
Normal file
@@ -0,0 +1,63 @@
|
|||||||
|
package xiaomi
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/hmac"
|
||||||
|
"crypto/sha1"
|
||||||
|
"encoding/hex"
|
||||||
|
"fmt"
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// SignatureHelper 签名辅助类
|
||||||
|
type SignatureHelper struct{}
|
||||||
|
|
||||||
|
// hmacSHA1 计算HMAC-SHA1哈希值
|
||||||
|
func hmacSHA1(data, key string) string {
|
||||||
|
h := hmac.New(sha1.New, []byte(key))
|
||||||
|
h.Write([]byte(data))
|
||||||
|
return hex.EncodeToString(h.Sum(nil))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sign 计算hmac-sha1签名
|
||||||
|
func (m *MiPay) Sign(params map[string]string, secretKey string) string {
|
||||||
|
if _, ok := params["signature"]; ok {
|
||||||
|
delete(params, "signature")
|
||||||
|
}
|
||||||
|
for k, v := range params {
|
||||||
|
if v == "" || v == "0" {
|
||||||
|
delete(params, k)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
sortString := m.buildSortString(params)
|
||||||
|
signature := hmacSHA1(sortString, secretKey)
|
||||||
|
return signature
|
||||||
|
}
|
||||||
|
|
||||||
|
// VerifySignature 验证签名
|
||||||
|
func (m *MiPay) VerifySignature(params map[string]string, signature, secretKey string) bool {
|
||||||
|
tmpSign := m.Sign(params, secretKey)
|
||||||
|
return tmpSign == signature
|
||||||
|
}
|
||||||
|
|
||||||
|
// buildSortString 构造排序字符串
|
||||||
|
func (m *MiPay) buildSortString(params map[string]string) string {
|
||||||
|
if len(params) == 0 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
// 按键排序
|
||||||
|
keys := make([]string, 0, len(params))
|
||||||
|
for k := range params {
|
||||||
|
keys = append(keys, k)
|
||||||
|
}
|
||||||
|
sort.Strings(keys)
|
||||||
|
|
||||||
|
// 构建排序字符串
|
||||||
|
var fields []string
|
||||||
|
for _, k := range keys {
|
||||||
|
fields = append(fields, fmt.Sprintf("%s=%s", k, params[k]))
|
||||||
|
}
|
||||||
|
|
||||||
|
return strings.Join(fields, "&")
|
||||||
|
}
|
||||||
20
package/pay/xiaomi/model.go
Normal file
20
package/pay/xiaomi/model.go
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
package xiaomi
|
||||||
|
|
||||||
|
import "github.com/gogf/gf/v2/os/gtime"
|
||||||
|
|
||||||
|
type PayCallback struct {
|
||||||
|
AppID string `json:"appId" dc:"游戏ID" required:"true"`
|
||||||
|
CPOrderID string `json:"cpOrderId" dc:"开发商订单ID" required:"true"`
|
||||||
|
CPUserInfo string `json:"cpUserInfo" dc:"开发商透传信息" required:"false"`
|
||||||
|
OrderConsumeType int `json:"orderConsumeType" dc:"订单类型:10:普通订单 11:直充直消订单" required:"false"`
|
||||||
|
OrderID string `json:"orderId" dc:"游戏平台订单ID" required:"true"`
|
||||||
|
OrderStatus string `json:"orderStatus" dc:"订单状态,TRADE_SUCCESS代表成功" required:"true"`
|
||||||
|
PayFee int `json:"payFee" dc:"支付金额,单位为分,即0.01米币。(请务必使用payFee字段值与游戏发起订单金额做校验,确保订单金额一致性)" required:"true"`
|
||||||
|
PayTime *gtime.Time `json:"payTime" dc:"支付时间,格式yyyy-MM-dd HH:mm:ss" required:"true"`
|
||||||
|
ProductCode string `json:"productCode" dc:"商品代码" required:"true"`
|
||||||
|
ProductCount int `json:"productCount" dc:"商品数量" required:"true"`
|
||||||
|
ProductName string `json:"productName" dc:"商品名称" required:"true"`
|
||||||
|
UID string `json:"uid" dc:"用户ID" required:"true"`
|
||||||
|
PartnerGiftConsume int64 `json:"partnerGiftConsume" dc:"使用游戏券金额(如果订单使用游戏券则有,long型),如果有则参与签名" required:"false"`
|
||||||
|
Signature string `json:"signature" dc:"签名,签名方法见后面说明" required:"true"`
|
||||||
|
}
|
||||||
29
package/pay/xiaomi/xiaomi.go
Normal file
29
package/pay/xiaomi/xiaomi.go
Normal file
@@ -0,0 +1,29 @@
|
|||||||
|
package xiaomi
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
|
"github.com/gogf/gf/v2/os/gctx"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Config 小米支付配置信息
|
||||||
|
type Config struct {
|
||||||
|
AppID string `json:"app_id"` // 应用ID
|
||||||
|
AppSecret string `json:"app_secret"` // 应用密钥
|
||||||
|
//PrivateKey string // 商户私钥(如需证书)
|
||||||
|
//MIAPIURL string // 小米支付API基础地址
|
||||||
|
//IsSandbox bool // 是否沙箱环境
|
||||||
|
}
|
||||||
|
|
||||||
|
// Miipay 小米支付客户端
|
||||||
|
type MiPay struct {
|
||||||
|
config *Config
|
||||||
|
}
|
||||||
|
|
||||||
|
func New() *MiPay {
|
||||||
|
_cfg, _ := g.Cfg().Get(gctx.New(), "pay.xiaomi")
|
||||||
|
var cfg *Config
|
||||||
|
_cfg.Scan(&cfg)
|
||||||
|
return &MiPay{
|
||||||
|
config: cfg,
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -270,7 +270,7 @@ func (s *Mod) GetPath(url string) (filePath string) {
|
|||||||
return url[len(get+bucketName)+1:]
|
return url[len(get+bucketName)+1:]
|
||||||
}
|
}
|
||||||
|
|
||||||
// 复制文件
|
// CopyObject 复制文件
|
||||||
func (s *Mod) CopyObject(bucketName string, dstStr string, srcStr string) (err error) {
|
func (s *Mod) CopyObject(bucketName string, dstStr string, srcStr string) (err error) {
|
||||||
|
|
||||||
// 原始文件
|
// 原始文件
|
||||||
@@ -288,3 +288,15 @@ func (s *Mod) CopyObject(bucketName string, dstStr string, srcStr string) (err e
|
|||||||
_, err = s.client.CopyObject(ctx, dst, src)
|
_, err = s.client.CopyObject(ctx, dst, src)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Rename 重命名文件
|
||||||
|
func (s *Mod) Rename(bucketName string, name string, newName string) (err error) {
|
||||||
|
// 复制文件到新的名称
|
||||||
|
err = s.CopyObject(bucketName, name, newName)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// 删除原始文件
|
||||||
|
err = s.RemoveObject(bucketName, name)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,6 +1,9 @@
|
|||||||
package aycache
|
package aycache
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
|
"math"
|
||||||
|
|
||||||
v1 "github.com/ayflying/utility_go/api/system/v1"
|
v1 "github.com/ayflying/utility_go/api/system/v1"
|
||||||
"github.com/ayflying/utility_go/internal/boot"
|
"github.com/ayflying/utility_go/internal/boot"
|
||||||
"github.com/ayflying/utility_go/pkg/aycache/drive"
|
"github.com/ayflying/utility_go/pkg/aycache/drive"
|
||||||
@@ -9,7 +12,6 @@ import (
|
|||||||
"github.com/gogf/gf/v2/os/gcache"
|
"github.com/gogf/gf/v2/os/gcache"
|
||||||
"github.com/prometheus/client_golang/prometheus"
|
"github.com/prometheus/client_golang/prometheus"
|
||||||
"github.com/prometheus/client_golang/prometheus/promauto"
|
"github.com/prometheus/client_golang/prometheus/promauto"
|
||||||
"math"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// Mod 定义缓存模块结构体,包含一个 gcache.Cache 客户端实例
|
// Mod 定义缓存模块结构体,包含一个 gcache.Cache 客户端实例
|
||||||
@@ -19,6 +21,7 @@ type Mod struct {
|
|||||||
|
|
||||||
// QPSCount 记录缓存的 QPS 计数
|
// QPSCount 记录缓存的 QPS 计数
|
||||||
var QPSCount int
|
var QPSCount int
|
||||||
|
|
||||||
// QPS 是一个 Prometheus 指标,用于记录当前缓存的 QPS 数量
|
// QPS 是一个 Prometheus 指标,用于记录当前缓存的 QPS 数量
|
||||||
var QPS = promauto.NewGauge(
|
var QPS = promauto.NewGauge(
|
||||||
prometheus.GaugeOpts{
|
prometheus.GaugeOpts{
|
||||||
@@ -31,7 +34,7 @@ var QPS = promauto.NewGauge(
|
|||||||
func init() {
|
func init() {
|
||||||
boot.AddFunc(func() {
|
boot.AddFunc(func() {
|
||||||
// 初始化指标,每分钟计算一次平均 QPS 并重置计数器
|
// 初始化指标,每分钟计算一次平均 QPS 并重置计数器
|
||||||
service.SystemCron().AddCron(v1.CronType_MINUTE, func() error {
|
service.SystemCron().AddCronV2(v1.CronType_MINUTE, func(context.Context) error {
|
||||||
QPS.Set(math.Round(float64(QPSCount) / 60))
|
QPS.Set(math.Round(float64(QPSCount) / 60))
|
||||||
QPSCount = 0
|
QPSCount = 0
|
||||||
return nil
|
return nil
|
||||||
@@ -53,8 +56,13 @@ func New(_name ...string) gcache.Adapter {
|
|||||||
// 创建内存缓存适配器
|
// 创建内存缓存适配器
|
||||||
cacheAdapterObj = drive2.NewAdapterMemory()
|
cacheAdapterObj = drive2.NewAdapterMemory()
|
||||||
case "redis":
|
case "redis":
|
||||||
|
//第二个参数为配置名称,默认为default
|
||||||
|
var typ = "default"
|
||||||
|
if len(_name) >= 2 {
|
||||||
|
typ = _name[1]
|
||||||
|
}
|
||||||
// 创建 Redis 缓存适配器
|
// 创建 Redis 缓存适配器
|
||||||
cacheAdapterObj = drive2.NewAdapterRedis()
|
cacheAdapterObj = drive2.NewAdapterRedis(typ)
|
||||||
case "file":
|
case "file":
|
||||||
// 创建文件缓存适配器,指定缓存目录为 "runtime/cache"
|
// 创建文件缓存适配器,指定缓存目录为 "runtime/cache"
|
||||||
cacheAdapterObj = drive2.NewAdapterFile("runtime/cache")
|
cacheAdapterObj = drive2.NewAdapterFile("runtime/cache")
|
||||||
|
|||||||
@@ -7,19 +7,22 @@ import (
|
|||||||
"github.com/gogf/gf/v2/os/gctx"
|
"github.com/gogf/gf/v2/os/gctx"
|
||||||
)
|
)
|
||||||
|
|
||||||
var adapterRedisClient gcache.Adapter
|
var adapterRedisClient = make(map[string]gcache.Adapter)
|
||||||
var adapterRedisCache = gcache.New()
|
var adapterRedisCache = make(map[string]*gcache.Cache)
|
||||||
|
|
||||||
func NewAdapterRedis() gcache.Adapter {
|
func NewAdapterRedis(name string) gcache.Adapter {
|
||||||
|
if adapterRedisClient[name] == nil {
|
||||||
if adapterRedisClient == nil {
|
_cfg, err := g.Cfg().Get(gctx.New(), "redis."+name)
|
||||||
_cfg, _ := g.Cfg().Get(gctx.New(), "redis.default")
|
if err != nil {
|
||||||
|
panic("当前redis配置不存在")
|
||||||
|
}
|
||||||
var cfg *gredis.Config
|
var cfg *gredis.Config
|
||||||
_cfg.Scan(&cfg)
|
_cfg.Scan(&cfg)
|
||||||
redisObj, _ := gredis.New(cfg)
|
redisObj, _ := gredis.New(cfg)
|
||||||
//adapterRedisClient = gcache.NewAdapterRedis(g.Redis("default"))
|
//adapterRedisClient[name] = gcache.NewAdapterRedis(g.Redis(name))
|
||||||
adapterRedisClient = gcache.NewAdapterRedis(redisObj)
|
adapterRedisClient[name] = gcache.NewAdapterRedis(redisObj)
|
||||||
adapterRedisCache.SetAdapter(adapterRedisClient)
|
adapterRedisCache[name] = gcache.New()
|
||||||
|
adapterRedisCache[name].SetAdapter(adapterRedisClient[name])
|
||||||
}
|
}
|
||||||
return adapterRedisCache
|
return adapterRedisCache[name]
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,7 +3,9 @@ package elasticsearch
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
|
||||||
"github.com/elastic/go-elasticsearch/v8"
|
"github.com/elastic/go-elasticsearch/v8"
|
||||||
"github.com/elastic/go-elasticsearch/v8/typedapi/core/bulk"
|
"github.com/elastic/go-elasticsearch/v8/typedapi/core/bulk"
|
||||||
"github.com/elastic/go-elasticsearch/v8/typedapi/core/delete"
|
"github.com/elastic/go-elasticsearch/v8/typedapi/core/delete"
|
||||||
@@ -12,6 +14,7 @@ import (
|
|||||||
"github.com/elastic/go-elasticsearch/v8/typedapi/types"
|
"github.com/elastic/go-elasticsearch/v8/typedapi/types"
|
||||||
"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/gctx"
|
||||||
|
"github.com/gogf/gf/v2/util/gconv"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -23,6 +26,13 @@ type Elastic struct {
|
|||||||
name string
|
name string
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type elkBulk struct {
|
||||||
|
Index struct {
|
||||||
|
Index string `json:"_index"`
|
||||||
|
Id string `json:"_id"`
|
||||||
|
} `json:"index"`
|
||||||
|
}
|
||||||
|
|
||||||
func NewV1(name string) *Elastic {
|
func NewV1(name string) *Elastic {
|
||||||
var cfg elasticsearch.Config
|
var cfg elasticsearch.Config
|
||||||
_cfg := g.Cfg().MustGetWithEnv(gctx.New(), "elasticsearch")
|
_cfg := g.Cfg().MustGetWithEnv(gctx.New(), "elasticsearch")
|
||||||
@@ -52,7 +62,7 @@ func NewV1(name string) *Elastic {
|
|||||||
// fmt.Printf("index:%#v\n", resp.Index)
|
// fmt.Printf("index:%#v\n", resp.Index)
|
||||||
//}
|
//}
|
||||||
|
|
||||||
// Set 索引文档
|
// Set 添加文档索引文档
|
||||||
func (s *Elastic) Set(ctx context.Context, key string, data interface{}) (err error) {
|
func (s *Elastic) Set(ctx context.Context, key string, data interface{}) (err error) {
|
||||||
// 添加文档
|
// 添加文档
|
||||||
_, err = s.client.Index(s.name).Id(key).Document(data).Do(ctx)
|
_, err = s.client.Index(s.name).Id(key).Document(data).Do(ctx)
|
||||||
@@ -61,11 +71,43 @@ func (s *Elastic) Set(ctx context.Context, key string, data interface{}) (err er
|
|||||||
|
|
||||||
// SetBulk 批量添加文档
|
// SetBulk 批量添加文档
|
||||||
func (s *Elastic) SetBulk(ctx context.Context, data []any) (err error) {
|
func (s *Elastic) SetBulk(ctx context.Context, data []any) (err error) {
|
||||||
var save *bulk.Request
|
var save bulk.Request
|
||||||
save = &bulk.Request{
|
save = make(bulk.Request, 0)
|
||||||
data,
|
for _, v := range data {
|
||||||
|
val := gconv.Map(v)
|
||||||
|
var saveIndex = elkBulk{}
|
||||||
|
saveIndex.Index.Index = s.name
|
||||||
|
if _, ok := val["uuid"]; ok {
|
||||||
|
saveIndex.Index.Id = val["uuid"].(string)
|
||||||
|
}
|
||||||
|
save = append(save, saveIndex)
|
||||||
|
save = append(save, v)
|
||||||
|
}
|
||||||
|
//save = data
|
||||||
|
response, err2 := s.client.Bulk().Index(s.name).Request(&save).Do(ctx)
|
||||||
|
if err2 != nil {
|
||||||
|
err = err2
|
||||||
|
return
|
||||||
|
}
|
||||||
|
//需要接收返回信息,判断是否全部执行成功
|
||||||
|
if response.Errors { //未全部完成
|
||||||
|
//是否需要删除已成功导入的部分数据
|
||||||
|
for _, item := range response.Items {
|
||||||
|
for _, v := range item {
|
||||||
|
if v.Error != nil { //失败
|
||||||
|
g.Log().Errorf(ctx, "导入数据出错 err: %v", *v.Error.Reason)
|
||||||
|
|
||||||
|
} else {
|
||||||
|
//删除已导入成功的数据
|
||||||
|
_, err = s.Delete(ctx, *v.Id_)
|
||||||
|
if err != nil {
|
||||||
|
g.Log().Errorf(ctx, "删除数据错误, err:%v\n", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return errors.New("部分数据导入失败")
|
||||||
}
|
}
|
||||||
s.client.Bulk().Index(s.name).Request(save).Do(ctx)
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -97,7 +139,7 @@ func (s *Elastic) Delete(ctx context.Context, key string) (res *delete.Response,
|
|||||||
// Select 查询
|
// Select 查询
|
||||||
func (s *Elastic) Select(ctx context.Context, query *types.MatchAllQuery) (res *search.Response, err error) {
|
func (s *Elastic) Select(ctx context.Context, query *types.MatchAllQuery) (res *search.Response, err error) {
|
||||||
res, err = s.client.Search(). //Index("my_index").
|
res, err = s.client.Search(). //Index("my_index").
|
||||||
Request(&search.Request{
|
Request(&search.Request{
|
||||||
Query: &types.Query{
|
Query: &types.Query{
|
||||||
MatchAll: &types.MatchAllQuery{},
|
MatchAll: &types.MatchAllQuery{},
|
||||||
},
|
},
|
||||||
|
|||||||
43
pkg/s3/s3.go
43
pkg/s3/s3.go
@@ -150,6 +150,7 @@ func (s *Mod) PutObject(f io.Reader, name string, bucketName string, _size ...in
|
|||||||
}
|
}
|
||||||
|
|
||||||
// RemoveObject 从指定存储桶中删除指定名称的文件
|
// RemoveObject 从指定存储桶中删除指定名称的文件
|
||||||
|
// Deprecation: to新方法 RemoveObjectV2
|
||||||
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,
|
ForceDelete: true,
|
||||||
@@ -161,6 +162,18 @@ func (s *Mod) RemoveObject(name string, bucketName string) (err error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// RemoveObjectV2 从指定存储桶中删除指定名称的文件
|
||||||
|
func (s *Mod) RemoveObjectV2(bucketName string, name string) (err error) {
|
||||||
|
opts := minio.RemoveObjectOptions{
|
||||||
|
ForceDelete: true,
|
||||||
|
//GovernanceBypass: true,
|
||||||
|
//VersionID: "myversionid",
|
||||||
|
}
|
||||||
|
// 调用 S3 客户端删除文件
|
||||||
|
err = s.client.RemoveObject(ctx, bucketName, name, opts)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
// ListObjects 获取指定存储桶中指定前缀的文件列表
|
// ListObjects 获取指定存储桶中指定前缀的文件列表
|
||||||
// 返回一个包含文件信息的通道
|
// 返回一个包含文件信息的通道
|
||||||
func (s *Mod) ListObjects(bucketName string, prefix string) (res <-chan minio.ObjectInfo, err error) {
|
func (s *Mod) ListObjects(bucketName string, prefix string) (res <-chan minio.ObjectInfo, err error) {
|
||||||
@@ -216,21 +229,43 @@ func (s *Mod) GetPath(url string) (filePath string) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// CopyObject 在指定存储桶内复制文件
|
// CopyObject 在指定存储桶内复制文件
|
||||||
// 支持指定源文件和目标文件路径
|
// bucketName 存储桶名称
|
||||||
|
// dstStr 目标文件路径
|
||||||
|
// srcStr 源文件路径
|
||||||
|
// 返回操作过程中可能出现的错误
|
||||||
func (s *Mod) CopyObject(bucketName string, dstStr string, srcStr string) (err error) {
|
func (s *Mod) CopyObject(bucketName string, dstStr string, srcStr string) (err error) {
|
||||||
// 定义目标文件选项
|
// 定义目标文件的复制选项,包含存储桶名称和目标文件路径
|
||||||
var dst = minio.CopyDestOptions{
|
var dst = minio.CopyDestOptions{
|
||||||
Bucket: bucketName,
|
Bucket: bucketName,
|
||||||
Object: dstStr,
|
Object: dstStr,
|
||||||
}
|
}
|
||||||
|
|
||||||
// 定义源文件选项
|
// 定义源文件的复制选项,包含存储桶名称和源文件路径
|
||||||
var src = minio.CopySrcOptions{
|
var src = minio.CopySrcOptions{
|
||||||
Bucket: bucketName,
|
Bucket: bucketName,
|
||||||
Object: srcStr,
|
Object: srcStr,
|
||||||
}
|
}
|
||||||
|
|
||||||
// 调用 S3 客户端复制文件
|
// 调用 S3 客户端的 CopyObject 方法,将源文件复制到目标位置
|
||||||
|
// 忽略返回的复制信息,仅关注是否发生错误
|
||||||
_, err = s.client.CopyObject(ctx, dst, src)
|
_, err = s.client.CopyObject(ctx, dst, src)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Rename 重命名文件
|
||||||
|
func (s *Mod) Rename(bucketName string, oldName string, newName string) (err error) {
|
||||||
|
// 复制文件到新的名称
|
||||||
|
g.Log().Debugf(nil, "仓库=%v,rename %s to %s", bucketName, oldName, newName)
|
||||||
|
err = s.CopyObject(bucketName, newName, oldName)
|
||||||
|
if err != nil {
|
||||||
|
g.Log().Error(ctx, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// 删除原始文件
|
||||||
|
err = s.RemoveObjectV2(bucketName, oldName)
|
||||||
|
if err != nil {
|
||||||
|
g.Log().Error(ctx, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|||||||
@@ -7,9 +7,6 @@ package service
|
|||||||
|
|
||||||
type (
|
type (
|
||||||
IIp2Region interface {
|
IIp2Region interface {
|
||||||
// Load 加载到内存中
|
|
||||||
//
|
|
||||||
// @Description: 加载ip2region数据库到内存中。
|
|
||||||
// @receiver s *sIp2region: sIp2region的实例。
|
// @receiver s *sIp2region: sIp2region的实例。
|
||||||
Load()
|
Load()
|
||||||
GetIp(ip string) (res []string)
|
GetIp(ip string) (res []string)
|
||||||
|
|||||||
@@ -38,7 +38,8 @@ type (
|
|||||||
// @receiver s: sSystemCron的实例,代表一个调度系统。
|
// @receiver s: sSystemCron的实例,代表一个调度系统。
|
||||||
// @param typ: 任务的类型,决定该任务将被添加到哪个列表中。对应不同的时间间隔。
|
// @param typ: 任务的类型,决定该任务将被添加到哪个列表中。对应不同的时间间隔。
|
||||||
// @param _func: 要添加的任务函数,该函数执行时应该返回一个error。
|
// @param _func: 要添加的任务函数,该函数执行时应该返回一个error。
|
||||||
AddCronV2(typ v1.CronType, _func func(context.Context) error)
|
// @param unique: 是否只在唯一服务器上执行
|
||||||
|
AddCronV2(typ v1.CronType, _func func(context.Context) error, unique ...bool)
|
||||||
// StartCron 开始计划任务执行
|
// StartCron 开始计划任务执行
|
||||||
//
|
//
|
||||||
// @Description:
|
// @Description:
|
||||||
|
|||||||
@@ -1,11 +1,15 @@
|
|||||||
package utility_go
|
package utility_go
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/ayflying/utility_go/config"
|
"github.com/ayflying/utility_go/config"
|
||||||
"github.com/ayflying/utility_go/internal/boot"
|
"github.com/ayflying/utility_go/internal/boot"
|
||||||
_ "github.com/ayflying/utility_go/internal/logic"
|
_ "github.com/ayflying/utility_go/internal/logic"
|
||||||
"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/gctx"
|
||||||
|
"github.com/gogf/gf/v2/os/gtimer"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -14,9 +18,12 @@ var (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
|
var err error
|
||||||
g.Log().Debug(ctx, "utility_go init启动完成")
|
g.Log().Debug(ctx, "utility_go init启动完成")
|
||||||
// 初始化配置
|
// 初始化配置
|
||||||
var err = boot.Boot()
|
gtimer.SetTimeout(ctx, time.Second*5, func(ctx context.Context) {
|
||||||
|
err = boot.Boot()
|
||||||
|
})
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
|
|||||||
Reference in New Issue
Block a user