return message id

zuoxuefei · Sep 6, 2016 · 308dca4 · 308dca4
1 parent 3e29cf9
commit 308dca4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/network/json/json.go b/network/json/json.go
@@ -34,7 +34,7 @@ func NewProcessor() *Processor {
 }
 
 // It's dangerous to call the method on routing or marshaling (unmarshaling)
-func (p *Processor) Register(msg interface{}) {
+func (p *Processor) Register(msg interface{}) string {
 	msgType := reflect.TypeOf(msg)
 	if msgType == nil || msgType.Kind() != reflect.Ptr {
 		log.Fatal("json message pointer required")
@@ -50,6 +50,7 @@ func (p *Processor) Register(msg interface{}) {
 	i := new(MsgInfo)
 	i.msgType = msgType
 	p.msgInfo[msgID] = i
+	return msgID
 }
 
 // It's dangerous to call the method on routing or marshaling (unmarshaling)

diff --git a/network/protobuf/protobuf.go b/network/protobuf/protobuf.go
@@ -47,7 +47,7 @@ func (p *Processor) SetByteOrder(littleEndian bool) {
 }
 
 // It's dangerous to call the method on routing or marshaling (unmarshaling)
-func (p *Processor) Register(msg proto.Message) {
+func (p *Processor) Register(msg proto.Message) uint16 {
 	msgType := reflect.TypeOf(msg)
 	if msgType == nil || msgType.Kind() != reflect.Ptr {
 		log.Fatal("protobuf message pointer required")
@@ -62,7 +62,9 @@ func (p *Processor) Register(msg proto.Message) {
 	i := new(MsgInfo)
 	i.msgType = msgType
 	p.msgInfo = append(p.msgInfo, i)
-	p.msgID[msgType] = uint16(len(p.msgInfo) - 1)
+	id := uint16(len(p.msgInfo) - 1)
+	p.msgID[msgType] = id
+	return id
 }
 
 // It's dangerous to call the method on routing or marshaling (unmarshaling)