73 lines
1.9 KiB
Go
73 lines
1.9 KiB
Go
package main
|
||
|
||
import (
|
||
"context"
|
||
"gitea.timerzz.com/kedaya_haitao/cn-coach-spider/pkg/options"
|
||
"gitea.timerzz.com/kedaya_haitao/cn-coach-spider/server"
|
||
"gitea.timerzz.com/kedaya_haitao/cn-coach-spider/spider"
|
||
coach_client "gitea.timerzz.com/kedaya_haitao/common/pkg/coach-client"
|
||
"gitea.timerzz.com/kedaya_haitao/common/pkg/database"
|
||
"github.com/gofiber/fiber/v3"
|
||
"github.com/gofiber/fiber/v3/middleware/cors"
|
||
"github.com/golang/glog"
|
||
"gorm.io/gorm/logger"
|
||
"log"
|
||
"log/slog"
|
||
"os"
|
||
"os/signal"
|
||
"time"
|
||
)
|
||
|
||
func main() {
|
||
slog.Info(">>> BEGIN INIT<<<")
|
||
ctx, cancel := signal.NotifyContext(context.Background(), os.Kill, os.Interrupt)
|
||
defer cancel()
|
||
cfg, err := options.LoadConfig()
|
||
if err != nil {
|
||
glog.Fatalf("获取配置失败：%v", err)
|
||
}
|
||
db, err := database.InitDatabase(&cfg.DB)
|
||
if err != nil {
|
||
glog.Fatalf("初始化数据库失败：%v", err)
|
||
}
|
||
db.Logger = logger.New(
|
||
log.New(os.Stdout, "\r\n", log.LstdFlags),
|
||
logger.Config{
|
||
SlowThreshold: time.Second,
|
||
LogLevel: logger.Warn,
|
||
IgnoreRecordNotFoundError: false,
|
||
Colorful: true,
|
||
},
|
||
)
|
||
shopType := os.Getenv("SHOP_TYPE")
|
||
if shopType != options.ShopType_coach && shopType != options.ShopType_outlet {
|
||
glog.Fatalf("SHOP_TYPE环境变量错误，请设置SHOP_TYPE=coach或SHOP_TYPE=outlet")
|
||
}
|
||
// coach client
|
||
var xMac = coach_client.XMac_Coach
|
||
if shopType == options.ShopType_outlet {
|
||
xMac = coach_client.XMac_Outlet
|
||
}
|
||
client, err := coach_client.CNClient(xMac)
|
||
if err != nil {
|
||
glog.Fatalf("初始化coach client失败：%v", err)
|
||
}
|
||
|
||
spiderCtl := spider.NewController(client, db, cfg.Interval, shopType)
|
||
|
||
go spiderCtl.Run(ctx)
|
||
|
||
r := fiber.New()
|
||
r.Use(cors.New())
|
||
|
||
r.Get("/health", func(ctx fiber.Ctx) error {
|
||
return ctx.SendString("ok")
|
||
})
|
||
api := r.Group("/api/v1")
|
||
|
||
server.NewSpiderSvc(spiderCtl, shopType).RegistryRouter(api)
|
||
if err = r.Listen(":8080"); err != nil {
|
||
glog.Warningf("server over: %v", err)
|
||
}
|
||
}
|