ft: improve some decoding paths

supabase · Jun 28, 2024 · 2413263 · 2413263
1 parent 91df94f
commit 2413263
Showing 1 changed file with 6 additions and 10 deletions.
diff --git a/lib/supavisor/protocol/client.ex b/lib/supavisor/protocol/client.ex
@@ -103,22 +103,18 @@ defmodule Supavisor.Protocol.Client do
   end
 
   def decode_payload(:simple_query, payload) do
-    case String.split(payload, <<0>>) do
+    case String.split(payload, <<0>>, parts: 2) do
       [query, ""] -> query
       _ -> :undefined
     end
   end
 
+  def decode_payload(:parse_message, <<0>>), do: :undefined
+
   def decode_payload(:parse_message, payload) do
-    case String.split(payload, <<0>>) do
-      [""] ->
-        :undefined
-
-      other ->
-        case Enum.filter(other, &(&1 != "")) do
-          [sql] -> sql
-          message -> message
-        end
+    case String.split(payload, <<0>>, trim: true) do
+      [sql] -> sql
+      message -> message
     end
   end