hmq/plugins/bridge/bridge.go

package bridge

import "github.com/fhmq/hmq/logger"

const (
	//Connect mqtt connect
	Connect = "connect"
	//Publish mqtt publish
	Publish = "publish"
	//Subscribe mqtt sub
	Subscribe = "subscribe"
	//Unsubscribe mqtt sub
	Unsubscribe = "unsubscribe"
	//Disconnect mqtt disconenct
	Disconnect = "disconnect"
)

var (
	log = logger.Get().Named("bridge")
)

//Elements kafka publish elements
type Elements struct {
	ClientID  string `json:"clientid"`
	Username  string `json:"username"`
	Topic     string `json:"topic"`
	Payload   string `json:"payload"`
	Timestamp int64  `json:"ts"`
	Size      int32  `json:"size"`
	Action    string `json:"action"`
}

const (
	//Kafka plugin name
	Kafka  = "kafka"
	CSVLog = "csvlog"
)

type BridgeMQ interface {
	// Publish return true to cost the message
	Publish(e *Elements) (bool, error)
}

func NewBridgeMQ(name string) BridgeMQ {
	switch name {
	case Kafka:
		return InitKafka()
	case CSVLog:
		return InitCSVLog()
	default:
		return &mockMQ{}
	}
}