104 lines
3.1 KiB
Go
Executable File
104 lines
3.1 KiB
Go
Executable File
// Made by Leandro Antônio Farias Machado
|
|
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"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"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"sync"
|
|
"syscall"
|
|
|
|
"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)
|
|
|
|
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", "oktopus/+/status/+", "That's the topic mqtt broker end new devices info.")
|
|
flSubTopic := flag.String("sub", "oktopus/+/controller/+", "That's the topic agent must publish to, and the controller keeps on listening.")
|
|
flBrokerAddr := flag.String("a", "localhost", "Mqtt broker adrress")
|
|
flBrokerPort := flag.String("p", "1883", "Mqtt broker port")
|
|
flTlsCert := flag.Bool("tls", false, "Connect to broker over TLS")
|
|
flBrokerUsername := flag.String("u", "", "Mqtt broker username")
|
|
flBrokerPassword := flag.String("P", "", "Mqtt broker password")
|
|
flBrokerClientId := flag.String("i", "", "A clientid for the Mqtt connection")
|
|
flBrokerQos := flag.Int("q", 0, "Quality of service of mqtt messages delivery")
|
|
flAddrDB := flag.String("mongo", "mongodb://localhost:27017/", "MongoDB URI")
|
|
flApiPort := flag.String("ap", "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!")
|
|
|
|
}
|