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
import (
"flag"
"fmt"
"os"
"polaris/db"
"polaris/log"
@@ -8,6 +10,9 @@ import (
)
func main() {
port := flag.Int("port", 8080, "port to listen on")
flag.Parse()
if os.Getenv("GIN_MODE") == "release" {
log.InitLogger(true)
}
@@ -19,7 +24,7 @@ func main() {
}
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)
}
select {} //wait indefinitely