60 lines
2.2 KiB
Go
60 lines
2.2 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"github.com/cyrilix/robocar-arduino/arduino"
|
||
|
|
"github.com/cyrilix/robocar-base/mqttdevice"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"strconv"
|
||
|
|
)
|
||
|
|
|
||
|
|
const DefaultClientId = "robocar-arduino"
|
||
|
|
|
||
|
|
func setDefaultValueFromEnv(value *string, key string, defaultValue string) {
|
||
|
|
if os.Getenv(key) != "" {
|
||
|
|
*value = os.Getenv(key)
|
||
|
|
} else {
|
||
|
|
*value = defaultValue
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
var mqttBroker, username, password, qos, clientId, topicBase string
|
||
|
|
var device string
|
||
|
|
var baud int
|
||
|
|
var pubFrequency float64
|
||
|
|
|
||
|
|
setDefaultValueFromEnv(&clientId, "MQTT_CLIENT_ID", DefaultClientId)
|
||
|
|
setDefaultValueFromEnv(&mqttBroker, "MQTT_BROKER", "tcp://127.0.0.1:1883")
|
||
|
|
setDefaultValueFromEnv(&qos, "MQTT_QOS", "0")
|
||
|
|
mqttQos, err := strconv.Atoi(qos)
|
||
|
|
if err != nil {
|
||
|
|
log.Panicf("invalid mqtt qos value: %v", qos)
|
||
|
|
}
|
||
|
|
_, mqttRetain := os.LookupEnv("MQTT_RETAIN")
|
||
|
|
|
||
|
|
flag.StringVar(&mqttBroker, "mqtt-broker", mqttBroker, "Broker Uri, use MQTT_BROKER env if arg not set")
|
||
|
|
flag.StringVar(&username, "mqtt-username", os.Getenv("MQTT_USERNAME"), "Broker Username, use MQTT_USERNAME env if arg not set")
|
||
|
|
flag.StringVar(&password, "mqtt-password", os.Getenv("MQTT_PASSWORD"), "Broker Password, MQTT_PASSWORD env if args not set")
|
||
|
|
flag.StringVar(&clientId, "mqtt-client-id", clientId, "Mqtt client id, use MQTT_CLIENT_ID env if args not set")
|
||
|
|
flag.IntVar(&mqttQos, "mqtt-qos", mqttQos, "Qos to pusblish message, use MQTT_QOS env if arg not set")
|
||
|
|
flag.StringVar(&topicBase, "mqtt-topic-base", os.Getenv("MQTT_TOPIC_BASE"), "Mqtt topic prefix, use MQTT_TOPIC_BASE if args not set")
|
||
|
|
flag.BoolVar(&mqttRetain, "mqtt-retain", mqttRetain, "Retain mqtt message, if not set, true if MQTT_RETAIN env variable is set")
|
||
|
|
flag.Float64Var(&pubFrequency, "mqtt-pub-frequency", 25., "Number of messages to publish per second")
|
||
|
|
flag.StringVar(&device, "device", "/dev/serial0", "Serial device")
|
||
|
|
flag.IntVar(&baud, "baud", 115200, "Serial baud")
|
||
|
|
|
||
|
|
flag.Parse()
|
||
|
|
if len(os.Args) <= 1 {
|
||
|
|
flag.PrintDefaults()
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
|
||
|
|
pubSub := mqttdevice.NewPahoMqttPubSub(mqttBroker, username, password, clientId, mqttQos, mqttRetain)
|
||
|
|
defer pubSub.Close()
|
||
|
|
|
||
|
|
a := arduino.NewArduinoPart(device, baud, pubSub,topicBase, pubFrequency)
|
||
|
|
a.Start()
|
||
|
|
}
|