Skip to content

Commit a01081a

Browse files
committed
refactor: 移除旧的命令处理文件,整合初始化逻辑到 main.go
1 parent 39028a9 commit a01081a

File tree

6 files changed

+45
-84
lines changed

6 files changed

+45
-84
lines changed

cmd/flag.go

Lines changed: 0 additions & 8 deletions
This file was deleted.

cmd/init.go

Lines changed: 0 additions & 13 deletions
This file was deleted.

cmd/logo.go

Lines changed: 0 additions & 21 deletions
This file was deleted.

cmd/setup.go

Lines changed: 0 additions & 25 deletions
This file was deleted.

cmd/shutdown.go

Lines changed: 0 additions & 12 deletions
This file was deleted.

main.go

Lines changed: 45 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,51 @@
11
package main
22

3-
import "MediaWarp/cmd"
3+
import (
4+
"MediaWarp/constants"
5+
"MediaWarp/internal/config"
6+
"MediaWarp/internal/logger"
7+
"MediaWarp/internal/router"
8+
"MediaWarp/pkg"
9+
"flag"
10+
"fmt"
11+
12+
"github.com/gin-gonic/gin"
13+
"github.com/sirupsen/logrus"
14+
)
15+
16+
var (
17+
isDebug bool
18+
)
19+
20+
func init() {
21+
printLOGO()
22+
flag.BoolVar(&isDebug, "debug", false, "是否启用调试模式")
23+
flag.Parse()
24+
}
425

526
func main() {
6-
cmd.PrintLOGO()
7-
cmd.InitFlag()
27+
if isDebug {
28+
logger.SetLevel(logrus.DebugLevel)
29+
logger.Warning("已启用调试模式")
30+
} else {
31+
gin.SetMode(gin.ReleaseMode)
32+
}
33+
34+
logger.Info("MediaWarp 监听端口:", config.Port)
35+
ginR := router.InitRouter() // 路由初始化
36+
logger.Info("MediaWarp 启动成功")
37+
ginR.Run(config.ListenAddr()) // 启动服务
38+
}
839

9-
defer cmd.ShutDown()
10-
cmd.SetUP()
40+
// 打印LOGO
41+
func printLOGO() {
42+
fmt.Print(
43+
constants.LOGO,
44+
pkg.Center(
45+
fmt.Sprintf(" MediaWarp %s 启动中 ", config.Version()),
46+
75,
47+
"=",
48+
),
49+
"\n\n",
50+
)
1151
}

0 commit comments

Comments
 (0)