Golang Message.Decode方法代码示例

func (this *NodeComponent) ForwardPush(msg *structure.Packet) {
	m := protocol.Message{}
	err := m.Decode(msg.Data)
	if err != nil {
		fmt.Println("NodeComponent Error:MSG Decode Failed", err)
		return
	}
	this.ForwardMsgs <- msg
}

func (this *NodeComponent) ReceiveMsg() {

	defer func() {
		if err := recover(); err != nil {
			fmt.Println("NodeComponent Error: " + err.(error).Error())
		}
	}()

	headbuff := make([]byte, define.PACKET_HEAD_LEN)
	for {

		var head structure.Head
		var Packet structure.Packet
		//如果无连接，尝试重新建立tcp连接
		if this.StreeConn == nil {
			this.ReConnect()
			continue
		}
		//从tcp连接中先读取头部信息
		n, err := this.StreeConn.Read(headbuff)
		//如果读取出错，则尝试重新建立tcp连接
		if n == 0 || n != define.PACKET_HEAD_LEN || err != nil {
			fmt.Println("NodeComponent Error:Read STree msg")
			this.ReConnect()
			continue
		}

		err = head.Decode(headbuff)
		if err != nil {
			fmt.Println("NodeComponent Error: Decode Stree msg")
			return
		}

		databuff := make([]byte, head.DataLen)
		//再从连接中读取剩余数据
		n, err = this.StreeConn.Read(databuff)
		if n == 0 || n != int(head.DataLen) || err != nil {
			fmt.Println("NodeComponent Error: Read Stree Data")
			this.ReConnect()
			continue
		}

		Packet.Head = head
		Packet.Data = databuff
		msg := protocol.Message{}
		//Packet.Data至少44位
		err = msg.Decode(Packet.Data)
		if err != nil {
			fmt.Println("NodeComponent Error:MSG Decode Failed", err)
			return
		}

		this.ReceiveMsgHandle(&Packet)
	}
}

func (this *DaughterHandler) ReceiveFromDaughter(conn *net.Conn) {
	for {
		//PACKET_HEAD_LEN = 19
		headBuf := make([]byte, define.PACKET_HEAD_LEN)
		//MAX_BUF = 2048
		data := make([]byte, define.MAX_BUF)
		//读取头部信息
		_, err := (*conn).Read(headBuf)
		if err != nil && err != io.EOF {
			fmt.Println(err.Error())
			continue
		} else if err == io.EOF {
			fmt.Println("\n", time.Now().String(), " In DaughterHander.ReceiverFromDaughter, connection ", (*conn).RemoteAddr(), " is invalid", "\n")
			(*conn).Close()
			return
		}
		head := structure.Head{}
		err = head.Decode(headBuf)
		if err != nil {
			err = errors.New("component: STree.Receive failed, can not decode the head buf, " + err.Error())
			fmt.Println(err.Error())
			continue
		}
		if head.DataLen > 0 {
			if int(head.DataLen) > len(data) {
				data = make([]byte, head.DataLen)
			}
			//从conn中读取Data[]部分
			_, err = (*conn).Read(data[:head.DataLen])
			if err != nil && err != io.EOF {
				fmt.Println("Got a error ", err)
				continue
			} else if err == io.EOF {
				fmt.Println("\n", time.Now().String(), " In DaughterHander.ReceiverFromDaughter, connection ", (*conn).RemoteAddr(), " is invalid", "\n")
				(*conn).Close()
				return
			}
		}

		packet := &structure.Packet{head, data[:head.DataLen]}
		if packet.Number == define.FORWARD {
			msg := push_protocol.Message{}
			err = msg.Decode(packet.Data)
			if err != nil {
				fmt.Println("MSG Decode Failed, ", err)
				return
			}
			fmt.Println("\n", time.Now().String(), " ReceiveFromDaughter, packet.Uuid: ", packet.Uuid, " msg.UserId: ", msg.UserId, " msg.DestId: ", msg.DestId, "\n")
		}
		this.Dispatch(packet, conn)
	}
}

func (this *NodeComponent) NoCacheForward(Uuid string, Data []byte) {
	Packet := GetPacket(define.FORWARD, Uuid, Data)

	buff, err := Packet.Encode()
	if err != nil {
		fmt.Println("NodeComponent Error:Encode No Cache Forward Packet")
		return
	}

	n, err := this.Write(buff)
	if n == 0 || err != nil {
		return
	}

	msg := protocol.Message{}
	err = msg.Decode(Packet.Data)
	if err != nil {
		fmt.Println("NodeComponent Error: MSG Decode Failed, ", err)
		return
	}
}

func (this *DaughterHandler) SendToDaughter() {
	for {
		//	fmt.Println("\n", time.Now().String(), " SendToDaughter Directly", "\n")
		//        packet := <-this.toDaughter
		//从toDaughter中接收数据
		packet := this.ToDaughterPop()
		//返回一个*net.Conn
		conn := GetSTree().UserMap.Get(packet.Uuid)
		if conn == nil && packet.Number == define.ALLOC_SERVER {
			fmt.Println("\n", "Geta Min LoadConn ", "\n")
			conn = GetSTree().GetMinLoadConn()
		}
		if conn == nil {
			err := errors.New("component: STree.SendToDaughter() failed, can not find user in user's list")
			fmt.Println(err.Error())
			fmt.Println("\n", "destID = ", packet.Uuid, "\n")
			continue
		}
		buf, err := packet.Encode()
		if err != nil {
			err = errors.New("component: STree.SendToDaughter() failed, can not get buf from packet, " + err.Error())
			fmt.Println(err.Error())
			continue
		}
		//向conn中写入Packet的buf
		_, err = (*conn).Write(buf)
		if err != nil {
			fmt.Println("\n", time.Now().String(), " in SendToDaughter, Got an error ", err, "\n")
			fmt.Println(err.Error())
			continue
		}
		msg := push_protocol.Message{}
		err = msg.Decode(packet.Data)
		if err != nil {
			fmt.Println("MSG Decode Failed, ", err)
			return
		}
		fmt.Println("\n", time.Now().String(), " SendToDaughter Directly OK, packet.Uuid: ", packet.Uuid, " msg.UserId: ", msg.UserId, " msg.DestId: ", msg.DestId, "\n")
	}
}