ecs/servers/admin/server.go

package main

import (
	"ecs/servers/admin/handler"
	"github.com/oylshe1314/framework/client/rpc"
	"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 ServerConfig struct {
	Subscriptions []string `json:"Subscriptions"`
}

type adminServer struct {
	server.NamedServer

	serverConfig *ServerConfig

	ExterServer server.HttpServer

	sdConfig        *sd.Config
	HttpRpcClient   rpc.HttpRpcClient
	SubscribeClient sd.SubscribeClient

	innerHandler *handler.InnerHandler
}

func NewAdminServer() server.Server {
	return &adminServer{}
}

func (this *adminServer) WithServerConfig(serverConfig *ServerConfig) {
	this.serverConfig = serverConfig
}

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

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

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

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

	var subscriptions = map[string]sd.SubscribeCallback{}
	for _, svrName := range this.serverConfig.Subscriptions {
		subscriptions[svrName] = this.HttpRpcClient.SubscribeCallback
	}
	this.SubscribeClient = zk.NewSubscribeClient(this, this.sdConfig, subscriptions)
	err = this.SubscribeClient.Init()
	if err != nil {
		return errors.Errorf("the 'SubscribeClient' init failed, %v", err)
	}

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

	//this.ExterServer.PostHandler("/server/detect", this.innerHandler.ServerDetect)
	//
	//this.ExterServer.GetHandler("/gate/setting/get", this.innerHandler.GateSettingGet)
	//this.ExterServer.PostHandler("/gate/setting/save", this.innerHandler.GateSettingSave)
	//
	//this.ExterServer.GetHandler("/gate/cdkey/get", this.innerHandler.GateCdkeyGet)
	//this.ExterServer.GetHandler("/gate/cdkey/list", this.innerHandler.GateCdkeyList)
	//this.ExterServer.PostHandler("/gate/cdkey/add", this.innerHandler.GateCdkeyAdd)
	//this.ExterServer.PostHandler("/gate/cdkey/generate", this.innerHandler.GateCdkeyGenerate)
	//this.ExterServer.PostHandler("/gate/cdkey/delete", this.innerHandler.GateCdkeyDelete)
	//
	//this.ExterServer.GetHandler("/gate/user/query", this.innerHandler.GateUserQuery)
	//this.ExterServer.PostHandler("/gate/user/operate", this.innerHandler.GateUserOperate)
	//
	//this.ExterServer.PostHandler("/game_1/data/reload", this.innerHandler.GameDataReload)
	//
	//this.ExterServer.GetHandler("/game_1/mail/list", this.innerHandler.GameMailList)
	//this.ExterServer.PostHandler("/game_1/mail/send", this.innerHandler.GameMailSend)
	//this.ExterServer.PostHandler("/game_1/mail/delete", this.innerHandler.GameMailDelete)
	//
	//this.ExterServer.GetHandler("/game_1/player/query", this.innerHandler.GamePlayerQuery)
	//this.ExterServer.PostHandler("/game_1/player/operate", this.innerHandler.GamePlayerOperate)
	return nil
}

func (this *adminServer) Serve() error {
	return util.WaitAny(this.ExterServer.Serve, this.SubscribeClient.Work)
}

func (this *adminServer) Close() error {

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

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

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

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