79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
package cmd
|
||
|
||
import (
|
||
"ai-css/middleware"
|
||
"ai-css/router"
|
||
"ai-css/tools"
|
||
"ai-css/ws"
|
||
"fmt"
|
||
"log"
|
||
"os"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/spf13/cobra"
|
||
"github.com/zh-five/xdaemon"
|
||
)
|
||
|
||
var (
|
||
port string
|
||
daemon bool
|
||
)
|
||
var serverCmd = &cobra.Command{
|
||
Use: "server",
|
||
Short: "启动http服务",
|
||
Example: "gofly server -p 8082",
|
||
Run: func(cmd *cobra.Command, args []string) {
|
||
run()
|
||
},
|
||
}
|
||
|
||
func init() {
|
||
serverCmd.PersistentFlags().StringVarP(&port, "port", "p", "8081", "监听端口号")
|
||
serverCmd.PersistentFlags().BoolVarP(&daemon, "daemon", "d", false, "是否为守护进程模式")
|
||
}
|
||
func run() {
|
||
if daemon == true {
|
||
logFilePath := ""
|
||
if dir, err := os.Getwd(); err == nil {
|
||
logFilePath = dir + "/logs/"
|
||
}
|
||
_, err := os.Stat(logFilePath)
|
||
if os.IsNotExist(err) {
|
||
if err := os.MkdirAll(logFilePath, 0777); err != nil {
|
||
log.Println(err.Error())
|
||
}
|
||
}
|
||
d := xdaemon.NewDaemon(logFilePath + "go-fly.log")
|
||
d.MaxCount = 10
|
||
d.Run()
|
||
}
|
||
|
||
baseServer := "0.0.0.0:" + port
|
||
log.Println("start server...\r\ngo：http://" + baseServer)
|
||
tools.Logger().Println("start server...\r\ngo：http://" + baseServer)
|
||
|
||
engine := gin.Default()
|
||
engine.LoadHTMLGlob("static/templates/*")
|
||
engine.Static("/assets", "./static")
|
||
engine.Static("/static", "./static")
|
||
engine.Use(tools.Session("gofly"))
|
||
engine.Use(middleware.CrossSite)
|
||
//性能监控
|
||
//pprof.Register(engine)
|
||
|
||
//记录日志
|
||
engine.Use(middleware.NewMidLogger())
|
||
router.InitViewRouter(engine)
|
||
router.InitApiRouter(engine)
|
||
//记录pid
|
||
os.WriteFile("gofly.sock", []byte(fmt.Sprintf("%d,%d", os.Getppid(), os.Getpid())), 0666)
|
||
//限流类
|
||
tools.NewLimitQueue()
|
||
//清理
|
||
ws.CleanVisitorExpire()
|
||
//后端websocket
|
||
go ws.WsServerBackend()
|
||
|
||
engine.Run(baseServer)
|
||
}
|