debug.go

package mqttx

import (
	"log"

	mqtt "github.com/eclipse/paho.mqtt.golang"
)

const (
	LevelEmergency = iota
	LevelAlert
	LevelCritical
	LevelError
	LevelWarning
	LevelNotice
	LevelInformational
	LevelDebug
)

var debugEnable bool

type TraceFunc func(format string, level int, v ...interface{})

var UserTrace TraceFunc = nil

func init() {
	debugEnable = false
	log.SetPrefix("[mqttx] TRACE: ")
	log.SetFlags(log.LstdFlags | log.Llongfile)
}

// Debug Enable debug
func Debug(enable bool) {
	debugEnable = enable
}

// MqttLibDebug Enable mqtt lib debug
func MqttLibDebug(enable bool, level int) {
	if enable {
		mqtt.ERROR = mqttLibDebugNew(LevelError)
		mqtt.CRITICAL = mqttLibDebugNew(LevelCritical)
		mqtt.WARN = mqttLibDebugNew(LevelWarning)
		mqtt.DEBUG = mqttLibDebugNew(LevelDebug)
	}
}

// SetUserDebug 配置其他日志输出
func SetUserDebug(f TraceFunc) {
	UserTrace = f
}

// TraceInfo 调试信息日志
func TraceInfo(format string, v ...interface{}) {
	if debugEnable {
		if UserTrace != nil {
			UserTrace(format, LevelInformational, v...)
		} else {
			log.Printf(format, v...)
		}
	}
}

// TraceError 错误日志
func TraceError(format string, v ...interface{}) {
	if debugEnable {
		if UserTrace != nil {
			UserTrace(format, LevelError, v...)
		} else {
			log.Printf(format, v...)
		}
	}
}

// TraceDebug Debug调试信息日志
func TraceDebug(format string, v ...interface{}) {
	if debugEnable {
		if UserTrace != nil {
			UserTrace(format, LevelDebug, v...)
		} else {
			log.Printf(format, v...)
		}
	}
}