Compare commits
12 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 14cf759ce1 | |||
| 19a19c1ff1 | |||
| 183d6d8b10 | |||
| d0cad61028 | |||
| efb34e0c5b | |||
|
|
8190e9f6b7 | ||
|
|
27435b57b7 | ||
|
|
0628882533 | ||
|
|
f68655eee6 | ||
|
|
d8491f0aba | ||
|
|
999f1f6a84 | ||
|
|
039acea0af |
@@ -3,6 +3,10 @@ package gameAct
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/ayflying/utility_go/internal/model/do"
|
||||
"github.com/ayflying/utility_go/internal/model/entity"
|
||||
"github.com/ayflying/utility_go/pkg"
|
||||
@@ -13,9 +17,6 @@ import (
|
||||
"github.com/gogf/gf/v2/os/gctx"
|
||||
"github.com/gogf/gf/v2/os/gtime"
|
||||
"github.com/gogf/gf/v2/util/gconv"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
var (
|
||||
@@ -206,37 +207,29 @@ func (s *sGameAct) Save(ctx context.Context, actId int) (err error) {
|
||||
g.Log().Error(ctx, err2)
|
||||
return
|
||||
}
|
||||
////获取多少个数据,删除不是当前修改的数据
|
||||
//count, _ := g.Model(Name).Where(do.GameAct{
|
||||
// Uid: v.Uid,
|
||||
// ActId: v.ActId,
|
||||
//}).Count()
|
||||
//if count > 1 {
|
||||
// g.Model(Name).Where(do.GameAct{
|
||||
// Uid: v.Uid,
|
||||
// ActId: v.ActId,
|
||||
// }).WhereNot("updated_at", v.UpdatedAt).Delete()
|
||||
//}
|
||||
}
|
||||
//dbRes, err2 := g.Model(Name).Batch(50).Data(add).Update()
|
||||
update = make([]*entity.GameAct, 0)
|
||||
dbRes, err2 := g.Model(Name).Batch(50).Data(add).Save()
|
||||
add = make([]*entity.GameAct, 0)
|
||||
if err2 != nil {
|
||||
g.Log().Error(ctx, err2)
|
||||
return
|
||||
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 {
|
||||
_, err2 = g.Redis().Del(ctx, v)
|
||||
if err2 != nil {
|
||||
g.Log().Error(ctx, err2)
|
||||
return
|
||||
_, err = g.Redis().Del(ctx, v)
|
||||
if err != nil {
|
||||
g.Log().Error(ctx, err)
|
||||
}
|
||||
}
|
||||
delKey = make([]string, 0)
|
||||
|
||||
count, _ := dbRes.RowsAffected()
|
||||
g.Log().Debugf(ctx, "当前 %v 写入数据库: %v 条", actId, count)
|
||||
}
|
||||
|
||||
|
||||
@@ -2,16 +2,17 @@ package gameKv
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/ayflying/utility_go/pkg"
|
||||
"github.com/ayflying/utility_go/service"
|
||||
"github.com/ayflying/utility_go/tools"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/os/gctx"
|
||||
"github.com/gogf/gf/v2/os/gtime"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
var (
|
||||
@@ -107,8 +108,7 @@ func (s *sGameKv) SavesV1() (err error) {
|
||||
for _, v := range delKey {
|
||||
_, err2 = g.Redis().Del(ctx, v)
|
||||
if err2 != nil {
|
||||
g.Log().Errorf(ctx, "删除存档错误:%v,err=%v", v, err2)
|
||||
return
|
||||
g.Log().Errorf(ctx, "删除存档失败:%v,err=%v", v, err2)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -1,13 +1,14 @@
|
||||
package ip2region
|
||||
|
||||
import (
|
||||
"net"
|
||||
"strings"
|
||||
|
||||
"github.com/ayflying/utility_go/service"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/os/gctx"
|
||||
"github.com/gogf/gf/v2/os/gfile"
|
||||
"github.com/lionsoul2014/ip2region/binding/golang/xdb"
|
||||
"net"
|
||||
"strings"
|
||||
)
|
||||
|
||||
var (
|
||||
@@ -81,7 +82,7 @@ func (s *sIp2region) Load() {
|
||||
|
||||
func (s *sIp2region) GetIp(ip string) (res []string) {
|
||||
//初始化加载
|
||||
if s.searcher != nil {
|
||||
if s.searcher == nil {
|
||||
s.Load()
|
||||
}
|
||||
|
||||
|
||||
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()
|
||||
})
|
||||
}
|
||||
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
|
||||
}
|
||||
@@ -1,7 +1,9 @@
|
||||
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"
|
||||
@@ -10,7 +12,8 @@ import (
|
||||
"strings"
|
||||
)
|
||||
|
||||
func (p *Pay) VerifySign(bm g.Map, key string) bool {
|
||||
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")
|
||||
|
||||
@@ -1,6 +1,8 @@
|
||||
package aycache
|
||||
|
||||
import (
|
||||
"math"
|
||||
|
||||
v1 "github.com/ayflying/utility_go/api/system/v1"
|
||||
"github.com/ayflying/utility_go/internal/boot"
|
||||
"github.com/ayflying/utility_go/pkg/aycache/drive"
|
||||
@@ -9,7 +11,6 @@ import (
|
||||
"github.com/gogf/gf/v2/os/gcache"
|
||||
"github.com/prometheus/client_golang/prometheus"
|
||||
"github.com/prometheus/client_golang/prometheus/promauto"
|
||||
"math"
|
||||
)
|
||||
|
||||
// Mod 定义缓存模块结构体,包含一个 gcache.Cache 客户端实例
|
||||
@@ -19,6 +20,7 @@ type Mod struct {
|
||||
|
||||
// QPSCount 记录缓存的 QPS 计数
|
||||
var QPSCount int
|
||||
|
||||
// QPS 是一个 Prometheus 指标,用于记录当前缓存的 QPS 数量
|
||||
var QPS = promauto.NewGauge(
|
||||
prometheus.GaugeOpts{
|
||||
@@ -53,8 +55,13 @@ func New(_name ...string) gcache.Adapter {
|
||||
// 创建内存缓存适配器
|
||||
cacheAdapterObj = drive2.NewAdapterMemory()
|
||||
case "redis":
|
||||
//第二个参数为配置名称,默认为default
|
||||
var typ = "default"
|
||||
if len(_name) >= 2 {
|
||||
typ = _name[1]
|
||||
}
|
||||
// 创建 Redis 缓存适配器
|
||||
cacheAdapterObj = drive2.NewAdapterRedis()
|
||||
cacheAdapterObj = drive2.NewAdapterRedis(typ)
|
||||
case "file":
|
||||
// 创建文件缓存适配器,指定缓存目录为 "runtime/cache"
|
||||
cacheAdapterObj = drive2.NewAdapterFile("runtime/cache")
|
||||
|
||||
@@ -7,19 +7,22 @@ import (
|
||||
"github.com/gogf/gf/v2/os/gctx"
|
||||
)
|
||||
|
||||
var adapterRedisClient gcache.Adapter
|
||||
var adapterRedisCache = gcache.New()
|
||||
var adapterRedisClient = make(map[string]gcache.Adapter)
|
||||
var adapterRedisCache = make(map[string]*gcache.Cache)
|
||||
|
||||
func NewAdapterRedis() gcache.Adapter {
|
||||
|
||||
if adapterRedisClient == nil {
|
||||
_cfg, _ := g.Cfg().Get(gctx.New(), "redis.default")
|
||||
func NewAdapterRedis(name string) gcache.Adapter {
|
||||
if adapterRedisClient[name] == nil {
|
||||
_cfg, err := g.Cfg().Get(gctx.New(), "redis."+name)
|
||||
if err != nil {
|
||||
panic("当前redis配置不存在")
|
||||
}
|
||||
var cfg *gredis.Config
|
||||
_cfg.Scan(&cfg)
|
||||
redisObj, _ := gredis.New(cfg)
|
||||
//adapterRedisClient = gcache.NewAdapterRedis(g.Redis("default"))
|
||||
adapterRedisClient = gcache.NewAdapterRedis(redisObj)
|
||||
adapterRedisCache.SetAdapter(adapterRedisClient)
|
||||
//adapterRedisClient[name] = gcache.NewAdapterRedis(g.Redis(name))
|
||||
adapterRedisClient[name] = gcache.NewAdapterRedis(redisObj)
|
||||
adapterRedisCache[name] = gcache.New()
|
||||
adapterRedisCache[name].SetAdapter(adapterRedisClient[name])
|
||||
}
|
||||
return adapterRedisCache
|
||||
return adapterRedisCache[name]
|
||||
}
|
||||
|
||||
@@ -3,7 +3,9 @@ package elasticsearch
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
|
||||
"github.com/elastic/go-elasticsearch/v8"
|
||||
"github.com/elastic/go-elasticsearch/v8/typedapi/core/bulk"
|
||||
"github.com/elastic/go-elasticsearch/v8/typedapi/core/delete"
|
||||
@@ -82,7 +84,30 @@ func (s *Elastic) SetBulk(ctx context.Context, data []any) (err error) {
|
||||
save = append(save, v)
|
||||
}
|
||||
//save = data
|
||||
_, err = s.client.Bulk().Index(s.name).Request(&save).Do(ctx)
|
||||
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("部分数据导入失败")
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
@@ -114,7 +139,7 @@ func (s *Elastic) Delete(ctx context.Context, key string) (res *delete.Response,
|
||||
// Select 查询
|
||||
func (s *Elastic) Select(ctx context.Context, query *types.MatchAllQuery) (res *search.Response, err error) {
|
||||
res, err = s.client.Search(). //Index("my_index").
|
||||
Request(&search.Request{
|
||||
Request(&search.Request{
|
||||
Query: &types.Query{
|
||||
MatchAll: &types.MatchAllQuery{},
|
||||
},
|
||||
|
||||
Reference in New Issue
Block a user