package customer

import (
	"context"
	"net/http"
	"server/call"
	"server/config"
	"server/db"
	edb "server/db/es"
	mdb "server/db/mysql"
	rdb "server/db/redis"
	"server/modules/customer/bdb"
	"server/modules/customer/routers"
	"server/util"
	"time"

	"github.com/liangdas/mqant/conf"
	"github.com/liangdas/mqant/log"
	"github.com/liangdas/mqant/module"
	basemodule "github.com/liangdas/mqant/module/base"
)

var (
	Module = func() module.Module {
		this := new(Customer)
		return this
	}
	BackDB = new(mdb.MysqlClient)
)

type Customer struct {
	basemodule.BaseModule
	httpSvr *http.Server
	addr    string
}

func (b *Customer) GetType() string {
	// 很关键,需要与配置文件中的Module配置对应
	return "customer"
}
func (b *Customer) Version() string {
	// 可以在监控时了解代码版本
	return "1.0.0"
}
func (b *Customer) OnInit(app module.App, settings *conf.ModuleSettings) {
	b.BaseModule.OnInit(b, app, settings)

	call.NewCaller(b)

	db.InitDB(&mdb.MysqlClient{}, &rdb.RedisClient{}, &edb.EsClient{})

	bdb.InitMysql()

	// 自动初始化后台数据库
	bdb.MigrateDB()

	// 加载配置
	util.Go(func() {
		loadConfig()
	})

	b.addr = config.GetConfig().Customer.Addr

	call.NewSnowflake(int64(config.GetConfig().WorkID))

	call.InitReload(b.App.Transport())
}

func (b *Customer) startHttpServer() {
	router := routers.SetUpRouter()
	srv := &http.Server{
		Addr:    b.addr,
		Handler: router,
	}

	go func() {
		if err := srv.ListenAndServe(); err != nil {
			panic(err)
		}
	}()
	// returning reference so caller can call Shutdown()
	b.httpSvr = srv
}

func (b *Customer) Run(closeSig chan bool) {
	log.Info("customer: starting HTTP server :%s", b.addr)
	call.InitTimeWheel(closeSig)
	call.InitWarn(b.App.Transport())
	b.startHttpServer()
	<-closeSig
	log.Info("customer: stopping HTTP server")

}

func (b *Customer) OnDestroy() {
	// The context is used to inform the server it has 5 seconds to finish
	// the request it is currently handling
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()

	// now close the server gracefully ("shutdown")
	// timeout could be given instead of nil as a https://golang.org/pkg/context/
	if err := b.httpSvr.Shutdown(ctx); err != nil {
		log.Error("OnDestroy Customer Shutdown error:%v", err)
	}

	log.Info("Customer: done. exiting")

	// 一定别忘了继承
	b.BaseModule.OnDestroy()

	log.Info("Customer 模块已销毁")
}