98 lines
2.2 KiB
Go
98 lines
2.2 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,
|
|
DebugInfo: "Ack",
|
|
},
|
|
Data: data,
|
|
}
|
|
err = m.peer.Send(rep)
|
|
if err != nil {
|
|
log.With(
|
|
zap.String("peerId", m.peer.Id()),
|
|
zap.String("header", rep.GetHeader().String())).
|
|
Error("failed sending ack to peer", zap.Error(err))
|
|
} else {
|
|
log.With(
|
|
zap.String("peerId", m.peer.Id()),
|
|
zap.String("header", rep.GetHeader().String())).
|
|
Debug("sent ack to peer")
|
|
}
|
|
return
|
|
}
|
|
|
|
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,
|
|
DebugInfo: "AckError",
|
|
},
|
|
Data: data,
|
|
}
|
|
if err != nil {
|
|
log.With(
|
|
zap.String("peerId", m.peer.Id()),
|
|
zap.String("header", rep.GetHeader().String())).
|
|
Error("failed sending ackError to peer", zap.Error(err))
|
|
} else {
|
|
log.With(
|
|
zap.String("peerId", m.peer.Id()),
|
|
zap.String("header", rep.GetHeader().String())).
|
|
Debug("sent ackError to peer")
|
|
}
|
|
return m.peer.Send(rep)
|
|
}
|