增加消息发送

network/processor.go

@@ -1,5 +1,6 @@
 package network
 
+
 type Processor interface {
 	// must goroutine safe
 	Route(msg interface{}, userData interface{}) error

network/processor/pbprocessor.go

@@ -12,7 +12,8 @@ type MessageInfo struct {
 	msgHandler MessageHandler
 }
 
-
+type MessageHandler func(clientid uint64,msg proto.Message)
+const MsgTypeSize = 2
 type PBProcessor struct {
 	mapMsg map[uint16]MessageInfo
 	LittleEndian bool
@@ -65,10 +66,22 @@ func (slf *PBProcessor ) Unmarshal(data []byte) (interface{}, error) {
 
 // must goroutine safe
 func (slf *PBProcessor ) Marshal(msg interface{}) ([]byte, error){
-	return proto.Marshal(msg.(proto.Message))
-}
+	bytes,err := proto.Marshal(msg.(proto.Message))
+	if err != nil {
+		return nil,err
+	}
 
-type MessageHandler func(clientid uint64,msg proto.Message)
+	buff := make([]byte, 0, len(bytes)+MsgTypeSize)
+	pMsg := msg.(*PBPackInfo)
+	if slf.LittleEndian == true {
+		binary.LittleEndian.PutUint16(buff[:2],pMsg.typ)
+	}else{
+		binary.BigEndian.PutUint16(buff[:2],pMsg.typ)
+	}
+
+	buff = append(buff,bytes...)
+	return buff,nil
+}
 
 func (slf *PBProcessor) Register(msgtype uint16,msg proto.Message,handle MessageHandler)  {
 	var info MessageInfo
@@ -77,3 +90,7 @@ func (slf *PBProcessor) Register(msgtype uint16,msg proto.Message,handle Message
 	info.msgHandler = handle
 	slf.mapMsg[msgtype] = info
 }
+
+func (slf *PBProcessor) MakeMsg(msgType uint16,protoMsg proto.Message) *PBPackInfo {
+	return &PBPackInfo{typ:msgType,msg:protoMsg}
+}