ecs/servers/center/server.go

package main

import (
	"ecs/servers/center/handler"
	"ecs/servers/center/logic"
	"github.com/oylshe1314/framework/client/sd"
	"github.com/oylshe1314/framework/client/sd/zk"
	"github.com/oylshe1314/framework/errors"
	"github.com/oylshe1314/framework/server"
	"github.com/oylshe1314/framework/util"
	"time"
)

type centerServer struct {
	server.NamedServer

	InnerServer server.HttpServer

	sdConfig       *sd.Config
	RegisterClient sd.RegisterClient

	centerManager *logic.CenterManager

	innerHandler *handler.InnerHandler
}

func NewCenterServer() server.Server {
	return &centerServer{}
}

func (this *centerServer) WithSdConfig(config *sd.Config) {
	this.sdConfig = config
}

func (this *centerServer) Init() (err error) {
	err = this.NamedServer.Init()
	if err != nil {
		return err
	}

	this.InnerServer.SetServer(this)
	err = this.InnerServer.Init()
	if err != nil {
		return errors.Errorf("the 'InnerServer' init failed, %v", err)
	}

	this.RegisterClient = zk.NewRegisterClient(this, this.sdConfig, &this.InnerServer.Listener, nil)
	err = this.RegisterClient.Init()
	if err != nil {
		return errors.Errorf("the 'RegisterClient' init failed, %v", err)
	}

	this.centerManager = logic.NewCenterManager(this)

	err = this.centerManager.Init()
	if err != nil {
		return errors.Errorf("the 'centerManager' init failed, %v", err)
	}

	this.innerHandler = handler.NewInnerHandler(this, this.centerManager)

	this.InnerServer.GetHandler("/wechat/accessToken/get", this.innerHandler.GetWechatAccessToken)
	this.InnerServer.PostHandler("/wechat/accessToken/refresh", this.innerHandler.RefreshWechatAccessToken)

	return nil
}

func (this *centerServer) Serve() error {
	return util.WaitAny(this.RegisterClient.Work, this.InnerServer.Serve)
}

func (this *centerServer) Close() error {

	this.Logger().Info("The 'SubscribeClient' is closing")
	_ = this.RegisterClient.Close()

	this.Logger().Info("The 'InnerServer' is closing")
	_ = this.InnerServer.Close()

	this.Logger().Info("Server closed")

	time.Sleep(time.Second)
	_ = this.NamedServer.Close()
	return nil
}