MiFaZhan
58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
// package queues
|
|
// @Link https://github.com/xiujiecn/xiu-admin
|
|
// @Copyright Copyright (c) 2025 LiXiujie
|
|
// @Author Lxj <xxx@example.com>
|
|
// @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
|
|
package queues
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/worker"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
)
|
|
|
|
var (
|
|
processes = make(map[string]*worker.Process)
|
|
queues = make(map[string]*worker.Queue)
|
|
)
|
|
|
|
func RegisterQueueProcess(name string, process worker.Process) {
|
|
processes[name] = &process
|
|
}
|
|
|
|
func GetQueue(name string) *worker.Queue {
|
|
if queue, ok := queues[name]; ok {
|
|
return queue
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func Run(ctx context.Context) {
|
|
for name, process := range processes {
|
|
queues[name] = worker.RegisterQueueProcess(*process)
|
|
g.Log().Infof(ctx, "queues[%s] register success", name)
|
|
}
|
|
}
|
|
|
|
func Push(ctx context.Context, topic string, data []byte, timeout int) (err error) {
|
|
queue := GetQueue(topic)
|
|
if queue != nil {
|
|
err = queue.Push(context.Background(), topic, data, timeout)
|
|
} else {
|
|
g.Log().Errorf(ctx, "server/internal/queues/base.go Push queue[%s] not found", topic)
|
|
return errors.New("queue not found")
|
|
}
|
|
return
|
|
}
|
|
|
|
func Stop(ctx context.Context) {
|
|
for name, _ := range processes {
|
|
queue := GetQueue(name)
|
|
if queue != nil {
|
|
queue.Stop()
|
|
}
|
|
}
|
|
}
|