diff options
Diffstat (limited to 'internal/server/api/helpers.go')
| -rw-r--r-- | internal/server/api/helpers.go | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/internal/server/api/helpers.go b/internal/server/api/helpers.go index 4327a49..2ed0694 100644 --- a/internal/server/api/helpers.go +++ b/internal/server/api/helpers.go @@ -79,8 +79,7 @@ func NetworkPropagateWithFilter( context, cancel := context.WithTimeout(context.Background(), timeout) go func() { defer cancel() - pkt := packet.NewPacket(packet.NewMsgPackEncoder(payload)) - if ok := session.Write(context, pkt); !ok { + if ok := session.Write(context, WrapPayload(payload)); !ok { log.Println(sess.Addr(), "propagation to", session.Addr(), "failed") } }() @@ -122,8 +121,7 @@ func UserPropagate( context, cancel := context.WithTimeout(context.Background(), timeout) go func() { defer cancel() - pkt := packet.NewPacket(packet.NewMsgPackEncoder(payload)) - if ok := session.Write(context, pkt); !ok { + if ok := session.Write(context, WrapPayload(payload)); !ok { log.Println(sess.Addr(), "propagation to", session.Addr(), "failed") } }() @@ -191,3 +189,7 @@ func getNotifications(ctx context.Context, userId snowflake.ID) (packet.Notifica } return items, nil } + +func WrapPayload(payload packet.Payload) packet.Packet { + return packet.NewPacket(packet.NewMsgPackEncoder(payload)) +} |