cn-coach-spider/cmd/main.go
timerzz d55cdfc0d5
All checks were successful
Build image / build (push) Successful in 1m8s
feat 支持coach和outlet
2024-06-16 11:08:28 +08:00

73 lines
1.9 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

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)
}
}