github.com/isyscore/isc-gobase@v1.5.3-0.20231218061332-cbc7451899e9/extend/emqx/emqx.go

package emqx

import (
	"errors"
	mqtt "github.com/eclipse/paho.mqtt.golang"
	"github.com/isyscore/isc-gobase/config"
	"github.com/isyscore/isc-gobase/logger"
	t0 "time"
)

func init() {
	config.LoadConfig()

	if config.ExistConfigFile() && config.GetValueBoolDefault("base.emqx.enable", false) {
		err := config.GetValueObject("base.emqx", &config.EmqxCfg)
		if err != nil {
			logger.Warn("读取emqx配置异常, %v", err.Error())
			return
		}

		mqtt.DEBUG = emqxLogger{"DEBUG"}
		mqtt.WARN = emqxLogger{"WARN"}
		mqtt.CRITICAL = emqxLogger{"CRITICAL"}
		mqtt.ERROR = emqxLogger{"ERROR"}
	}
}

type emqxLogger struct {
	Level string
}

func (log emqxLogger) Println(v ...interface{}) {
	switch log.Level {
	case "DEBUG":
		logger.DebugDirect(v...)
	case "WARN":
		logger.WarnDirect(v...)
	case "CRITICAL":
		logger.ErrorDirect(v...)
	case "ERROR":
		logger.ErrorDirect(v...)
	}
}
func (log emqxLogger) Printf(format string, v ...interface{}) {
	switch log.Level {
	case "DEBUG":
		logger.Debug(format, v...)
	case "WARN":
		logger.Warn(format, v...)
	case "CRITICAL":
		logger.Error(format, v...)
	case "ERROR":
		logger.Error(format, v...)
	}
}

func NewEmqxClient() (mqtt.Client, error) {
	if !config.GetValueBoolDefault("base.emqx.enable", false) {
		logger.Error("emqx没有配置，请先配置")
		return nil, errors.New("emqx没有配置，请先配置")
	}

	_emqxClient := mqtt.NewClient(localEmqxOptions())
	if token := _emqxClient.Connect(); token.Wait() && token.Error() != nil {
		logger.Error("链接emqx client失败, %v", token.Error().Error())
		return nil, token.Error()
	}
	return _emqxClient, nil
}

func localEmqxOptions() *mqtt.ClientOptions {
	opts := mqtt.NewClientOptions()
	for _, server := range config.EmqxCfg.Servers {
		opts.AddBroker(server)
	}

	if config.EmqxCfg.ClientId != "" {
		opts.SetClientID(config.EmqxCfg.ClientId)
	}

	if config.EmqxCfg.Username != "" {
		opts.SetUsername(config.EmqxCfg.Username)
	}

	if config.EmqxCfg.Password != "" {
		opts.SetPassword(config.EmqxCfg.Password)
	}

	if config.EmqxCfg.CleanSession != true && config.GetValueString("base.emqx.clean-session") != "" {
		opts.SetCleanSession(config.EmqxCfg.CleanSession)
	}

	if config.EmqxCfg.Order != true && config.GetValueString("base.emqx.order") != "" {
		opts.SetOrderMatters(config.EmqxCfg.Order)
	}

	if config.EmqxCfg.WillEnabled != false && config.GetValueString("base.emqx.will-enabled") != "" {
		opts.WillEnabled = config.EmqxCfg.WillEnabled
	}

	if config.EmqxCfg.WillTopic != "" {
		opts.WillTopic = config.EmqxCfg.WillTopic
	}

	if config.EmqxCfg.WillQos != 0 {
		opts.WillQos = config.EmqxCfg.WillQos
	}

	if config.EmqxCfg.WillRetained != false && config.GetValueString("base.emqx.will-retained") != "" {
		opts.WillRetained = config.EmqxCfg.WillRetained
	}

	if config.EmqxCfg.ProtocolVersion != 0 {
		opts.ProtocolVersion = config.EmqxCfg.ProtocolVersion
	}

	if config.EmqxCfg.KeepAlive != 30 && config.GetValueString("base.emqx.keep-alive") != "" {
		opts.KeepAlive = config.EmqxCfg.KeepAlive
	}

	if config.EmqxCfg.PingTimeout != "10s" && config.GetValueString("base.emqx.ping-timeout") != "" {
		t, err := t0.ParseDuration(config.EmqxCfg.PingTimeout)
		if err != nil {
			logger.Warn("读取配置【base.emqx.ping-timeout】异常：%v", err.Error())
		} else {
			opts.PingTimeout = t
		}
	}

	if config.EmqxCfg.ConnectTimeout != "30s" && config.GetValueString("base.emqx.connect-timeout") != "" {
		t, err := t0.ParseDuration(config.EmqxCfg.ConnectTimeout)
		if err != nil {
			logger.Warn("读取配置【base.emqx.connect-timeout】异常：%v", err.Error())
		} else {
			opts.PingTimeout = t
		}
	}

	if config.EmqxCfg.MaxReconnectInterval != "10m" && config.GetValueString("base.emqx.max-reconnect-interval") != "" {
		t, err := t0.ParseDuration(config.EmqxCfg.MaxReconnectInterval)
		if err != nil {
			logger.Warn("读取配置【base.emqx.max-reconnect-interval】异常：%v", err.Error())
		} else {
			opts.MaxReconnectInterval = t
		}
	}

	if config.EmqxCfg.AutoReconnect != true && config.GetValueString("base.emqx.auto-reconnect") != "" {
		opts.AutoReconnect = config.EmqxCfg.AutoReconnect
	}

	if config.EmqxCfg.ConnectRetryInterval != "30s" && config.GetValueString("base.emqx.connect-retry-interval") != "" {
		t, err := t0.ParseDuration(config.EmqxCfg.ConnectRetryInterval)
		if err != nil {
			logger.Warn("读取配置【base.emqx.connect-retry-interval】异常：%v", err.Error())
		} else {
			opts.ConnectRetryInterval = t
		}
	}

	if config.EmqxCfg.ConnectRetry != false && config.GetValueString("base.emqx.connect-retry") != "" {
		opts.ConnectRetry = config.EmqxCfg.ConnectRetry
	}

	if config.EmqxCfg.WriteTimeout != "0" && config.GetValueString("base.emqx.write-timeout") != "" {
		t, err := t0.ParseDuration(config.EmqxCfg.WriteTimeout)
		if err != nil {
			logger.Warn("读取配置【base.emqx.write-timeout】异常：%v", err.Error())
		} else {
			opts.WriteTimeout = t
		}
	}

	if config.EmqxCfg.ResumeSubs != false && config.GetValueString("base.emqx.resume-subs") != "" {
		opts.ResumeSubs = config.EmqxCfg.ResumeSubs
	}

	if config.EmqxCfg.MaxResumePubInFlight != 0 {
		opts.MaxResumePubInFlight = config.EmqxCfg.MaxResumePubInFlight
	}

	if config.EmqxCfg.AutoAckDisabled != false && config.GetValueString("base.emqx.auto-ack-disabled") != "" {
		opts.AutoAckDisabled = config.EmqxCfg.AutoAckDisabled
	}

	return opts
}