提交origin2.0版本

example/GateService/GateService.go

@@ -0,0 +1,43 @@
+package GateService
+
+import (
+	"github.com/duanhf2012/originnet/event"
+	"github.com/duanhf2012/originnet/node"
+	"github.com/duanhf2012/originnet/service"
+	"github.com/duanhf2012/originnet/sysservice"
+)
+
+type GateService struct {
+	processor *PBProcessor
+	service.Service
+}
+
+func (slf *GateService) OnInit() error{
+	tcpervice := node.GetService("TcpService").(*sysservice.TcpService)
+	tcpervice.SetProcessor(&PBProcessor{})
+	return nil
+}
+
+
+func (slf *GateService) OnEventHandler(ev *event.Event) error{
+	if ev.Type == event.Sys_Event_Tcp_RecvPack {
+		pPack := ev.Data.(*sysservice.TcpPack)
+		slf.processor.Route(pPack.Data,pPack.ClientId)
+	}else if ev.Type == event.Sys_Event_Tcp_Connected {
+		pPack := ev.Data.(*sysservice.TcpPack)
+		slf.OnConnected(pPack.ClientId)
+	}else if ev.Type == event.Sys_Event_Tcp_DisConnected {
+		pPack := ev.Data.(*sysservice.TcpPack)
+		slf.OnDisconnected(pPack.ClientId)
+	}
+	return nil
+}
+
+func (slf *GateService) OnConnected(clientid uint64){
+
+}
+
+
+func (slf *GateService) OnDisconnected(clientid uint64){
+
+}

example/GateService/pbprocessor.go

@@ -0,0 +1,79 @@
+package GateService
+
+import (
+	"encoding/binary"
+	"fmt"
+	"github.com/golang/protobuf/proto"
+	"reflect"
+)
+
+type MessageInfo struct {
+	msgType    reflect.Type
+	msgHandler MessageHandler
+}
+
+
+type PBProcessor struct {
+	mapMsg map[uint16]MessageInfo
+	LittleEndian bool
+}
+
+func (slf *PBProcessor) SetLittleEndian(littleEndian bool){
+	slf.LittleEndian = littleEndian
+}
+
+type PackInfo struct {
+	typ uint16
+	msg proto.Message
+}
+
+// must goroutine safe
+func (slf *PBProcessor ) Route(msg interface{},userdata interface{}) error{
+	pPackInfo := msg.(*PackInfo)
+	v,ok := slf.mapMsg[pPackInfo.typ]
+	if ok == false {
+		return fmt.Errorf("cannot find msgtype %d is register!",pPackInfo.typ)
+	}
+
+
+	v.msgHandler(userdata.(uint64),pPackInfo.msg)
+	return nil
+}
+
+// must goroutine safe
+func (slf *PBProcessor ) Unmarshal(data []byte) (interface{}, error) {
+	var msgType uint16
+	if slf.LittleEndian == true {
+		msgType = binary.LittleEndian.Uint16(data[:2])
+	}else{
+		msgType = binary.BigEndian.Uint16(data[:2])
+	}
+
+	info,ok := slf.mapMsg[msgType]
+	if ok == false {
+		return nil,fmt.Errorf("cannot find register %d msgtype!",msgType)
+	}
+	msg := reflect.New(info.msgType.Elem()).Interface()
+	protoMsg := msg.(proto.Message)
+	err := proto.Unmarshal(data[2:], protoMsg)
+	if err != nil {
+		return nil,err
+	}
+
+	return &PackInfo{typ:msgType,msg:protoMsg},nil
+}
+
+// must goroutine safe
+func (slf *PBProcessor ) Marshal(msg interface{}) ([]byte, error){
+	return proto.Marshal(msg.(proto.Message))
+}
+
+type MessageHandler func(clientid uint64,msg proto.Message)
+
+func (slf *PBProcessor) Register(msgtype uint16,msg proto.Message,handle MessageHandler)  {
+	var info MessageInfo
+
+	info.msgType = reflect.TypeOf(msg.(proto.Message))
+	info.msgHandler = handle
+	slf.mapMsg[msgtype] = info
+}