86 lines
1.9 KiB
Go
86 lines
1.9 KiB
Go
package cmd
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/sirupsen/logrus"
|
|
"hy2xs-admin/dao"
|
|
"hy2xs-admin/middleware"
|
|
"hy2xs-admin/model/constant"
|
|
"hy2xs-admin/router"
|
|
"hy2xs-admin/service"
|
|
"hy2xs-admin/util"
|
|
"net/http"
|
|
"os"
|
|
)
|
|
|
|
func runServer(port string) error {
|
|
defer releaseResource()
|
|
|
|
middleware.InitLog()
|
|
service.InitForward()
|
|
if err := initFile(); err != nil {
|
|
return err
|
|
}
|
|
if err := dao.InitSql(port); err != nil {
|
|
return err
|
|
}
|
|
if err := middleware.InitCron(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.InitHysteria2(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.InitTableAndChain(); err != nil {
|
|
logrus.Errorf(err.Error())
|
|
}
|
|
if err := service.InitPortHopping(); err != nil {
|
|
logrus.Errorf(err.Error())
|
|
}
|
|
config, err := dao.GetConfig("key = ?", constant.HUIWebContext)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
r := gin.Default()
|
|
router.Router(r, config.Value)
|
|
|
|
serverPort, crtPath, keyPath, err := service.GetServerPortAndCert()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
service.InitServer(fmt.Sprintf(":%d", serverPort), r)
|
|
if err := service.StartServer(crtPath, keyPath); err != nil && err != http.ErrServerClosed {
|
|
logrus.Errorf("start server err: %v", err)
|
|
return errors.New("start server err")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func releaseResource() {
|
|
if err := dao.CloseSqliteDB(); err != nil {
|
|
logrus.Errorf(err.Error())
|
|
}
|
|
if err := service.ReleaseHysteria2(); err != nil {
|
|
logrus.Errorf(err.Error())
|
|
}
|
|
if err := service.RemoveByComment(); err != nil {
|
|
logrus.Errorf(err.Error())
|
|
}
|
|
}
|
|
|
|
func initFile() error {
|
|
var dirs = []string{constant.LogDir, constant.SqliteDBDir, constant.BinDir, constant.ExportPathDir}
|
|
for _, item := range dirs {
|
|
if !util.Exists(item) {
|
|
if err := os.Mkdir(item, os.ModePerm); err != nil {
|
|
logrus.Errorf("%s create err: %v", item, err)
|
|
return errors.New(fmt.Sprintf("%s create err", item))
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|