feat: add port flag

This commit is contained in:
Simon Ding
2025-08-30 14:46:42 +08:00
parent d265f17712
commit 1ddba4ddce

View File

@@ -1,6 +1,8 @@
package main package main
import ( import (
"flag"
"fmt"
"os" "os"
"polaris/db" "polaris/db"
"polaris/log" "polaris/log"
@@ -8,6 +10,9 @@ import (
) )
func main() { func main() {
port := flag.Int("port", 8080, "port to listen on")
flag.Parse()
if os.Getenv("GIN_MODE") == "release" { if os.Getenv("GIN_MODE") == "release" {
log.InitLogger(true) log.InitLogger(true)
} }
@@ -19,7 +24,7 @@ func main() {
} }
s := server.NewServer(dbClient) s := server.NewServer(dbClient)
if _, err := s.Start(":8080"); err != nil { if _, err := s.Start(fmt.Sprintf(":%d", *port)); err != nil {
log.Errorf("server start error: %v", err) log.Errorf("server start error: %v", err)
} }
select {} //wait indefinitely select {} //wait indefinitely