142 lines
3.1 KiB
Go
142 lines
3.1 KiB
Go
|
//go:build !windows
|
|||
|
|
|
|||
|
|
package main
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"errors"
|
|||
|
|
"flag"
|
|||
|
|
"fmt"
|
|||
|
|
"io/fs"
|
|||
|
|
"log/slog"
|
|||
|
|
"net/http"
|
|||
|
|
"os"
|
|||
|
|
"os/signal"
|
|||
|
|
"path/filepath"
|
|||
|
|
"syscall"
|
|||
|
|
|
|||
|
|
"bj_power_mes/internal/handler"
|
|||
|
|
"bj_power_mes/internal/middleware"
|
|||
|
|
|
|||
|
|
xhttp "bj_power_mes/common/httpx"
|
|||
|
|
xlog "bj_power_mes/common/logx"
|
|||
|
|
"bj_power_mes/internal/config"
|
|||
|
|
"bj_power_mes/internal/svc"
|
|||
|
|
|
|||
|
|
"github.com/dromara/carbon/v2"
|
|||
|
|
"github.com/pkg/browser"
|
|||
|
|
"github.com/zeromicro/go-zero/core/conf"
|
|||
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|||
|
|
"github.com/zeromicro/go-zero/rest"
|
|||
|
|
"github.com/zeromicro/go-zero/rest/httpx"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// macOS/Linux 无互斥锁支持，直接返回成功
|
|||
|
|
var ErrAlreadyExists = errors.New("already exists")
|
|||
|
|
|
|||
|
|
const AppName = "bj_power_mes"
|
|||
|
|
|
|||
|
|
var configFile = flag.String("f", "etc/bj_power_mes-api.yaml", "the config file")
|
|||
|
|
|
|||
|
|
var AppPath, _ = filepath.Abs(os.Args[0])
|
|||
|
|
|
|||
|
|
var baseUrl string
|
|||
|
|
|
|||
|
|
func main() {
|
|||
|
|
flag.Parse()
|
|||
|
|
|
|||
|
|
// 业务影响：所有时间显示使用北京时间
|
|||
|
|
carbon.SetDefault(carbon.Default{
|
|||
|
|
Layout: carbon.DateTimeLayout,
|
|||
|
|
Timezone: carbon.PRC,
|
|||
|
|
Locale: "zh-CN",
|
|||
|
|
WeekStartsAt: carbon.Monday,
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
var c config.Config
|
|||
|
|
conf.MustLoad(*configFile, &c)
|
|||
|
|
baseUrl = fmt.Sprintf("http://127.0.0.1:%d", c.Port)
|
|||
|
|
|
|||
|
|
logger := xlog.NewLogger(c.Logger)
|
|||
|
|
xlog.SetDefault(logger)
|
|||
|
|
logx.SetWriter(logger)
|
|||
|
|
|
|||
|
|
// 业务影响：初始化数据库连接、PLC 连接
|
|||
|
|
svcCtx := svc.NewServiceContext(c)
|
|||
|
|
|
|||
|
|
var publicFS http.FileSystem
|
|||
|
|
if _, err := os.Stat("public"); err == nil {
|
|||
|
|
publicFS = http.Dir("public")
|
|||
|
|
} else {
|
|||
|
|
sub, _ := fs.Sub(public, "public")
|
|||
|
|
publicFS = http.FS(sub)
|
|||
|
|
}
|
|||
|
|
spaHandler := xhttp.NewNotFoundHandler(publicFS)
|
|||
|
|
|
|||
|
|
server := rest.MustNewServer(
|
|||
|
|
c.RestConf,
|
|||
|
|
rest.WithCors("*"),
|
|||
|
|
rest.WithNotFoundHandler(spaHandler),
|
|||
|
|
)
|
|||
|
|
logx.DisableStat()
|
|||
|
|
logx.SetWriter(logger)
|
|||
|
|
defer server.Stop()
|
|||
|
|
|
|||
|
|
handler.RegisterHandlers(server, svcCtx)
|
|||
|
|
|
|||
|
|
// 写操作审计中间件（兜底覆盖所有写接口）
|
|||
|
|
server.Use(middleware.WriteAudit(svcCtx))
|
|||
|
|
|
|||
|
|
// 业务影响：统一错误响应格式，前端不用处理多种错误格式
|
|||
|
|
httpx.SetValidator(xhttp.NewValidator())
|
|||
|
|
httpx.SetOkHandler(xhttp.OkJsonCtx)
|
|||
|
|
httpx.SetErrorHandlerCtx(xhttp.ErrorCtx)
|
|||
|
|
|
|||
|
|
fmt.Printf("Starting server at %s:%d...\n", c.Host, c.Port)
|
|||
|
|
slog.Info(fmt.Sprintf("Starting server at %s:%d...", c.Host, c.Port))
|
|||
|
|
|
|||
|
|
go server.Start()
|
|||
|
|
|
|||
|
|
_ = browser.OpenURL(baseUrl)
|
|||
|
|
|
|||
|
|
// 阻塞等待退出信号（Ctrl+C / kill）
|
|||
|
|
signalChan := make(chan os.Signal, 1)
|
|||
|
|
signal.Notify(signalChan, syscall.SIGINT, syscall.SIGTERM)
|
|||
|
|
<-signalChan
|
|||
|
|
|
|||
|
|
slog.Info("shutting down server...")
|
|||
|
|
server.Stop()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// macOS/Linux 下 Windows 特有功能的空实现
|
|||
|
|
|
|||
|
|
var mutex uintptr
|
|||
|
|
|
|||
|
|
func createMutex(name string) (uintptr, error) {
|
|||
|
|
return 1, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var procCloseHandle = closeHandleHelper{}
|
|||
|
|
|
|||
|
|
type closeHandleHelper struct{}
|
|||
|
|
|
|||
|
|
func (closeHandleHelper) Call(uintptr) uintptr {
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func MessageBox(title, message string) {
|
|||
|
|
fmt.Printf("[Dialog] %s: %s\n", title, message)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func setAutoStart(name, path string) error {
|
|||
|
|
slog.Info("setAutoStart not supported on macOS")
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func removeAutoStart(name string) error {
|
|||
|
|
slog.Info("removeAutoStart not supported on macOS")
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func isAutoStartEnabled(name string) bool {
|
|||
|
|
return false
|
|||
|
|
}
|