overflow_probes/auth/client/socket_handlers.go

package client

import (
	"fmt"
	"net/http"

	"git.loafle.net/commons_go/logging"
	cwfc "git.loafle.net/commons_go/websocket_fasthttp/client"
	ooccn "git.loafle.net/overflow/overflow_commons_go/config/noauthprobe"
	oocmn "git.loafle.net/overflow/overflow_commons_go/modules/noauthprobe"
	oopar "git.loafle.net/overflow/overflow_probes/auth/rpc"
)

type SocketHandlers struct {
	cwfc.SocketHandlers

	napService *oopar.NoAuthProbeService
}

func (sh *SocketHandlers) OnConnect(socketContext cwfc.SocketContext, res *http.Response) {
	logging.Logger().Info(fmt.Sprintf("Auth: client has been connected res[%v]", res))

	switch sh.napService.Config.State() {
	case ooccn.NoAuthProbeStateTypeNotRegisterd:
		tempProbeKey := res.Header.Get(oocmn.HTTPResponseHeaderKey_NoAuthProbe_SetTempProbeKey)
		sh.napService.SetTempProbeKey(tempProbeKey)
	case ooccn.NoAuthProbeStateTypeRegisterd:
	}

}

func (sh *SocketHandlers) OnDisconnect(soc cwfc.Socket) {
	logging.Logger().Info(fmt.Sprintf("Auth: client has been disconnected soc[%v]", soc))

}

func newSocketHandler(napService *oopar.NoAuthProbeService) cwfc.SocketHandler {
	return &SocketHandlers{
		napService: napService,
	}
}