Пример #1
0
func (chn *Channel) handleWriteSocket(proto *base.Proto) {

	if atomic.CompareAndSwapUint32(&chn.valid, 0, 0) {
		return
	}
	cmdID := proto.RD.BaseHD.CommandId & 0x7f

	if cmdID == base.HEART_CMD {
		heartHdr := new(base.BaseHeader)
		heartHdr.CommandId = proto.RD.BaseHD.CommandId
		msg := heartHdr.Encode()
		_, err := chn.connSocket.Write(msg.Bytes())

		if err != nil {
			chn.handleError(base.DTerror{"Send Error"})
		}
		return
	}
	fmt.Println("handleWriteSocket ", cmdID)
	msg := proto.EncodeHdr()
	_, err := chn.connSocket.Write(msg.Bytes())
	if err != nil {
		chn.handleError(base.DTerror{"Send Error"})

	}
	_, dErr := chn.connSocket.Write(proto.BD.Data)
	if dErr != nil {
		chn.handleError(base.DTerror{"Send Error"})
	}
	base.PutProto(proto)

}
Пример #2
0
func (chn *Channel) handleReadSocketTask() {

	var productID uint32 = 1

	defer func() {
		if r := recover(); r != nil {
			chn.handleError(base.DTerror{"handleReadSocketTask Panic"})
		}
	}()
	for {

		proto := base.GetProto()
		e := proto.ReadBinaryProto(chn.connSocket)
		if e != nil {

			chn.handleError(base.DTerror{"handleReadSocketTask"})
			break
		}

		consumerID := proto.RD.HD.ClientIdent
		cmdID := proto.RD.BaseHD.CommandId & 0x7f
		isRequest := proto.RD.BaseHD.CommandId & 0x80

		switch cmdID {

		case base.HEART_CMD:
			if isRequest == 0 {

				heartProto := base.GetProto()

				heartProto.RD.BaseHD.CommandId = base.HEART_CMD | 0x80

				chn.Send(heartProto)

			} else {
				chn.heartChn <- true
			}

		case base.OPEN_RESOURCE_CMD:
			// 打开资源
			if isRequest == 0 {
				//fmt.Println("打开资源")
				go chn.handleReqOpenRes(consumerID, proto.BD.Data)

			} else {
				//fmt.Println("回应资源")
				go chn.handleRspRes(proto)
			}

		case base.CLOSE_RESOURCE_CMD:
			// 关闭资源
			if isRequest == 0 {
				go chn.handleReqCloseRes(consumerID, proto.BD.Data)
			} else {
				go chn.handleRspRes(proto)
			}

		case base.CRTL_RESOURCE_CMD:
			// 控制资源
		case base.REGISTER_RESOURCE:
			// 注册资源
			go chn.handleRegisterRes(productID, proto.BD.Data)
			productID++
		case base.UNREGISTER_RESOURCE:
			// 注销资源

		case base.DATA_STREAM:
			// 数据流
			chn.handleRspRes(proto)

		default:
			go chn.handleOtherCMD(proto)
		}
		base.PutProto(proto)
	}

	//fmt.Println("End Cu Read Task")
}