Ejemplo n.º 1
0
// ReadMessage reads from the websocket and unmarshals it into a birpc.Message
func (c *codec) ReadMessage(msg *birpc.Message) error {
	c.rmu.Lock()
	defer c.rmu.Unlock()

	mt, r, err := c.ws.NextReader() // ignoring message type
	if mt != websocket.BinaryMessage {
		return ErrInvalidMsg
	}

	if err != nil {
		return err
	}
	c.r.Reset(r)

	m := &mpc.Message{}
	err = m.DecodeMsg(c.r)
	if err != nil {
		return err
	}

	msg.ID = m.ID
	msg.Func = m.Func
	msg.Args = m.Args
	msg.Result = m.Result
	if m.Error != nil {
		*msg.Error = birpc.Error{Msg: m.Error.Msg}
	}

	return nil
}
Ejemplo n.º 2
0
func (c *codec) ReadMessage(msg *birpc.Message) error {
	var jm jsonMessage
	err := c.dec.Decode(&jm)
	if err != nil {
		return err
	}
	msg.ID = jm.ID
	msg.Func = jm.Func
	msg.Args = jm.Args
	msg.Result = jm.Result
	msg.Error = jm.Error
	return nil
}
Ejemplo n.º 3
0
// ReadMessage reads from the connection and unmarshals the message
// into a birpc.Message
func (c *codec) ReadMessage(msg *birpc.Message) error {
	m := &mpc.Message{}
	err := m.DecodeMsg(c.r)
	if err != nil {
		return err
	}

	msg.ID = m.ID
	msg.Func = m.Func
	msg.Args = m.Args
	msg.Result = m.Result
	if m.Error != nil {
		*msg.Error = birpc.Error{Msg: m.Error.Msg}
	}

	return nil
}
Ejemplo n.º 4
0
// ReadMessage listens on the redis pub-sub channel and unmarshals messages
// into a birpc.Message
func (c *codec) ReadMessage(msg *birpc.Message) error {
	c.rmu.Lock()
	defer c.rmu.Unlock()

	for {
		if c.sub == nil {
			err := c.setupSubscription()
			if err != nil {
				return err
			}
		}

		result, err := c.sub.Pop()
		if err != nil {
			c.sub = nil
			return err
		}

		if result.Value.IsNil() {
			continue
		}

		b := result.Value.Bytes()
		m := &mpc.Message{}
		_, err = m.UnmarshalMsg(b)
		if err != nil {
			return err
		}

		msg.ID = m.ID
		msg.Func = m.Func
		msg.Args = m.Args
		msg.Result = m.Result
		if m.Error != nil {
			*msg.Error = birpc.Error{Msg: m.Error.Msg}
		}

		return nil
	}
}