package main

import (
	"fmt"
	"time"

	"github.com/pingcap/dbaas/billing2/handler/cloudmarketplacehandler"
	"github.com/pingcap/dbaas/billing2/handler/cloudmarketplacehandler/cloudimpl"

	"github.com/pingcap/dbaas/billing2/handler/paymenthandler"
	"github.com/pingcap/dbaas/billing2/handler/paymenthandler/impl"
	"github.com/pingcap/dbaas/billing2/handler/ratehandler"

	"github.com/gin-contrib/pprof"
	"github.com/gin-gonic/gin"
	"github.com/pingcap/dbaas/billing2/api"
	"github.com/pingcap/dbaas/billing2/config"
	"github.com/pingcap/dbaas/billing2/db"
	"github.com/pingcap/dbaas/billing2/handler/cachehandler"
	"github.com/pingcap/dbaas/billing2/handler/eventbushandler"
	"github.com/pingcap/dbaas/billing2/handler/gcphandler/auth"
	"github.com/pingcap/dbaas/billing2/handler/sqsmeasurehandler"
	"github.com/pingcap/dbaas/billing2/migrate"
	"github.com/pingcap/dbaas/billing2/utils/log"
	"github.com/pingcap/dbaas/billing2/utils/memlimit"
	"github.com/pingcap/dbaas/billing2/utils/panicIf"
	"github.com/pingcap/dbaas/billing2/utils/sentry"
	"github.com/pingcap/dbaas/billing2/workers"

	"go.uber.org/zap"
)

func main() {
	gin.DisableConsoleColor()
	panicIf.Error(config.Init())
	memlimit.SetupFromCgroup()
	// gin.SetMode(gin.ReleaseMode)
	_, err := db.Initialize()
	panicIf.Error(err)
	panicIf.Error(api.RegisterEventHandler())
	panicIf.Error(migrate.MigrateDB(db.GetDB(), true))
	cachehandler.Init()
	ratehandler.Init()
	panicIf.Error(auth.InitWebIdentityEnv())
	panicIf.Error(sentry.SetupLoggerAndSentry())
	engine, err := api.NewWorkerService()
	panicIf.Error(err)
	pprof.Register(engine)
	panicIf.Error(eventbushandler.Get().Run())
	panicIf.Error(sqsmeasurehandler.SetDefaultMeasureSender())
	paymenthandler.SetPaymentHandler(&impl.PaymentHandlerImpl{})
	cloudmarketplacehandler.SetMarketplaceHandler(&cloudimpl.MarketplaceHandlerImpl{})
	workers.RunWorker(15 * time.Second)

	log.Fatal("start billing worker failed", zap.Error(engine.Run(fmt.Sprintf(":%d", config.Get().GetInt(config.BillingServerPort)))))
}