refactor: Update main.go to process Civial Aviation Authority Telegram messages and add CLI flag for Nats server address and topic

This commit is contained in:
windyboy
2024-08-05 10:58:20 +08:00
parent c54bcbc0a8
commit 3c917c93af
3 changed files with 49 additions and 23 deletions
+43 -17
View File
@@ -12,6 +12,10 @@ import (
"github.com/urfave/cli/v2"
)
var (
cfg *config.Config
)
func main() {
app := setupApp()
if err := app.Run(os.Args); err != nil {
@@ -21,16 +25,40 @@ func main() {
func setupApp() *cli.App {
app := &cli.App{
Name: "serial-read",
Usage: "A serial port reading CLI application",
Name: "telegram message process",
Usage: "A Civial Aviation Authority Telegram message processor",
Before: func(c *cli.Context) error {
// parameter = config.GetParameter()
cfg, err := config.LoadConfig()
if err != nil {
fmt.Printf("Error loading configuration: %v\n", err)
return err
}
if err := config.ValidateConfig(cfg); err != nil {
fmt.Printf("Invalid configuration: %v\n", err)
return err
}
overrideConfig(c)
return nil
},
Commands: []*cli.Command{
{
Name: "listen",
Usage: "Listen to nats messages",
Name: "listen",
Usage: "Listen to nats messages",
Flags: []cli.Flag{
&cli.StringFlag{
Name: "nats",
Usage: "Nats server address",
Value: "nats://localhost:4222",
EnvVars: []string{"NATS_SERVER"},
},
&cli.StringFlag{
Name: "topic",
Usage: "Nats topic to listen to",
Value: "Telegram.Serial",
EnvVars: []string{"NATS_SUBJECT"},
},
},
Action: executeListen,
},
},
@@ -38,21 +66,19 @@ func setupApp() *cli.App {
return app
}
func overrideConfig(c *cli.Context) {
if c.IsSet("nats") {
cfg.Nats.URL = c.String("nats")
}
if c.IsSet("topic") {
cfg.Subscription.Topic = c.String("topic")
}
}
func executeListen(c *cli.Context) error {
cfg, err := config.LoadConfig()
// cfg, err := config.LoadConfig()
log := utils.GetLogger()
if err != nil {
fmt.Printf("Error loading configuration: %v\n", err)
return err
}
if err := config.ValidateConfig(cfg); err != nil {
fmt.Printf("Invalid configuration: %v\n", err)
return err
}
fmt.Println("Loaded configuration successfully")
log.Info("Starting nats subscriber")
handler := handlers.NewNatsHandler(cfg)