SheXiang_mq

type AllocateMessageQueueStrategy interface {
	// Name The strategy name
	Name() string

	// Allocate To allocate result of given strategy
	Allocate() []MessageQueue
}

type ConsumeConcurrentlyStatus int

type Consumer interface {
	// Start the consumer with the given parameters and return immediately
	Start() error

	// Shutdown Stop the consumer with the given parameters and return immediately
	Shutdown()

	ShutdownCallback(callback func())

	// Subscribe with the given parameters and return immediately
	Subscribe(topic string) bool

	// Unsubscribe with the given parameters and return immediately
	Unsubscribe(topics ...string)

	// RegisterMessageListener with the given parameters and return immediately and
	RegisterMessageListener(topic string, l func(message Message) ConsumeConcurrentlyStatus)
}

type ConsumerConfig struct {
	PoolSize         int
	ConsumerGroup    string
	MessageListeners map[string]func(message Message) ConsumeConcurrentlyStatus
}

type DefaultConsumer struct {
	ConsumerGroup string
	ServiceState  ServiceState
	Listeners     map[string]func(message Message) ConsumeConcurrentlyStatus
	// contains filtered or unexported fields
}

type DefaultProducer struct {
	ProducerGroup string
	ServiceState  ServiceState
	MqFactory     *MqFactory
}

type FactoryConfig struct {
	ToPicConfigs []*ToPicConfig
	//Topic 队列选择器 选填默认随机 优先ToPicConfigs 中
	Selector MsgQueueSelector
}

type Message struct {
	Topic      string
	Body       []byte
	Properties map[string]any
}

type MessageQueue struct {
	QueueId string
	Message chan Message
}

type MessageQueueSelector interface {
	Select(messageQueues []*MessageQueue) *MessageQueue
}

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

type MonitorListener interface {
	Surround(messageListener func(message Message) ConsumeConcurrentlyStatus, msg Message)

	Print()

	Info() map[string]*Msg

	TurnMonitor()

	CloseMonitor()
}

type Mq struct {
	Consumer        Consumer
	Producer        Producer
	MonitorListener MonitorListener
}

type MqConfig struct {
	//Topic 队列选择器 //默认随机 ToPicConfig 不配置 默认去mq
	Selector             MsgQueueSelector
	ProducerConfig       *ProducerConfig
	ConsumerConfig       *ConsumerConfig
	ToPicConfigs         []*ToPicConfig
	DefaultTopicPoolSize int
}

type MqFactory struct {
	TopicPublishInfoTable map[string]*TopicPublishInfo
	// contains filtered or unexported fields
}

type Msg struct {
	// topicName name for monitoring
	TopicName string
	// consumerGroup name for monitoring
	ConsumerGroup string
	// taskCount task count for monitoring
	TaskCount int64
	// timeCount time count for monitoring
	TimeCount int64
	// maxTime max time task  for monitoring
	MaxTime int64
	// minTime min time task for monitoring
	MinTime int64

	Running int64

	Waiting int64
	// started
	StartTime time.Time
	//
	LastTaskTime int64
	// contains filtered or unexported fields
}

type MsgQueueSelector int

type Pool struct {
	Pool ants.Pool
}

type Producer interface {
	Start() error

	Shutdown()

	Send(msg Message) error
}

type ProducerConfig struct {
	ProducerGroupName string
}

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

type SelectMessageQueueByRandom struct{}

type ServiceState int

type ToPicConfig struct {
	//名称
	TopicName string
	//队列长度
	MessageQueueLength int
	//队列缓存容量
	MessageCapLength int
	//Topic 队列选择器 //默认随机
	Selector MsgQueueSelector
	// 可用线程数
	PoolSize int
	// contains filtered or unexported fields
}

type TopicPublishInfo struct {
	ToPicConfig          *ToPicConfig
	MessageQueueSelector MessageQueueSelector
	// contains filtered or unexported fields
}