message.go

package rmq

import (
	"encoding/json"
	"time"

	uuid "github.com/satori/go.uuid"
)

type Messages []*Message

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"`
}

func NewMsg() *Message {
	now := Now()
	msg := &Message{
		Id:        uuid.NewV4().String(),
		RunAt:     now,
		ExpiredAt: now.Add(24 * time.Hour), // 默认24小时过期
		Meta:      RetryMeta,
		CreatedAt: now,
	}
	return msg
}

func NewBlankMsg() *Message {
	now := Now()
	msg := &Message{
		Id:        uuid.NewV4().String(),
		RunAt:     now,
		ExpiredAt: now.Add(24 * time.Hour), // 默认24小时过期
		Meta:      DefaultMeta,
		CreatedAt: now,
	}
	return msg
}

func (m *Message) SetTask(task Task) (message *Message, err error) {
	m.Task = task.TaskName()
	if impl, ok := task.(TaskValuer); ok {
		m.Data, err = impl.Value()
	} else {
		m.Data, err = json.Marshal(task)
	}
	message = m
	return
}

func (m *Message) SetRawTask(name string, data any) (msg *Message, err error) {
	m.Task = name
	m.Data, err = json.Marshal(data)
	msg = m
	return
}

func (m *Message) SetMeta(meta Meta) *Message {
	m.Meta = meta
	return m
}

func (m *Message) SetData(data any) *Message {
	m.Data, _ = json.Marshal(data)
	return m
}

func (m *Message) SetRawData(data json.RawMessage) *Message {
	m.Data = data
	return m
}

func (m *Message) SetMaxRetry(retry int) *Message {
	m.Meta.Retry[1] = retry
	return m
}

func (m *Message) SetRetryRule(rule []int) *Message {
	m.Meta.RetryRule = rule
	return m
}

func (m *Message) SetDelay(delay time.Duration) *Message {
	m.Meta.Delay = int(delay.Seconds())
	m.RunAt = m.RunAt.Add(delay)
	return m
}

func (m *Message) SetExpiredAt(t time.Time) *Message {
	m.ExpiredAt = NewTimestamp(t)
	return m
}

func (m *Message) SetExpire(d time.Duration) *Message {
	m.ExpiredAt = m.RunAt.Add(d)
	return m
}

func (m *Message) SetTimeout(t time.Duration) *Message {
	m.Meta.Timeout = int(t.Seconds())
	return m
}

func (m *Message) SetTraceId(traceId string) *Message {
	m.Meta.TraceId = traceId
	return m
}

func (m *Message) TryRetry(delay time.Duration) *Message {
	m.Meta.Retry[0] += 1
	m.Meta.Delay = int(delay.Seconds())
	m.RunAt = Now().Add(delay)
	return m
}

func (m *Message) String() string {
	data, _ := json.Marshal(m)
	return string(data)
}