migrate from github

pkg/mq/mq_api.go

@@ -0,0 +1,86 @@
+package mq
+
+import (
+	"context"
+	"strings"
+	"time"
+
+	"github.com/XingfenD/yoresee_doc/pkg/errs"
+)
+
+type Message struct {
+	ID        string
+	Topic     string
+	Key       string
+	Body      []byte
+	Headers   map[string]string
+	Timestamp time.Time
+}
+
+type PublishMessage struct {
+	Topic   string
+	Key     string
+	Body    []byte
+	Headers map[string]string
+}
+
+type ConsumeMode string
+
+const (
+	ConsumeModeFanout ConsumeMode = "fanout"
+	ConsumeModeGroup  ConsumeMode = "group"
+)
+
+type ErrorAction string
+
+const (
+	ErrorActionDrop    ErrorAction = "drop"
+	ErrorActionRequeue ErrorAction = "requeue"
+)
+
+type ConsumeOptions struct {
+	Topic    string
+	Mode     ConsumeMode
+	Group    string
+	Consumer string
+	AutoAck  bool
+	OnError  ErrorAction
+}
+
+func (o ConsumeOptions) normalize() ConsumeOptions {
+	n := o
+	n.Topic = strings.TrimSpace(n.Topic)
+	n.Group = strings.TrimSpace(n.Group)
+	n.Consumer = strings.TrimSpace(n.Consumer)
+	if n.Mode == "" {
+		n.Mode = ConsumeModeFanout
+	}
+	if n.OnError == "" {
+		n.OnError = ErrorActionRequeue
+	}
+	return n
+}
+
+func (o ConsumeOptions) validate() error {
+	switch o.Mode {
+	case ConsumeModeFanout, ConsumeModeGroup:
+	default:
+		return errs.Detailf(errs.ErrInvalidConsumeMode, "%s", o.Mode)
+	}
+
+	switch o.OnError {
+	case ErrorActionDrop, ErrorActionRequeue:
+	default:
+		return errs.Detailf(errs.ErrInvalidErrorAction, "%s", o.OnError)
+	}
+
+	return nil
+}
+
+type MessageHandler func(ctx context.Context, msg Message) error
+
+type MessageQueue interface {
+	Publish(ctx context.Context, msg PublishMessage) error
+	Consume(ctx context.Context, opts ConsumeOptions, handler MessageHandler) error
+	Close() error
+}