fix: cache for user query, server_name_indication, change log level

VERSION

@@ -1 +1 @@
-0.9.10
+0.9.11

lib/supavisor/client_handler.ex

@@ -119,7 +119,7 @@ defmodule Supavisor.ClientHandler do
   def handle_event(:internal, {:hello, {user, external_id}}, :exchange, %{sock: sock} = data) do
     sni_hostname = try_get_sni(sock)
 
-    case Tenants.get_user(user, external_id, sni_hostname) do
+    case Tenants.get_user_cache(user, external_id, sni_hostname) do
       {:ok, info} ->
         id = Supavisor.id(info.tenant.external_id, user, data.mode, info.user.mode_type)
         Registry.register(Supavisor.Registry.TenantClients, id, [])

lib/supavisor/db_handler.ex

@@ -72,7 +72,7 @@
 
         case try_ssl_handshake({:gen_tcp, sock}, auth) do
           {:ok, sock} ->
             case send_startup(sock, auth) do
               :ok ->
                 :ok = activate(sock)
                 {:next_state, :authentication, %{data | sock: sock}}
@@ -225,7 +225,7 @@
 
   def handle_event(:internal, :check_buffer, :idle, %{buffer: buff} = data) do
     if buff != [] do
-      Logger.warning("Buffer is not empty, try to send #{IO.iodata_length(buff)} bytes")
+      Logger.info("Buffer is not empty, try to send #{IO.iodata_length(buff)} bytes")
       buff = Enum.reverse(buff)
       :ok = sock_send(data.sock, buff)
     end
@@ -252,7 +252,7 @@
   end
 
   def handle_event({:call, {pid, _} = from}, {:db_call, bin}, state, %{buffer: buff} = data) do
-    Logger.warning(
+    Logger.info(
       "state #{state} <-- <-- bin #{inspect(byte_size(bin))} bytes, caller: #{inspect(pid)}"
     )
 
@@ -329,6 +329,7 @@
           [
             verify: :verify_peer,
             cacerts: [auth.upstream_tls_ca],
+            server_name_indication: String.to_charlist(auth.host),
             customize_hostname_check: [{:match_fun, fn _, _ -> true end}]
           ]
 

lib/supavisor/helpers.ex

@@ -29,6 +29,7 @@
           [
             {:verify, :verify_peer},
             {:cacerts, [upstream_cert(params["upstream_tls_ca"])]},
+            {:server_name_indication, String.to_charlist(params["db_host"])},
             {:customize_hostname_check, [{:match_fun, fn _, _ -> true end}]}
           ]
         end
@@ -51,7 +52,7 @@
         Postgrex.query(conn, "select version()", [])
         |> case do
           {:ok, %{rows: [[version]]}} ->
             if !params["require_user"] do
               case get_user_secret(conn, params["auth_query"], user["db_user"]) do
                 {:ok, _} ->
                   {:halt, {:ok, version}}

lib/supavisor/tenants.ex

@@ -43,6 +43,19 @@ defmodule Supavisor.Tenants do
     Tenant |> Repo.get_by(external_id: external_id) |> Repo.preload(:users)
   end
 
+  @spec get_user_cache(String.t(), String.t() | nil, String.t() | nil) ::
+          {:ok, map()} | {:error, any()}
+  def get_user_cache(user, external_id, sni_hostname) do
+    cache_key = {:user_cache, user, external_id, sni_hostname}
+
+    case Cachex.fetch(Supavisor.Cache, cache_key, fn _key ->
+           {:commit, {:cached, get_user(user, external_id, sni_hostname)}, ttl: 5_000}
+         end) do
+      {_, {:cached, value}} -> value
+      {_, {:cached, value}, _} -> value
+    end
+  end
+
   @spec get_user(String.t(), String.t() | nil, String.t() | nil) ::
           {:ok, map()} | {:error, any()}
   def get_user(_, nil, nil) do