overflow_gateway_websocket/server/server_handlers.go

package server

import (
	"fmt"
	"net"
	"time"

	"git.loafle.net/commons_go/logging"
	cwf "git.loafle.net/commons_go/websocket_fasthttp"
	"git.loafle.net/overflow/overflow_gateway_websocket/config"
	"git.loafle.net/overflow/overflow_gateway_websocket/external"
	"git.loafle.net/overflow/overflow_gateway_websocket/external/redis"
	oogwisr "git.loafle.net/overflow/overflow_gateway_websocket/internal/server/rpc"
	"git.loafle.net/overflow/overflow_gateway_websocket/rpc"
	"git.loafle.net/overflow/overflow_gateway_websocket/servlet"
	oos "git.loafle.net/overflow/overflow_subscriber"
	oosr "git.loafle.net/overflow/overflow_subscriber/redis"
	"github.com/valyala/fasthttp"
)

type ServerHandlers struct {
	cwf.ServerHandlers

	redisSubscriber oos.Subscriber
	subscribers     []oos.SubscriberHandler
}

// Init invoked before the server is started
// If you override ths method, must call
func (sh *ServerHandlers) Init() error {
	if err := sh.ServerHandlers.Init(); nil != err {
		return err
	}

	external.ExternalInit()

	return nil
}

func (sh *ServerHandlers) Listen() (net.Listener, error) {
	return net.Listen(config.Config.Server.Network, config.Config.Server.Addr)
}

func (sh *ServerHandlers) OnStart() {
	sh.ServerHandlers.OnStart()

	sh.redisSubscriber = oosr.New(redis.RedisPool.Get())
	if err := sh.redisSubscriber.Start(); nil != err {
		logging.Logger().Panic(fmt.Sprintf("Gateway Websocket: Redist Subscriber did not start %v", err))
		return
	}

	for _, subscriber := range sh.subscribers {
		sh.redisSubscriber.Subscribe(subscriber)
	}
}

func (sh *ServerHandlers) OnError(ctx *fasthttp.RequestCtx, status int, reason error) {
	sh.OnError(ctx, status, reason)
}

func (sh *ServerHandlers) OnStop() {
	sh.redisSubscriber.Stop()

	external.ExternalDestroy()

	sh.ServerHandlers.OnStop()
}

func (sh *ServerHandlers) RegisterServlet(servletName string, servletHandler servlet.ServletHandler) {
	cfg := config.Config.Servlets[servletName]
	if nil == cfg {
		logging.Logger().Panic(fmt.Sprintf("Gateway Websocket: config of servlet[%s] is not exist", servletName))
		return
	}

	servletH := servletHandler.(*servlet.ServletHandlers)

	servletH.MaxMessageSize = cfg.Socket.MaxMessageSize
	servletH.WriteTimeout = cfg.Socket.WriteTimeout * time.Second
	servletH.ReadTimeout = cfg.Socket.ReadTimeout * time.Second
	servletH.PongTimeout = cfg.Socket.PongTimeout * time.Second
	servletH.PingTimeout = cfg.Socket.PingTimeout * time.Second
	servletH.PingPeriod = cfg.Socket.PingPeriod * time.Second
	// servletH.BinaryMessage = cfg.Socket.BinaryMessage

	sh.RegisterSocketHandler(cfg.Entry, servletH)

}

func (sh *ServerHandlers) RegisterRPCGatewayServlet(servletName string, rpcGatewayHandler rpc.RPCGatewayHandler) servlet.ServletHandler {
	cfg := config.Config.Servlets[servletName]
	if nil == cfg {
		logging.Logger().Panic(fmt.Sprintf("Gateway Websocket: config of servlet[%s] is not exist", servletName))
		return nil
	}

	rpcSH := oogwisr.NewRPCGatewayServletHandler(rpcGatewayHandler)
	rpcSHs := rpcSH.(*oogwisr.RPCGatewayServletHandlers)

	rpcSHs.MaxMessageSize = cfg.Socket.MaxMessageSize
	rpcSHs.WriteTimeout = cfg.Socket.WriteTimeout * time.Second
	rpcSHs.ReadTimeout = cfg.Socket.ReadTimeout * time.Second
	rpcSHs.PongTimeout = cfg.Socket.PongTimeout * time.Second
	rpcSHs.PingTimeout = cfg.Socket.PingTimeout * time.Second
	rpcSHs.PingPeriod = cfg.Socket.PingPeriod * time.Second

	sh.RegisterSocketHandler(cfg.Entry, rpcSHs)

	return rpcSH
}

func (sh *ServerHandlers) RegisterSubscriber(subscriberHandler oos.SubscriberHandler) {
	if nil == sh.subscribers {
		sh.subscribers = make([]oos.SubscriberHandler, 0)
	}
	sh.subscribers = append(sh.subscribers, subscriberHandler)
}

func (sh *ServerHandlers) Validate() {
	sh.ServerHandlers.Validate()
}