session_manager.go

/*
 *  Copyright (c) 2018, https://github.com/airwide-code
 *  All rights reserved.
 *
 *
 *
 */

package server

import (
	"encoding/binary"
	"github.com/airwide-code/airwide.datacenter/mtproto"
	"encoding/hex"
	"fmt"
	"github.com/golang/glog"
	"sync"
)

type sessionManager struct {
	sessions sync.Map // map[int64]*sessionClientList
}

func newSessionManager() *sessionManager {
	return &sessionManager{}
}

////////////////////////////////////////////////////////////////////////////////////////////////////////
func (s *sessionManager) onSessionData2(clientConnID, frontendConnID uint64, md *mtproto.ZProtoMetadata, buf []byte) error {
	if len(buf) > 10240 {
		glog.Infof("onSessionData: data: {session_id: %d, md: %v, buf_len: %d, buf: %s, buf_end: %s}",
			// conn.RemoteAddr(),
			clientConnID,
			md,
			len(buf),
			hex.EncodeToString(buf[:256]),
			hex.EncodeToString(buf[len(buf)-256:]))
	} else {
		glog.Infof("onSessionData: data: {session_id: %d, md: %v, buf_len: %d}", clientConnID, md, len(buf))
	}

	////
	authKeyId := int64(binary.LittleEndian.Uint64(buf))

	// TODO(@benqi): sync s.sessions
	var sessList *clientSessionManager
	if vv, ok := s.sessions.Load(authKeyId); !ok {
		authKey := getCacheAuthKey(authKeyId)
		if authKey == nil {
			err := fmt.Errorf("onSessionData - not found authKeyId: {%d}", authKeyId)
			glog.Error(err)
			return err
		}

		sessList = newClientSessionManager(authKeyId, authKey, 0)
		s.sessions.Store(authKeyId, sessList)
		s.onNewSessionClientManager(sessList)
	} else {
		sessList, _ = vv.(*clientSessionManager)
	}

	return sessList.OnSessionDataArrived(makeClientConnID(clientConnID, frontendConnID), md, buf)
}

func (s *sessionManager) onSyncData2(authKeyId, sessionId int64, md *mtproto.ZProtoMetadata, data *messageData) error {
	var sessList *clientSessionManager

	if vv, ok := s.sessions.Load(authKeyId); !ok {
		err := fmt.Errorf("pushToSessionData - not find sessionList by authKeyId: {%d}", authKeyId)
		glog.Warning(err)
		return err
	} else {
		sessList, _ = vv.(*clientSessionManager)
	}

	return sessList.OnSyncDataArrived(sessionId, md, data)
}

////////////////////////////////////////////////////////////////////////////////////////////////////////
// session event
func (s *sessionManager) onNewSessionClientManager(sess *clientSessionManager) {
	sess.Start()
}

func (s *sessionManager) onCloseSessionClientManager(authKeyId int64) {
	if vv, ok := s.sessions.Load(authKeyId); ok {
		vv.(*clientSessionManager).Stop()
		s.sessions.Delete(authKeyId)
	}
}