78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"path/filepath"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/joho/godotenv"
|
|
|
|
"everypublish/server/internal/api"
|
|
"everypublish/server/internal/cache"
|
|
"everypublish/server/internal/config"
|
|
"everypublish/server/internal/db"
|
|
"everypublish/server/internal/models"
|
|
"everypublish/server/internal/ws"
|
|
)
|
|
|
|
func main() {
|
|
_ = godotenv.Load() // 本地开发读 .env，生产用环境变量
|
|
cfg := config.Load()
|
|
|
|
gdb, err := db.Connect(cfg)
|
|
if err != nil {
|
|
log.Fatalf("db connect failed: %v", err)
|
|
}
|
|
if err = db.AutoMigrate(gdb); err != nil {
|
|
log.Fatalf("auto migrate failed: %v", err)
|
|
}
|
|
// 启动自愈：上次异常退出残留的 online 状态重置为 offline（真实状态以连接为准）
|
|
_ = gdb.Model(&models.AgentDevice{}).Where("status = ?", "online").Update("status", "offline").Error
|
|
log.Println("mysql connected & migrated")
|
|
|
|
rds := cache.New(cfg)
|
|
if err = rds.Ping(); err != nil {
|
|
log.Fatalf("redis ping failed: %v", err)
|
|
}
|
|
log.Println("redis connected")
|
|
|
|
serverKey, err := ws.LoadOrCreateServerKey("./data/server_ed25519.key")
|
|
if err != nil {
|
|
log.Fatalf("server ed25519 key failed: %v", err)
|
|
}
|
|
hub := ws.NewHub()
|
|
dsp := ws.NewDispatcher(gdb, hub, cfg)
|
|
|
|
ctx, stopFn := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
|
|
defer stopFn()
|
|
go dsp.Start(ctx.Done())
|
|
|
|
router := api.Router(cfg, gdb, rds, hub, serverKey, dsp)
|
|
|
|
// 托管前端生产构建（STATIC_DIR 存在 index.html 时启用），SPA 回退
|
|
if cfg.StaticDir != "" {
|
|
if _, err := os.Stat(filepath.Join(cfg.StaticDir, "index.html")); err == nil {
|
|
router.Static("/assets", filepath.Join(cfg.StaticDir, "assets"))
|
|
router.StaticFile("/favicon.ico", filepath.Join(cfg.StaticDir, "favicon.ico"))
|
|
router.NoRoute(func(c *gin.Context) {
|
|
if strings.HasPrefix(c.Request.URL.Path, "/api/") {
|
|
c.JSON(404, gin.H{"code": 404, "message": "接口不存在"})
|
|
return
|
|
}
|
|
c.File(filepath.Join(cfg.StaticDir, "index.html"))
|
|
})
|
|
log.Printf("serving frontend from %s", cfg.StaticDir)
|
|
}
|
|
}
|
|
|
|
log.Printf("everypublish-server listening on %s", cfg.ServerAddr)
|
|
if err = router.Run(cfg.ServerAddr); err != nil {
|
|
log.Fatalf("server exited: %v", err)
|
|
}
|
|
}
|