feat:完善项目
This commit is contained in:
@ -7,11 +7,13 @@ import (
|
||||
"ArmedPolice/utils"
|
||||
"fmt"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
type TaskListen struct {
|
||||
Produce chan *Task
|
||||
Consume chan *Task
|
||||
ticker *time.Ticker
|
||||
lock *sync.Mutex
|
||||
}
|
||||
|
||||
@ -35,12 +37,29 @@ func (this *TaskListen) Listen() {
|
||||
if err != nil {
|
||||
fmt.Printf("Task Produce Redis Sadd Error【%s】", err)
|
||||
} else {
|
||||
err = cache.Cache.HSet(config.RedisKeyForTaskQueueBody, p.ID, p)
|
||||
fmt.Printf("err:%v\n", err)
|
||||
_ = cache.Cache.HSet(config.RedisKeyForTaskQueueBody, p.ID, p)
|
||||
}
|
||||
case c := <-this.Consume:
|
||||
if err := c.Handle(); err != nil {
|
||||
fmt.Printf("err:%v\n", err)
|
||||
// 监听到数据,执行后续操作
|
||||
_ = c.Handle()
|
||||
case <-this.ticker.C:
|
||||
now := time.Now()
|
||||
|
||||
_cache, _ := cache.Cache.ZRangebyscore(config.RedisKeyForTaskQueue, &logic.ScoreRangeBy{Min: "0",
|
||||
Max: fmt.Sprintf("%d", now.Unix())})
|
||||
|
||||
if len(_cache) > 0 {
|
||||
for _, v := range _cache {
|
||||
body, _ := cache.Cache.HGet(config.RedisKeyForTaskQueueBody, v)
|
||||
// TODO:有Bug,读取接口数据为空
|
||||
task := new(Task)
|
||||
task.Body = new(Order)
|
||||
_ = task.UnmarshalBinary([]byte(body))
|
||||
task.Consume()
|
||||
}
|
||||
// 销毁信息
|
||||
_ = cache.Cache.ZRem(config.RedisKeyForTaskQueue, _cache)
|
||||
_ = cache.Cache.HDel(config.RedisKeyForTaskQueueBody, _cache...)
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -48,12 +67,11 @@ func (this *TaskListen) Listen() {
|
||||
}
|
||||
|
||||
func NewTaskListen() *TaskListen {
|
||||
if TaskListenEvent == nil {
|
||||
TaskListenEvent = &TaskListen{
|
||||
Produce: make(chan *Task, 1),
|
||||
Consume: make(chan *Task, 1),
|
||||
lock: new(sync.Mutex),
|
||||
}
|
||||
TaskListenEvent = &TaskListen{
|
||||
Produce: make(chan *Task, 1),
|
||||
Consume: make(chan *Task, 1),
|
||||
ticker: time.NewTicker(time.Second * 1),
|
||||
lock: new(sync.Mutex),
|
||||
}
|
||||
return TaskListenEvent
|
||||
}
|
||||
|
Reference in New Issue
Block a user