From 08d88c425b4cfe4918eabe83b003ee01314e636a Mon Sep 17 00:00:00 2001 From: Kyren223 Date: Sat, 2 Nov 2024 13:00:37 +0200 Subject: test(packet): test packet framer and packet encoding/decoding --- internal/server/server.go | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'internal/server/server.go') diff --git a/internal/server/server.go b/internal/server/server.go index 05af434..1b8dccb 100644 --- a/internal/server/server.go +++ b/internal/server/server.go @@ -152,7 +152,7 @@ func handleConnection(ctx context.Context, conn net.Conn, server *server) { sess := session.NewSession(server, addr, id, pubKey) server.AddSession(sess) ctx = session.NewContext(ctx, sess) - framer := packet.NewFramer(ctx) + framer := packet.NewFramer() defer func() { conn.Close() @@ -292,9 +292,11 @@ func processRequest(ctx context.Context, request packet.Payload) packet.Payload switch request := request.(type) { case *packet.SendMessage: - return timeout(100 * time.Millisecond, api.SendMessage, ctx, request) + return timeout(50 * time.Millisecond, api.SendMessage, ctx, request) + case *packet.GetMessagesRange: + return timeout(100 * time.Millisecond, api.GetMessages, ctx, request) default: - return &packet.ErrorMessage{Error: "use of unsupported packet type"} + return &packet.ErrorMessage{Error: "use of disallowed packet type for request"} } } -- cgit v1.3.1