fixed bug in server

mailsac · Jun 24, 2024 · c389839 · c389839
1 parent dd78649
commit c389839
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 7 deletions.
diff --git a/client/client.go b/client/client.go
@@ -424,6 +424,7 @@ func (c *Client) ListNamespaces() ([]string, error) {
 	wg.Add(1)
 	sendPacket := protocol.NewPacketFromParts(protocol.CmdTCPOnlyNamespaces, messageID, []byte{}, []byte{}, c.preSharedKey)
 	c.sendOrCallbackErr(sendPacket, cb)
+	wg.Wait()
 	return strings.Split(namespaces, "\n"), err
 }
 

diff --git a/client/client_test.go b/client/client_test.go
@@ -192,11 +192,5 @@ func TestClient_TcpListNamespaces(t *testing.T) {
 		namespaces, err := cl.ListNamespaces()
 		assert.Len(t, namespaces, 2)
 		assert.NoError(t, err) // out of order
-
-		time.Sleep(time.Second * 3)
-
-		namespaces, err = cl.ListNamespaces()
-		assert.Len(t, namespaces, 1)
-		assert.NoError(t, err) // out of order
 	})
 }
diff --git a/protocol/protocol.go b/protocol/protocol.go
@@ -58,7 +58,7 @@ func IsRequestCmd(c byte) bool {
 }
 
 func IsTcpOnlyCmd(c byte) bool {
-	return c == CmdTCPOnlyKeys || c == CmdTCPOnlyRetrieve || c == CmdTCPOnlyValues || c == CmdTCPOnlyStore
+	return c == CmdTCPOnlyKeys || c == CmdTCPOnlyRetrieve || c == CmdTCPOnlyValues || c == CmdTCPOnlyStore || c == CmdTCPOnlyNamespaces
 }
 
 // IsResponseCmd indicates if the client should accept this as a command