src/components/net/server.go

package net
import (
	"fmt"
	"github.com/golang/protobuf/proto"
	"net"
	"pro2d/conf"
	"pro2d/protos/pb"
	"pro2d/src/components/db"
	"pro2d/src/components/etcd"
	"pro2d/src/components/logger"
	"pro2d/src/models"
	"sync"
)
type ActionHandler func (msg *MsgPkg)  (int32, proto.Message)
var ActionMap map[pb.ProtoCode]ActionHandler
type Server struct {
	SConf      *conf.SConf
	Clients *sync.Map
	EtcdClient *etcd.EtcdClient
}
func NewServer(sConf *conf.SConf) *Server {
	return &Server{
		SConf:      sConf,
		Clients:    new(sync.Map),
	}
}
func (s *Server) OnRecv(msg *MsgPkg) {
	logger.Debug("cmd: %d, data: %s", msg.Head.Cmd, msg.Body)
	if md, ok := ActionMap[pb.ProtoCode(msg.Head.Cmd)]; ok {
		errCode, protomsg := md(msg)
		rsp, err := proto.Marshal(protomsg)
		if err != nil {
			msg.Conn.SendMsgByCode(-100, msg.Head.Cmd, nil)
			return
		}
		msg.Conn.SendMsgByCode(errCode, msg.Head.Cmd, rsp)
		return
	}
	logger.Error("protocode not handler: %d", msg.Head.Cmd)
}
func (s *Server) OnClose(conn *Connection) {
	s.Clients.Delete(conn.Id)
}
func (s *Server)Start() error {
	//mongo 初始化
	db.MongoDatabase = db.MongoClient.Database(conf.GlobalConf.AccountConf.DBName)
	models.InitGameServerModels()
	//Etcd 初始化
	s.EtcdClient = etcd.NewEtcdClient(conf.GlobalConf.Etcd)
	s.EtcdClient.PutWithLeasePrefix(conf.GlobalConf.GameConf.Name, conf.GlobalConf.GameConf.ID, fmt.Sprintf("%s:%d", conf.GlobalConf.GameConf.IP, conf.GlobalConf.GameConf.Port), 5)
	port := fmt.Sprintf(":%d", s.SConf.Port)
	l, err := net.Listen("tcp", port)
	if err != nil {
		return err
	}
	logger.Debug("listen on %s\n", port)
	id := 0
	for {
		conn, err := l.Accept()
		if err != nil {
			return err
		}
		id++
		client := NewConn(id, conn, s)
		s.Clients.Store(id, client)
		go client.Start()
	}
}
func (s *Server)Stop()  {
}