mirror of
https://github.com/simon-ding/polaris.git
synced 2026-06-03 08:27:41 +08:00
feat: add port flag
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user