ekuiper-new/xstream/sinks/mqtt_sink.go

package sinks

import (
	"context"
	"engine/common"
	"fmt"
	MQTT "github.com/eclipse/paho.mqtt.golang"
	"github.com/google/uuid"
	"strings"
)

type MQTTSink struct {
	srv      string
	tpc      string
	clientid string
	pVersion uint
	uName 	string
	password string

	input chan interface{}
	conn MQTT.Client
	ruleId   string
	name 	 string
	//ctx context.Context
}

func NewMqttSink(name string, ruleId string, properties interface{}) (*MQTTSink, error) {
	ps, ok := properties.(map[string]interface{})
	if !ok {
		return nil, fmt.Errorf("expect map[string]interface{} type for the mqtt sink properties")
	}
	srv, ok := ps["server"]
	if !ok {
		return nil, fmt.Errorf("mqtt sink is missing property server")
	}
	tpc, ok := ps["topic"]
	if !ok {
		return nil, fmt.Errorf("mqtt sink is missing property topic")
	}
	clientid, ok := ps["clientId"]
	if !ok{
		if uuid, err := uuid.NewUUID(); err != nil {
			return nil, fmt.Errorf("mqtt sink fails to get uuid, the error is %s", err)
		}else{
			clientid = uuid.String()
		}
	}
	var pVersion uint = 3
	pVersionStr, ok := ps["protocol_version"];
	if ok {
		v, _ := pVersionStr.(string)
		if v == "3.1" {
			pVersion = 3
		} else if v == "3.1.1" {
			pVersion = 4
		} else {
			return nil, fmt.Errorf("Unknown protocol version {0}, the value could be only 3.1 or 3.1.1 (also refers to MQTT version 4).", pVersionStr)
		}
	}

	uName := ""
	un, ok := ps["username"];
	if ok {
		v, _ := un.(string)
		if strings.Trim(v, " ") != "" {
			uName = v
		}
	}

	password := ""
	pwd, ok := ps["password"];
	if ok {
		v, _ := pwd.(string)
		if strings.Trim(v, " ") != "" {
			password = v
		}
	}

	ms := &MQTTSink{name:name, ruleId: ruleId, input: make(chan interface{}), srv: srv.(string), tpc: tpc.(string), clientid: clientid.(string), pVersion:pVersion, uName:uName, password:password}
	return ms, nil
}

func (ms *MQTTSink) GetName() string {
	return ms.name
}

func (ms *MQTTSink) GetInput() (chan<- interface{}, string)  {
	return ms.input, ms.name
}

func (ms *MQTTSink) Open(ctx context.Context, result chan<- error) {
	log := common.GetLogger(ctx)
	log.Printf("Opening mqtt sink for rule %s", ms.ruleId)

	go func() {
		exeCtx, cancel := context.WithCancel(ctx)
		opts := MQTT.NewClientOptions().AddBroker(ms.srv).SetClientID(ms.clientid).SetProtocolVersion(ms.pVersion)
		if ms.uName != "" {
			opts = opts.SetUsername(ms.uName)
		}

		if ms.password != "" {
			opts = opts.SetPassword(ms.password)
		}

		c := MQTT.NewClient(opts)
		if token := c.Connect(); token.Wait() && token.Error() != nil {
			result <- fmt.Errorf("Found error: %s", token.Error())
			cancel()
		}
		log.Printf("The connection to server %s was established successfully", ms.srv)
		ms.conn = c

		for {
			select {
			case item := <-ms.input:
				log.Infof("publish %s", item)
				if token := c.Publish(ms.tpc, 0, false, item); token.Wait() && token.Error() != nil {
					result <- fmt.Errorf("Publish error: %s", token.Error())
				}

			case <-exeCtx.Done():
				c.Disconnect(5000)
				log.Infof("Closing mqtt sink")
				cancel()
				return
			}
		}

	}()
}