chore: 并入 iotlcm-app 历史到 app/（原提交 f40bda9）

2026-11-02 13:28:00 +08:00
parent 03ef151f5b f40bda99a8
commit 226495365e
1107 changed files with 147143 additions and 0 deletions
@@ -0,0 +1,84 @@
+// package worker
+// @Link  https://github.com/xiujiecn/xiu-admin
+// @Copyright  Copyright (c) 2025 LiXiujie
+// @Author  Lxj <li@example.com>
+// @License  https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
+package worker
+
+import (
+	"context"
+
+	"git.5iot.com/iotlcm-public/iotlcm-app/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
+}