feat: change Protocol.Server.cancel_message/2 function a macro

lib/supavisor/client_handler.ex

@@ -11,7 +11,6 @@
   @behaviour :ranch_protocol
   @behaviour :gen_statem
   @proto [:tcp, :ssl]
-  @cancel_query_msg <<16::32, 1234::16, 5678::16>>
   @switch_active_count Application.compile_env(:supavisor, :switch_active_count)
   @subscribe_retries Application.compile_env(:supavisor, :subscribe_retries)
   @timeout_subscribe 500
@@ -22,10 +21,11 @@
     Helpers,
     Monitoring.Telem,
     Protocol.Client,
-    Protocol.Server,
     Tenants
   }
 
+  require Supavisor.Protocol.Server, as: Server
+
   @impl true
   def start_link(ref, transport, opts) do
     pid = :proc_lib.spawn_link(__MODULE__, :init, [ref, transport, opts])
@@ -111,7 +111,7 @@
   end
 
   # cancel request
-  def handle_event(:info, {_, _, <<@cancel_query_msg, pid::32, key::32>>}, _, _) do
+  def handle_event(:info, {_, _, Server.cancel_message(pid, key)}, _, _) do
     Logger.debug("ClientHandler: Got cancel query for #{inspect({pid, key})}")
     :ok = HandlerHelpers.send_cancel_query(pid, key)
     {:stop, {:shutdown, :cancel_query}}
@@ -275,7 +275,7 @@
 
             key = {:secrets, tenant_or_alias, user}
 
             case auth_secrets(info, user, key, :timer.hours(24)) do
               {:ok, auth_secrets} ->
                 Logger.debug("ClientHandler: Authentication method: #{inspect(auth_secrets)}")
 
@@ -334,7 +334,7 @@
              Cachex.get(Supavisor.Cache, key) == {:ok, nil} do
           case auth_secrets(info, data.user, key, 15_000) do
             {:ok, {method2, secrets2}} = value ->
               if method != method2 or Map.delete(secrets.(), :client_key) != secrets2.() do
                 Logger.warning("ClientHandler: Update secrets and terminate pool")
 
                 Cachex.update(

lib/supavisor/handler_helpers.ex

@@ -2,7 +2,8 @@ defmodule Supavisor.HandlerHelpers do
   @moduledoc false
 
   alias Phoenix.PubSub
-  alias Supavisor.Protocol.Server
+
+  require Supavisor.Protocol.Server, as: Server
 
   @spec sock_send(Supavisor.sock(), iodata()) :: :ok | {:error, term()}
   def sock_send({mod, sock}, data) do

lib/supavisor/protocol/server.ex

@@ -28,6 +28,12 @@
           }
   end
 
+  defmacro cancel_message(pid, key) do
+    quote do
+      <<unquote(@msg_cancel_header)::binary, unquote(pid)::32, unquote(key)::32>>
+    end
+  end
+
   @spec decode(iodata()) :: [Pkt.t()] | []
   def decode(data) do
     decode(data, [])
@@ -456,11 +462,6 @@
     <<byte_size(bin) + 9::32, 0, 3, 0, 0, bin::binary, 0>>
   end
 
-  @spec cancel_message(non_neg_integer, non_neg_integer) :: iodata
-  def cancel_message(pid, key) do
-    [@msg_cancel_header, <<pid::32, key::32>>]
-  end
-
   @spec has_read_only_error?(list) :: boolean
   def has_read_only_error?(pkts) do
     Enum.any?(pkts, fn
@@ -473,5 +474,5 @@
   def application_name, do: @application_name
 
   @spec terminate_message() :: binary
   def terminate_message(), do: @terminate_message
 end

test/supavisor/protocol_test.exs

@@ -1,6 +1,9 @@
 defmodule Supavisor.ProtocolTest do
   use ExUnit.Case, async: true
-  alias Supavisor.Protocol.Server, as: S
+
+  @subject Supavisor.Protocol.Server
+
+  require Supavisor.Protocol.Server
 
   @initial_data %{
     "DateStyle" => "ISO, MDY",
@@ -27,42 +30,43 @@ defmodule Supavisor.ProtocolTest do
                     97, 105, 108, 101, 100, 0, 0>>
 
   test "encode_parameter_status/1" do
-    result = S.encode_parameter_status(@initial_data)
+    result = @subject.encode_parameter_status(@initial_data)
 
-    for {key, value} <- @initial_data do
-      assert :erlang.is_binary(key)
-      assert :erlang.is_binary(value)
-      encoded = S.encode_pkt(:parameter_status, key, value)
-      assert Enum.member?(result, encoded)
+    for entry <- @initial_data do
+      assert {key, value} = entry
+      assert is_binary(key)
+      assert is_binary(value)
+      encoded = @subject.encode_pkt(:parameter_status, key, value)
+      assert encoded in result
     end
   end
 
   test "encode_pkt/3" do
     key = "TimeZone"
     value = "UTC"
-    result = S.encode_pkt(:parameter_status, key, value)
+    result = @subject.encode_pkt(:parameter_status, key, value)
 
     assert result == [<<?S, 17::32>>, [key, <<0>>, value, <<0>>]]
   end
 
   test "backend_key_data/0" do
-    {header, payload} = S.backend_key_data()
+    {header, payload} = @subject.backend_key_data()
     len = byte_size(payload) + 4
 
     assert [
-             %S.Pkt{
+             %@subject.Pkt{
                tag: :backend_key_data,
                len: 13,
                payload: %{pid: _, key: _}
              }
-           ] = S.decode([header, payload] |> IO.iodata_to_binary())
+           ] = @subject.decode([header, payload] |> IO.iodata_to_binary())
 
     assert header == <<?K, len::32>>
     assert byte_size(payload) == 8
   end
 
   test "decode_payload for error_response" do
-    assert S.decode(@auth_bin_error) == [
+    assert @subject.decode(@auth_bin_error) == [
              %Supavisor.Protocol.Server.Pkt{
                tag: :error_response,
                len: 112,
@@ -84,7 +88,6 @@ defmodule Supavisor.ProtocolTest do
     key = 123_456
     expected = <<0, 0, 0, 16, 4, 210, 22, 46, 0, 0, 0, 123, 0, 1, 226, 64>>
 
-    assert S.cancel_message(pid, key)
-           |> IO.iodata_to_binary() == expected
+    assert @subject.cancel_message(pid, key) == expected
   end
 end