mirror of
https://github.com/mochi-mqtt/server.git
synced 2025-09-26 20:21:12 +08:00
57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
// SPDX-License-Identifier: MIT
|
|
// SPDX-FileCopyrightText: 2023 mochi-mqtt
|
|
// SPDX-FileContributor: dgduncan, mochi-co
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"github.com/mochi-mqtt/server/v2/config"
|
|
"log"
|
|
"log/slog"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
mqtt "github.com/mochi-mqtt/server/v2"
|
|
)
|
|
|
|
func main() {
|
|
slog.SetDefault(slog.New(slog.NewTextHandler(os.Stdout, nil))) // set basic logger to ensure logs before configuration are in a consistent format
|
|
|
|
configFile := flag.String("config", "config.yaml", "path to mochi config yaml or json file")
|
|
flag.Parse()
|
|
|
|
sigs := make(chan os.Signal, 1)
|
|
done := make(chan bool, 1)
|
|
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
|
|
go func() {
|
|
<-sigs
|
|
done <- true
|
|
}()
|
|
|
|
configBytes, err := os.ReadFile(*configFile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
options, err := config.FromBytes(configBytes)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
server := mqtt.New(options)
|
|
|
|
go func() {
|
|
err := server.Serve()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}()
|
|
|
|
<-done
|
|
server.Log.Warn("caught signal, stopping...")
|
|
_ = server.Close()
|
|
server.Log.Info("mochi mqtt shutdown complete")
|
|
}
|