🎨 flag params binding

main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"schisandra-cloud-album/cmd"
 	"schisandra-cloud-album/core"
 	"schisandra-cloud-album/global"
 	"schisandra-cloud-album/router"
@@ -11,11 +12,17 @@ func main() {
 	core.InitConfig()
 	core.InitLogger()
 	core.InitGorm()
+	// 命令行参数绑定
+	option := cmd.Parse()
+	if cmd.IsStopWeb(&option) {
+		cmd.SwitchOption(&option)
+		return
+	}
 	r := router.InitRouter()
 	addr := global.CONFIG.System.Addr()
 	global.LOG.Info("Server run on ", addr)
 	err := r.Run(addr)
 	if err != nil {
-		return
+		global.LOG.Fatalf(err.Error())
 	}
 }