rmq

type BatchError struct {
	// contains filtered or unexported fields
}

type Broker interface {
	Encode(*Message) ([]byte, error)
	Decode(bytes []byte) (msg *Message, err error)

	Push(context.Context, ...*Message) error
	Pop(ctx context.Context) (msg *Message, err error)
}

type BrokerAfterExit interface {
	AfterExit(ctx context.Context) error
}

type BrokerAfterStart interface {
	AfterStart(ctx context.Context) error
}

type BrokerBeforeExit interface {
	BeforeExit(ctx context.Context) error
}

type BrokerBeforeStart interface {
	BeforeStart(ctx context.Context) error
}

type Callback func(ctx context.Context, msg *Message) (string, error)

type CommandTask struct {
	Shell   string
	Command []string
}

type Hook struct {
	// contains filtered or unexported fields
}

type HttpTask struct {
	Url    string            `json:"url"`
	Method string            `json:"method"`
	Header map[string]string `json:"header,omitempty"`
	Body   json.RawMessage   `json:"body,omitempty"` // 因为json序列号，保证消息好看一点
	// contains filtered or unexported fields
}

type Logger interface {
	Errorf(format string, arg ...interface{})
	Warningf(format string, args ...interface{})
	Infof(format string, arg ...interface{})
}

type Message struct {
	Id        string          `json:"id"`
	Task      string          `json:"task"`
	Data      json.RawMessage `json:"data"`
	RunAt     Timestamp       `json:"run_at"`     // 应执行时间
	ExpiredAt Timestamp       `json:"expired_at"` // 过期时间
	CreatedAt Timestamp       `json:"created_at"` // 创建时间
	Meta      Meta            `json:"meta,omitempty"`
}

type Messages []*Message

type Meta struct {
	RetryRule []int  `json:"retry_rule,omitempty"` // 重试规则，单位秒
	Retry     [2]int `json:"retry,omitempty"`      // 当前执行次数，总共重试测试
	Delay     int    `json:"delay,omitempty"`      // 延迟时间
	Timeout   int    `json:"timeout,omitempty"`    // 超时时间，单位秒
	TraceId   string `json:"trace_id,omitempty"`   // 用于打通trace
}

type OnComplete interface {
	OnComplete(ctx context.Context)
}

type OnFail interface {
	OnFail(ctx context.Context)
}

type OnLoad interface {
	Load(ctx context.Context, msg *Message) error
}

type OnSuccess interface {
	OnSuccess(ctx context.Context)
}

type Process interface {
	Exec(ctx context.Context, msg *TaskRuntime) (err error)
}

type Producer interface {
	Push(ctx context.Context)
}

type RedisBroker struct {
	// contains filtered or unexported fields
}

type RedisBrokerConfig struct {
	Key          string        `json:"key" toml:"key" yaml:"key"`
	WaitDuration time.Duration `json:"wait_duration" toml:"wait_duration" yaml:"wait_duration"`

	DelayKey          string        `json:"delay_key" toml:"delay_key" yaml:"delay_key"`
	DelayWaitDuration time.Duration `json:"delay_wait_duration" yaml:"delay_wait_duration" json:"delay_wait_duration"`
}

type Register struct {
	sync.RWMutex
	// contains filtered or unexported fields
}

type Rmq struct {

	// 钩子
	Hook *Hook
	// contains filtered or unexported fields
}

type Task interface {
	TaskName() string
	Run(ctx context.Context) (result string, err error)
}

type TaskInfo struct {
	Name        string
	IsCallback  bool
	ReflectType reflect.Type
	Callback    Callback
}

type TaskName interface {
	TaskName() string
}

type TaskRuntime struct {
	Msg       *Message      `json:"msg"`
	StartTime time.Time     `json:"start_time"`
	EndTime   time.Time     `json:"end_time"`
	Duration  time.Duration `json:"duration"`
	TaskError error         `json:"task_error,omitempty"` // 执行错误
	Error     error         `json:"error,omitempty"`      // 最后的错误
	Result    string        `json:"result,omitempty"`     // 结果
}

type TaskScanner interface {
	Scan(src []byte) error
}

type TaskValuer interface {
	Value() ([]byte, error)
}

type Timestamp int64

type WorkerConfig struct {
	// 配置信息
	WorkerNum  int
	Concurrent int
}