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.Info, 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) } }