74 lines
1.6 KiB
Go
74 lines
1.6 KiB
Go
package pool
|
|
|
|
import (
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/service/net/peer"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/syncproto"
|
|
"go.uber.org/zap"
|
|
"gopkg.in/mgo.v2/bson"
|
|
)
|
|
|
|
type Message struct {
|
|
*syncproto.Message
|
|
peer peer.Peer
|
|
}
|
|
|
|
func (m *Message) Peer() peer.Peer {
|
|
return m.peer
|
|
}
|
|
|
|
func (m *Message) Reply(data []byte) (err error) {
|
|
rep := &syncproto.Message{
|
|
Header: &syncproto.Header{
|
|
TraceId: m.GetHeader().TraceId,
|
|
ReplyId: m.GetHeader().RequestId,
|
|
Type: syncproto.MessageType_MessageTypeSync,
|
|
},
|
|
Data: data,
|
|
}
|
|
return m.peer.Send(rep)
|
|
}
|
|
|
|
func (m *Message) Ack() (err error) {
|
|
ack := &syncproto.System{
|
|
Ack: &syncproto.SystemAck{},
|
|
}
|
|
data, err := ack.Marshal()
|
|
if err != nil {
|
|
return
|
|
}
|
|
rep := &syncproto.Message{
|
|
Header: &syncproto.Header{
|
|
TraceId: m.GetHeader().TraceId,
|
|
ReplyId: m.GetHeader().RequestId,
|
|
Type: syncproto.MessageType_MessageTypeSystem,
|
|
},
|
|
Data: data,
|
|
}
|
|
log.With(zap.String("header", rep.Header.String())).Info("sending ack to peer")
|
|
return m.peer.Send(rep)
|
|
}
|
|
|
|
func (m *Message) AckError(code syncproto.SystemErrorCode, description string) (err error) {
|
|
ack := &syncproto.System{
|
|
Ack: &syncproto.SystemAck{
|
|
Error: &syncproto.SystemError{
|
|
Code: code,
|
|
Description: description,
|
|
},
|
|
},
|
|
}
|
|
data, err := ack.Marshal()
|
|
if err != nil {
|
|
return
|
|
}
|
|
rep := &syncproto.Message{
|
|
Header: &syncproto.Header{
|
|
TraceId: []byte(bson.NewObjectId()),
|
|
ReplyId: m.GetHeader().RequestId,
|
|
Type: syncproto.MessageType_MessageTypeSystem,
|
|
},
|
|
Data: data,
|
|
}
|
|
return m.peer.Send(rep)
|
|
}
|