deprecated_overflow_gateway_app/server/server_handlers.go

package server

import (
	"fmt"
	"net"

	"git.loafle.net/commons_go/logging"
	cwf "git.loafle.net/commons_go/websocket_fasthttp"
	"git.loafle.net/overflow/overflow_gateway_app/conf"
	"git.loafle.net/overflow/overflow_gateway_app/external/redis"
	oos "git.loafle.net/overflow/overflow_subscriber"
	oosr "git.loafle.net/overflow/overflow_subscriber/redis"
	"github.com/valyala/fasthttp"
)

func newServerHandler() ServerHandler {
	sh := &ServerHandlers{}
	sh.Name = conf.Config.Server.Name
	sh.Concurrency = conf.Config.Server.Concurrency
	sh.MaxStopWaitTime = conf.Config.Server.MaxStopWaitTime

	sh.HandshakeTimeout = conf.Config.Websocket.HandshakeTimeout
	sh.ReadBufferSize = conf.Config.Websocket.ReadBufferSize
	sh.WriteBufferSize = conf.Config.Websocket.WriteBufferSize
	sh.EnableCompression = conf.Config.Websocket.EnableCompression

	return sh
}

type ServerHandler interface {
	cwf.ServerHandler

	RegisterSubscriberHandler(subscriberHandler oos.SubscriberHandler)
}

type ServerHandlers struct {
	cwf.ServerHandlers

	redisSubscriber oos.Subscriber
	subscribers     []oos.SubscriberHandler
}

func (sh *ServerHandlers) OnStart() {
	sh.redisSubscriber = oosr.New(redis.RedisPool.Get())
	if err := sh.redisSubscriber.Start(); nil != err {
		logging.Logger().Error(fmt.Sprintf("App: Redist Subscriber did not start %v", err))
		return
	}

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

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

func (sh *ServerHandlers) CheckOrigin(ctx *fasthttp.RequestCtx) bool {
	if origin := string(ctx.Request.Header.Peek("Origin")); origin != "" {
		ctx.Response.Header.Set("Access-Control-Allow-Origin", origin)
		if string(ctx.Method()) == "OPTIONS" && string(ctx.Request.Header.Peek("Access-Control-Request-Method")) != "" {
			ctx.Response.Header.Set("Access-Control-Allow-Headers", "Content-Type, Accept")
			ctx.Response.Header.Set("Access-Control-Allow-Methods", "GET, HEAD, POST, PUT, DELETE")
		}
	}
	return true
}

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

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

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