MiFaZhan
59eb34976a
iotlcm-admin-svr：管理后台服务。 - Go module：iotlcm → git.5iot.com/iotlcm-public/iotlcm-admin-svr（729 文件） - 凭据与地址脱敏：JWT signingKey（含代码里的 MustGet 默认值）、 MySQL DSN、Redis 口令、告警邮件发件人邮箱全部改为占位； 私网 IP → 文档地址段；库名 xiujie_admin → iotlcm - 真实手机号/邮箱 → 合法 11 位编造号 / example.com - 移除内部资产：CI 触发器、过时 DDL（sql/db_iotlcm.sql）、 私有依赖 go get 行、明文 registry 登录口令 - 启动 banner 统一 IOTLCM 方块字 + 社区惯例署名 - Dockerfile 改用公共基础镜像 - 拼错的 RADEME.md → README.md，README 重写 - 需要真实数据库的用例改为可用 IOTLCM_TEST_WITH_DB=1 显式启用 - 补 Apache-2.0 LICENSE，LF 行尾 + .gitattributes 数据库 schema 不随本仓发布（生产 DDL 已过时），README 已说明。
85 lines
2.0 KiB
Go
85 lines
2.0 KiB
Go
// package worker
|
|
// @Link https://github.com/xiujiecn/xiu-admin
|
|
// @Copyright Copyright (c) 2025 LiXiujie
|
|
// @Author Lxj <li@xiujie.cn>
|
|
// @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
|
|
package worker
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/utility/uuid32"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
)
|
|
|
|
type Queue struct {
|
|
topic string
|
|
w *Worker
|
|
}
|
|
|
|
type Process interface {
|
|
GetTopic() string // 获取消费主题
|
|
Handle(ctx context.Context, p Payload) (err error) // 处理过程的方法
|
|
GetAggregator() *AggregatorOptions // 获取聚合器配置，返回nil不使用聚合
|
|
}
|
|
|
|
func RegisterQueueProcess(p Process) (q *Queue) {
|
|
q = &Queue{
|
|
topic: p.GetTopic(),
|
|
w: New(WithWorkerGroup(p.GetTopic()), WithWorkerHandler(p.Handle), WithWorkerAggregator(p.GetAggregator())),
|
|
}
|
|
return
|
|
}
|
|
|
|
func (q *Queue) Push(ctx context.Context, topic string, data []byte, timeout int) (err error) {
|
|
err = q.w.Once(
|
|
WithTaskUid(uuid32.New()),
|
|
WithTaskGroup(topic),
|
|
WithTaskPayload(data),
|
|
WithTaskTimeout(timeout),
|
|
)
|
|
if err != nil {
|
|
g.Log().Debug(ctx, "Run Queue TaskWorker %s Error: %v", topic, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (q *Queue) Stop() {
|
|
if q.w != nil {
|
|
q.w.Stop()
|
|
}
|
|
}
|
|
|
|
type Scheduled struct {
|
|
topic string
|
|
w *Worker
|
|
}
|
|
|
|
func RegisterScheduledProcess(p Process) (q *Queue) {
|
|
q = &Queue{
|
|
topic: p.GetTopic(),
|
|
w: New(WithWorkerGroup(p.GetTopic()), WithWorkerHandler(p.Handle), WithWorkerAggregator(p.GetAggregator())),
|
|
}
|
|
return
|
|
}
|
|
func (s *Scheduled) Cron(ctx context.Context, topic, cronExpr string, data []byte) (entryID string, err error) {
|
|
s.topic = topic
|
|
entryID, err = s.w.Cron(
|
|
1,
|
|
WithTaskUid(uuid32.S()),
|
|
WithTaskGroup(topic),
|
|
WithTaskExpr(cronExpr),
|
|
WithTaskPayload(data),
|
|
)
|
|
if err != nil {
|
|
g.Log().Debug(ctx, "Run Cron TaskWorker %s Error: %v", topic, err)
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *Scheduled) Remove(entryID string) (err error) {
|
|
err = s.w.RemoveCron(entryID)
|
|
return
|
|
}
|