feat: add cookie cleaning for postgres identifiers

Logflare · Nov 15, 2023 · 3b01318 · 3b01318
1 parent 3433c22
commit 3b01318
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/logflare/cluster/postgres_strategy.ex b/lib/logflare/cluster/postgres_strategy.ex
@@ -34,7 +34,7 @@ defmodule Logflare.Cluster.PostgresStrategy do
     new_config =
       state.config
       |> Keyword.put_new(:heartbeat_interval, 5_000)
-      |> Keyword.put(:channel_name, Atom.to_string(Node.get_cookie()))
+      |> Keyword.put(:channel_name, clean_cookie(Node.get_cookie()))
       |> Keyword.put(:url, url)
 
     meta = %{
@@ -114,4 +114,9 @@ defmodule Logflare.Cluster.PostgresStrategy do
     database = Application.get_env(:logflare, Logflare.Repo)[:database]
     ~s|postgresql://#{username}:#{password}@#{hostname}:#{port}/#{database}|
   end
+
+defp clean_cookie(cookie) when is_atom(cookie), do: cookie |> Atom.to_string() |> clean_cookie()
+  defp clean_cookie(str) when is_binary(str) do
+    String.replace(str, ~r/\W/, "_") |> dbg()
+  end
 end