albert/cml04-falcon-system
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

wip

Browse Source
This commit is contained in:
Albert Espín 2024-10-22 09:56:39 +02:00
parent 3343c3ff09
commit cd4e313d06
3 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
cmd/labeler.go
View File

@ -19,6 +19,7 @@ var labelerCmd = &cobra.Command{
func init() { func init() {
rootCmd.AddCommand(labelerCmd) rootCmd.AddCommand(labelerCmd)
// flags // flags
labelerCmd.Flags().String("bind-addr", ":3000", "tcp bind address")
labelerCmd.Flags().String("db-username", "postgres", "database username") labelerCmd.Flags().String("db-username", "postgres", "database username")
labelerCmd.Flags().String("db-password", "Me8140@01", "database user password") labelerCmd.Flags().String("db-password", "Me8140@01", "database user password")
labelerCmd.Flags().String("db-host", "db", "database host address") labelerCmd.Flags().String("db-host", "db", "database host address")

5
labeler/app/app.go
View File

@ -15,6 +15,7 @@ import (
func Run(cmd *cobra.Command, args []string) { func Run(cmd *cobra.Command, args []string) {
// read flags // read flags
logLevel, _ := cmd.Flags().GetString("log-level") logLevel, _ := cmd.Flags().GetString("log-level")
bindAddr, _ := cmd.Flags().GetString("bind-addr")
userName, _ := cmd.Flags().GetString("db-username") userName, _ := cmd.Flags().GetString("db-username")
userPass, _ := cmd.Flags().GetString("db-password") userPass, _ := cmd.Flags().GetString("db-password")
dbHost, _ := cmd.Flags().GetString("db-host") dbHost, _ := cmd.Flags().GetString("db-host")
@ -51,8 +52,8 @@ func Run(cmd *cobra.Command, args []string) {
} }
defer svc.Close(ctx) defer svc.Close(ctx)
// create server // create server
server := server.New("", svc, log) server := server.New(bindAddr, svc, log)
server.Start() go server.Start()
// info banner // info banner
log.Info("started cml04-falcon-labeler service", logFields) log.Info("started cml04-falcon-labeler service", logFields)
// wait signal to finish // wait signal to finish

6
labeler/server/server.go
View File

@ -100,7 +100,7 @@ func (s *Server) handleConnection(conn net.Conn) {
break break
} }
s.log.Error("reading telegram failed", err, logger.LogFields{}) s.log.Error("reading telegram failed", err, logger.LogFields{})
continue // Keep reading from the connection return // Keep reading from the connection
} }
// Check if TelegramID is 6500, if not discard but keep connection // Check if TelegramID is 6500, if not discard but keep connection
@ -109,13 +109,13 @@ func (s *Server) handleConnection(conn net.Conn) {
"id": labelData.Header.TelegramID, "id": labelData.Header.TelegramID,
"from": conn.RemoteAddr(), "from": conn.RemoteAddr(),
}) })
continue // Discard the content but keep reading from the connection return // Discard the content but keep reading from the connection
} }
// Check if MessageLength matches the data received // Check if MessageLength matches the data received
if labelData.Header.MessageLength != int16(bytesRead) { if labelData.Header.MessageLength != int16(bytesRead) {
fmt.Printf("MessageLength mismatch: expected %d, got %d bytes from %s\n", labelData.Header.MessageLength, bytesRead, conn.RemoteAddr()) fmt.Printf("MessageLength mismatch: expected %d, got %d bytes from %s\n", labelData.Header.MessageLength, bytesRead, conn.RemoteAddr())
continue // Discard the content but keep reading from the connection return // Discard the content but keep reading from the connection
} }
// Locking if necessary when interacting with shared resources (svc, etc.) // Locking if necessary when interacting with shared resources (svc, etc.)