Пример #1
0
func ReplyErrorAndArgs(errcode int32, args ...interface{}) *rpc.Message {
	msg := rpc.NewMessage(rpc.MAX_BUF_LEN)

	if errcode > 0 {
		sr := util.NewStoreArchiver(msg.Header)
		sr.Write(int8(1))
		sr.Write(errcode)
		msg.Header = msg.Header[:sr.Len()]
	}

	if len(args) > 0 {
		mw := NewMessageWriter(msg)
		for i := 0; i < len(args); i++ {
			err := mw.Write(args[i])
			if err != nil {
				msg.Free()
				panic("write args failed")
				return nil
			}
		}
		mw.Flush()
	}

	return msg
}
Пример #2
0
//错误消息
func ReplyErrorMessage(errcode int32) *rpc.Message {
	msg := rpc.NewMessage(1)
	if errcode == 0 {
		return msg
	}
	sr := util.NewStoreArchiver(msg.Header)
	sr.Write(int8(1))
	sr.Write(errcode)
	msg.Header = msg.Header[:sr.Len()]
	return msg
}
Пример #3
0
//向客户端发包,存进一个消息队列,异步发送
func (c *ClientNode) Send(data []byte) error {
	if c.quit {
		return fmt.Errorf("client is quit")
	}
	message := rpc.NewMessage(len(data))
	message.Body = append(message.Body, data...)
	select {
	case c.Sendbuf <- message:
		return nil
	default:
		message.Free() //满了,直接扔掉
		return fmt.Errorf("send buffer full")
	}
}
Пример #4
0
func (c *ClientCodec) ReadRequest(maxrc uint16) (*rpc.Message, error) {
	for {
		id, data, err := util.ReadPkg(c.rwc, c.cachebuf)
		if err != nil {
			return nil, err
		}
		switch id {
		case share.C2S_PING:
			c.node.Ping()
			break
		case share.C2S_RPC:
			msg := rpc.NewMessage(len(data))
			ar := NewHeadWriter(msg)
			ar.Write(uint64(0))
			ar.Write(c.node.MailBox.Uid)
			ar.Write("C2SC2SHelper.Call")
			msg.Header = msg.Header[:ar.Len()]
			msg.Body = append(msg.Body, data...)
			return msg, nil
		}
	}
}
Пример #5
0
func (s *S2CHelper) call(src rpc.Mailbox, session int64, method string, out []byte) error {
	c := core.clientList.FindNode(session)
	if c == nil {
		return ErrClientNotFound
	}

	cachedata, exist := s.cachedata[session]

	if exist && cachedata != nil { //优先写入缓存
		if len(cachedata.Body)+len(out) <= packagesize { //可以写入
			cachedata.Body = append(cachedata.Body, out...)
			return nil
		}

		//超出长度,则先发送当前消息,保证消息顺序
		if err := c.SendMessage(cachedata); err != nil {
			log.LogError(session, err)
		}
		cachedata.Free()
		cachedata = nil
		s.cachedata[session] = nil
	}

	if len(out) > packagesize { //超出包体大小,直接发送
		//直接发送
		if err := c.Send(out); err != nil {
			log.LogError(session, err)
		}
		return nil
	}

	//写入缓冲区
	cachedata = rpc.NewMessage(packagesize)
	s.cachedata[session] = cachedata
	cachedata.Body = append(cachedata.Body, out...)
	return nil
}
Пример #6
0
func NewMessage() *BodyWriter {
	msg := rpc.NewMessage(rpc.MAX_BUF_LEN)
	w := &BodyWriter{util.NewStoreArchiver(msg.Body), msg}
	return w
}