chore: 初始开源发布
iotlcm-app:小程序端 + 用户端服务。 - app/:uni-app 小程序前端(基于 unibest,MIT) - 微信 AppID → YOUR_WX_APPID,uni-app AppID → __UNI__YOURAPPID - 内网后端地址 → admin.iotlcm.example.com - 应用标题与 unibest 模板 README 重写(保留上游致谢与 MIT 声明) - server/:GoFrame 用户端服务(基于 XiuAdmin,Apache-2.0) - module xiuadmin → git.5iot.com/iotlcm-public/iotlcm-app(360 文件) - 支付回调地址 → app.iotlcm.example.com,商户名与应用名泛化 - 真实手机号 → 合法 11 位编造号(保留号段前 3 位 + 后 4 位) - 依赖号码尾部的断言同步更新 - 补 server/LICENSE(Apache-2.0) - 移除内部资产:CI 触发器 - 补根 README,LF 行尾 + .gitattributes 已知:TestSysDept_RefreshDeptAncestors 在源仓同一提交上同样失败,属既有缺陷。
This commit is contained in:
@@ -0,0 +1,127 @@
|
||||
package queue
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/gogf/gf/v2/container/gqueue"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/net/gtrace"
|
||||
"github.com/gogf/gf/v2/os/gctx"
|
||||
"github.com/gogf/gf/v2/os/grpool"
|
||||
)
|
||||
|
||||
type tMemoryQueue struct {
|
||||
q *gqueue.Queue
|
||||
p Process
|
||||
grpool *grpool.Pool
|
||||
}
|
||||
|
||||
var (
|
||||
memRwLock sync.RWMutex
|
||||
memQueues = make(map[string]*tMemoryQueue)
|
||||
)
|
||||
|
||||
func RegisterMemoryQueue(ctx context.Context, topic string, p Process, config *Config) *tMemoryQueue {
|
||||
memRwLock.Lock()
|
||||
defer memRwLock.Unlock()
|
||||
q, ok := memQueues[topic]
|
||||
if !ok {
|
||||
q = &tMemoryQueue{
|
||||
q: gqueue.New(),
|
||||
p: p,
|
||||
}
|
||||
memQueues[topic] = q
|
||||
}
|
||||
workers := 1
|
||||
if config != nil && config.Workers > 0 {
|
||||
workers = config.Workers
|
||||
}
|
||||
if p != nil {
|
||||
q.p = p
|
||||
q.grpool = grpool.New(workers)
|
||||
go q.Handle(ctx)
|
||||
}
|
||||
g.Log().Infof(ctx, "RegisterMemoryQueue %s success, workers: %d, process: %v", topic, workers, p)
|
||||
return q
|
||||
}
|
||||
|
||||
func UnregisterMemoryQueue(ctx context.Context, topic string) {
|
||||
memRwLock.Lock()
|
||||
defer memRwLock.Unlock()
|
||||
q, ok := memQueues[topic]
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
q.q.Close()
|
||||
delete(memQueues, topic)
|
||||
}
|
||||
|
||||
func (q *tMemoryQueue) GetType() QueueType {
|
||||
return QueueTypeMemory
|
||||
}
|
||||
|
||||
func (q *tMemoryQueue) GetTopic() string {
|
||||
return q.p.GetTopic()
|
||||
}
|
||||
|
||||
func (q *tMemoryQueue) Push(ctx context.Context, topic string, data interface{}, expire int64) (err error) {
|
||||
jsonData, err := json.Marshal(data)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
payload := Payload{
|
||||
Ts: time.Now().UnixMilli(),
|
||||
TraceId: gctx.CtxId(ctx),
|
||||
Topic: topic,
|
||||
Data: json.RawMessage(jsonData),
|
||||
Expire: expire,
|
||||
}
|
||||
q.q.Push(payload)
|
||||
return
|
||||
}
|
||||
|
||||
func (q *tMemoryQueue) Handle(ctx context.Context) (err error) {
|
||||
if q.grpool == nil {
|
||||
g.Log().Errorf(ctx, "grpool is nil")
|
||||
return errors.New("grpool is nil")
|
||||
}
|
||||
p := q.p
|
||||
if p == nil {
|
||||
g.Log().Errorf(ctx, "process is nil")
|
||||
return errors.New("process is nil")
|
||||
}
|
||||
|
||||
for {
|
||||
|
||||
data := q.q.Pop()
|
||||
if data == nil {
|
||||
break
|
||||
}
|
||||
payload, ok := data.(Payload)
|
||||
if !ok {
|
||||
g.Log().Errorf(ctx, "memqueue process unmarshal error: %v", data)
|
||||
continue
|
||||
}
|
||||
subCtx := gctx.New()
|
||||
gtrace.WithTraceID(subCtx, payload.TraceId)
|
||||
q.grpool.Add(ctx, func(ctx context.Context) {
|
||||
defer func() {
|
||||
if r := recover(); r != nil {
|
||||
g.Log().Errorf(ctx, "memqueue process panic: %v, topic: %s, trace_id: %s, hex(data): %v",
|
||||
r, q.p.GetTopic(), payload.TraceId, hex.EncodeToString(payload.Data))
|
||||
}
|
||||
}()
|
||||
err := p.Handle(subCtx, payload)
|
||||
if err != nil {
|
||||
g.Log().Errorf(ctx, "memqueue process error: %v, topic: %s, trace_id: %s, hex(data): %v",
|
||||
err, q.p.GetTopic(), payload.TraceId, hex.EncodeToString(payload.Data))
|
||||
}
|
||||
})
|
||||
}
|
||||
return
|
||||
}
|
||||
Reference in New Issue
Block a user