142 lines
4.2 KiB
Go
Executable File
142 lines
4.2 KiB
Go
Executable File
// Made by Leandro Antônio Farias Machado
|
|
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"strconv"
|
|
"sync"
|
|
"syscall"
|
|
|
|
"github.com/joho/godotenv"
|
|
"github.com/leandrofars/oktopus/internal/api"
|
|
"github.com/leandrofars/oktopus/internal/db"
|
|
usp_msg "github.com/leandrofars/oktopus/internal/usp_message"
|
|
|
|
"github.com/leandrofars/oktopus/internal/mqtt"
|
|
"github.com/leandrofars/oktopus/internal/mtp"
|
|
)
|
|
|
|
const VERSION = "0.0.1"
|
|
|
|
func main() {
|
|
done := make(chan os.Signal, 1)
|
|
|
|
err := godotenv.Load()
|
|
|
|
localEnv := ".env.local"
|
|
if _, err := os.Stat(localEnv); err == nil {
|
|
_ = godotenv.Overload(localEnv)
|
|
log.Println("Loaded variables from '.env.local'")
|
|
} else {
|
|
log.Println("Loaded variables from '.env'")
|
|
}
|
|
|
|
if err != nil {
|
|
log.Println("Error to load environment variables:", err)
|
|
}
|
|
|
|
// Locks app running until it receives a stop command as Ctrl+C.
|
|
signal.Notify(done, syscall.SIGINT, syscall.SIGTERM)
|
|
|
|
log.SetFlags(log.LstdFlags | log.Lshortfile)
|
|
|
|
/*
|
|
App variables priority:
|
|
1º - Flag through command line.
|
|
2º - Env variables.
|
|
3º - Default flag value
|
|
*/
|
|
|
|
log.Println("Starting Oktopus Project TR-369 Controller Version:", VERSION)
|
|
// fl_endpointId := flag.String("endpoint_id", "proto::oktopus-controller", "Defines the enpoint id the Agent must trust on.")
|
|
flDevicesTopic := flag.String("d", lookupEnvOrString("DEVICES_STATUS_TOPIC", "oktopus/+/status/+"), "That's the topic mqtt broker end new devices info.")
|
|
flSubTopic := flag.String("sub", lookupEnvOrString("DEVICE_PUB_TOPIC", "oktopus/+/controller/+"), "That's the topic agent must publish to, and the controller keeps on listening.")
|
|
flBrokerAddr := flag.String("a", lookupEnvOrString("BROKER_ADDR", "localhost"), "Mqtt broker adrress")
|
|
flBrokerPort := flag.String("p", lookupEnvOrString("BROKER_PORT", "1883"), "Mqtt broker port")
|
|
flTlsCert := flag.Bool("tls", lookupEnvOrBool("BROKER_TLS", false), "Connect to broker over TLS")
|
|
flBrokerUsername := flag.String("u", lookupEnvOrString("BROKER_USERNAME", ""), "Mqtt broker username")
|
|
flBrokerPassword := flag.String("P", lookupEnvOrString("BROKER_PASSWORD", ""), "Mqtt broker password")
|
|
flBrokerClientId := flag.String("i", lookupEnvOrString("BROKER_CLIENTID", ""), "A clientid for the Mqtt connection")
|
|
flBrokerQos := flag.Int("q", lookupEnvOrInt("BROKER_QOS", 0), "Quality of service of mqtt messages delivery")
|
|
flAddrDB := flag.String("mongo", lookupEnvOrString("MONGO_URI", "mongodb://localhost:27017"), "MongoDB URI")
|
|
flApiPort := flag.String("ap", lookupEnvOrString("REST_API_PORT", "8000"), "Rest api port")
|
|
flHelp := flag.Bool("help", false, "Help")
|
|
|
|
flag.Parse()
|
|
|
|
if *flHelp {
|
|
flag.Usage()
|
|
os.Exit(0)
|
|
}
|
|
/*
|
|
This context suppress our needs, but we can use a more sofisticate
|
|
approach with cancel and timeout options passing it through paho mqtt functions.
|
|
*/
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
database := db.NewDatabase(ctx, *flAddrDB)
|
|
apiMsgQueue := make(map[string](chan usp_msg.Msg))
|
|
var m sync.Mutex
|
|
/*
|
|
If you want to use another message protocol just make it implement Broker interface.
|
|
*/
|
|
mqttClient := mqtt.Mqtt{
|
|
Addr: *flBrokerAddr,
|
|
Port: *flBrokerPort,
|
|
Id: *flBrokerClientId,
|
|
User: *flBrokerUsername,
|
|
Passwd: *flBrokerPassword,
|
|
Ctx: ctx,
|
|
QoS: *flBrokerQos,
|
|
SubTopic: *flSubTopic,
|
|
DevicesTopic: *flDevicesTopic,
|
|
TLS: *flTlsCert,
|
|
DB: database,
|
|
MsgQueue: apiMsgQueue,
|
|
QMutex: &m,
|
|
}
|
|
|
|
mtp.MtpService(&mqttClient, done)
|
|
a := api.NewApi(*flApiPort, database, &mqttClient, apiMsgQueue, &m)
|
|
api.StartApi(a)
|
|
|
|
<-done
|
|
cancel()
|
|
|
|
log.Println("(⌐■_■) Oktopus is out!")
|
|
|
|
}
|
|
|
|
func lookupEnvOrString(key string, defaultVal string) string {
|
|
if val, _ := os.LookupEnv(key); val != "" {
|
|
return val
|
|
}
|
|
return defaultVal
|
|
}
|
|
|
|
func lookupEnvOrInt(key string, defaultVal int) int {
|
|
if val, _ := os.LookupEnv(key); val != "" {
|
|
v, err := strconv.Atoi(val)
|
|
if err != nil {
|
|
log.Fatalf("LookupEnvOrInt[%s]: %v", key, err)
|
|
}
|
|
return v
|
|
}
|
|
return defaultVal
|
|
}
|
|
|
|
func lookupEnvOrBool(key string, defaultVal bool) bool {
|
|
if val, _ := os.LookupEnv(key); val != "" {
|
|
v, err := strconv.ParseBool(val)
|
|
if err != nil {
|
|
log.Fatalf("LookupEnvOrInt[%s]: %v", key, err)
|
|
}
|
|
return v
|
|
}
|
|
return defaultVal
|
|
}
|