origin/sysservice/tcpgateway/Router.go

package tcpgateway

import (
	"github.com/duanhf2012/origin/log"
	"github.com/duanhf2012/origin/node"
	"github.com/duanhf2012/origin/rpc"
	"github.com/duanhf2012/origin/sysservice/tcpservice"
	"github.com/golang/protobuf/proto"
	"strings"
)

type RouterCfg struct {
}

type Router struct {
	loadBalance ILoadBalance
	rpcHandler rpc.IRpcHandler

	mapMsgRouterInfo map[uint16]*MsgRouterInfo
	mapEventRouterInfo map[string]*EventRouterInfo

	tcpService *tcpservice.TcpService

	mapClientRouterCache map[uint64]map[string]int //map[clientid]nodeid
}

type MsgRouterInfo struct {
	Rpc string
	ServiceName string
	LoadBalanceType string
}

type EventRouterInfo struct {
	Rpc string
	ServiceName string
	LoadBalanceType string
}

func NewRouter(loadBalance ILoadBalance,rpcHandler rpc.IRpcHandler,cfg interface{}) IRouter {
	router := &Router{}
	router.loadBalance = loadBalance
	router.rpcHandler = rpcHandler
	router.tcpService =  node.GetService("TcpService").(*tcpservice.TcpService)
	router.loadCfg(cfg)
	router.mapClientRouterCache = map[uint64]map[string]int{}
	return router
}

func (slf *Router) loadCfg(cfg interface{}){
	slf.mapMsgRouterInfo = map[uint16]*MsgRouterInfo{}
	slf.mapEventRouterInfo = map[string]*EventRouterInfo{}

	mapRouter,ok := cfg.(map[string]interface{})
	if ok == false{
		//error....
		return
	}

	//parse MsgRouter
	 routerInfo,ok := mapRouter["MsgRouter"]
	 if ok == false{
	 	//error...
	 	return
	 }

	 //ar routerList []RouterItem
	routerList,ok := routerInfo.([]interface{})
	if ok == false{
		//error...
		return
	}

	for _,v := range routerList{
		mapItem := v.(map[string]interface{})
		var iMsgId interface{}
		var iRpc interface{}
		var iLoadBalanceType interface{}

		if iMsgId,ok = mapItem["MsgId"];ok == false {
			//error ...
			continue
		}
		if iRpc,ok = mapItem["Rpc"];ok == false {
			//error ...
			continue
		}
		if iLoadBalanceType,ok = mapItem["LoadBalanceType"];ok == false {
			//error ...
			continue
		}
		msgId,ok  := iMsgId.(float64)
		if ok == false {
			//error ...
			continue
		}

		strService := strings.Split(iRpc.(string),".")
		if len(strService)!=2 {
			//error ...
			continue
		}

		slf.mapMsgRouterInfo[uint16(msgId)] = &MsgRouterInfo{ServiceName:strService[0],Rpc: iRpc.(string),LoadBalanceType: iLoadBalanceType.(string)}
	}

	//parse EventRouter
	eventRouterInfo,ok := mapRouter["EventRouter"]
	if ok == false{
		//error...
		return
	}

	//ar routerList []RouterItem
	eRouterList,ok := eventRouterInfo.([]interface{})
	if ok == false{
		//error...
		return
	}

	for _,v := range eRouterList{
		mapItem := v.(map[string]interface{})
		var eventType interface{}
		var iRpc interface{}
		var iLoadBalanceType interface{}

		if eventType,ok = mapItem["EventType"];ok == false {
			//error ...
			continue
		}
		if iRpc,ok = mapItem["Rpc"];ok == false {
			//error ...
			continue
		}
		if iLoadBalanceType,ok = mapItem["LoadBalanceType"];ok == false {
			//error ...
			continue
		}
		strEventType,ok  := eventType.(string)
		if ok == false {
			//error ...
			continue
		}

		strService := strings.Split(iRpc.(string),".")
		if len(strService)!=2 {
			//error ...
			continue
		}

		slf.mapEventRouterInfo[strEventType] = &EventRouterInfo{ServiceName:strService[0],Rpc: iRpc.(string),LoadBalanceType: iLoadBalanceType.(string)}
	}
}

func (slf *Router) GetMsgRouterService(msgType uint16) *MsgRouterInfo{
	info,ok := slf.mapMsgRouterInfo[msgType]
	if ok == false {
		return nil
	}

	return info
}

func (slf *Router) GetEventRouterService(eventType string) *EventRouterInfo{
	info,ok := slf.mapEventRouterInfo[eventType]
	if ok == false {
		return nil
	}

	return info
}

func (slf *Router) GetRouterId(clientId uint64,serviceName *string) int {
	mapServiceRouter,ok := slf.mapClientRouterCache[clientId]
	if ok == false{
		return 0
	}

	routerId,ok := mapServiceRouter[*serviceName]
	if ok == false {
		return 0
	}

	return routerId
}

func (slf *Router) SetRouterId(clientId uint64,serviceName *string,routerId int){
	slf.mapClientRouterCache[clientId][*serviceName] = routerId
}

func (slf *Router) RouterMessage(clientId uint64,msgType uint16,msg []byte) {
	routerInfo:= slf.GetMsgRouterService(msgType)
	if routerInfo==nil {
		log.Error("The message type is %d with no configured route!",msgType)
		return
	}

	routerId := slf.GetRouterId(clientId,&routerInfo.ServiceName)
	if routerId ==0 {
		routerId = slf.loadBalance.SelectNode(routerInfo.ServiceName)
		slf.SetRouterId(clientId,&routerInfo.ServiceName,routerId)
	}

	if routerId>0 {
		slf.rpcHandler.RawGoNode(rpc.RPC_PROCESSOR_PB,routerId,routerInfo.Rpc,msg,proto.Uint64(clientId))
	}
}

func (slf *Router) Load(){
}

func (slf *Router) RouterEvent(clientId uint64,eventType string) bool{
	routerInfo:= slf.GetEventRouterService(eventType)
	if routerInfo==nil {
		log.Error("The event type is %s with no register!",eventType)
		return false
	}

	routerId := slf.GetRouterId(clientId,&routerInfo.ServiceName)
	if routerId ==0 {
		routerId = slf.loadBalance.SelectNode(routerInfo.ServiceName)
		slf.SetRouterId(clientId,&routerInfo.ServiceName,routerId)
	}

	if routerId>0 {
		slf.rpcHandler.RawGoNode(rpc.RPC_PROCESSOR_PB,routerId,routerInfo.Rpc,[]byte{},proto.Uint64(clientId))
		return true
	}

	return false
}


func (slf *Router) OnDisconnected(clientId uint64){
	delete(slf.mapClientRouterCache,clientId)
	//通知事件
	slf.RouterEvent(clientId,"DisConnect")
}

func (slf *Router) OnConnected(clientId uint64){
	slf.mapClientRouterCache[clientId] = map[string]int{}
	slf.RouterEvent(clientId,"Connect")
}