63 lines
1.3 KiB
Go
63 lines
1.3 KiB
Go
|
package task
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"Edu/utils"
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type ITask interface {
|
|||
|
|
MarshalBinary() ([]byte, error)
|
|||
|
|
UnmarshalBinary(data []byte) error
|
|||
|
|
Handle() error
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Task struct {
|
|||
|
|
ID string `json:"id"` // Job唯一标识
|
|||
|
|
Topic string `json:"topic"` // Job类型
|
|||
|
|
Delay int64 `json:"delay"` // Job需要延迟的时间, 单位：秒
|
|||
|
|
Body ITask `json:"body"` // Job的内容，供消费者做具体的业务处理
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type TaskHandle func(topic string, delay int64, body ITask) *Task
|
|||
|
|
|
|||
|
|
func (this *Task) MarshalBinary() ([]byte, error) {
|
|||
|
|
return json.Marshal(this)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *Task) UnmarshalBinary(data []byte) error {
|
|||
|
|
return utils.FromJSONBytes(data, this)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *Task) setID() string {
|
|||
|
|
return utils.Sha1String(fmt.Sprintf("%s%d%s", this.Topic, time.Now().UnixNano(), utils.GetRandomCode(6)))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *Task) DelayUnix() int64 {
|
|||
|
|
return time.Now().Unix() + this.Delay
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *Task) Push() {
|
|||
|
|
TaskListenEvent.Join(this)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Consume 消费
|
|||
|
|
func (this *Task) Consume() {
|
|||
|
|
TaskListenEvent.Quit(this)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *Task) Handle() error {
|
|||
|
|
// 处理各种方法
|
|||
|
|
this.Body.(*Order).Handle()
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewTask() TaskHandle {
|
|||
|
|
return func(topic string, delay int64, body ITask) *Task {
|
|||
|
|
task := &Task{Topic: topic, Delay: delay, Body: body}
|
|||
|
|
task.ID = task.setID()
|
|||
|
|
return task
|
|||
|
|
}
|
|||
|
|
}
|