Fixed everything on the server-side

1 files changed, 9 insertions, 4 deletions

diff --git a/internal/server/server.go b/internal/server/server.go
index 56c3a50..9abf51f 100644
--- a/internal/server/server.go
+++ b/internal/server/server.go
@@ -332,14 +332,16 @@ func processRequest(ctx context.Context, sess *session.Session, request packet.P
 	// Potentially even separate time for code vs DB operations
 	var response packet.Payload
 	switch request := request.(type) {
+
+	case *packet.SetUserData:
+		response = timeout(5*time.Millisecond, api.SetUserData, ctx, sess, request)
+	case *packet.GetUserData:
+		response = timeout(5*time.Millisecond, api.GetUserData, ctx, sess, request)
+
 	case *packet.CreateNetwork:
 		response = timeout(10*time.Millisecond, api.CreateNetwork, ctx, sess, request)
 	case *packet.DeleteNetwork:
 		response = timeout(500*time.Millisecond, api.DeleteNetwork, ctx, sess, request)
-	case *packet.SwapUserNetworks:
-		response = timeout(5*time.Millisecond, api.SwapUserNetworks, ctx, sess, request)
-	case *packet.SetMember:
-		response = timeout(50*time.Millisecond, api.SetMember, ctx, sess, request)
 
 	case *packet.CreateFrequency:
 		response = timeout(5*time.Millisecond, api.CreateFrequency, ctx, sess, request)
@@ -353,6 +355,9 @@ func processRequest(ctx context.Context, sess *session.Session, request packet.P
 	case *packet.RequestMessages:
 		response = timeout(50*time.Millisecond, api.RequestMessages, ctx, sess, request)
 
+	case *packet.SetMember:
+		response = timeout(50*time.Millisecond, api.SetMember, ctx, sess, request)
+
 	default:
 		response = &packet.Error{Error: "use of disallowed packet type for request"}
 	}