fix: alter id to uuid type (#1199)

Changes the existing ID to UUID type.

lib/realtime/api/message.ex

@@ -5,15 +5,15 @@ defmodule Realtime.Api.Message do
   use Ecto.Schema
   import Ecto.Changeset
 
+  @primary_key {:id, Ecto.UUID, autogenerate: true}
   @schema_prefix "realtime"
 
   schema "messages" do
-    field :uuid, :string
-    field :topic, :string
-    field :extension, Ecto.Enum, values: [:broadcast, :presence]
-    field :payload, :map
-    field :event, :string
-    field :private, :boolean
+    field(:topic, :string)
+    field(:extension, Ecto.Enum, values: [:broadcast, :presence])
+    field(:payload, :map)
+    field(:event, :string)
+    field(:private, :boolean)
 
     timestamps()
   end
@@ -27,8 +27,7 @@ defmodule Realtime.Api.Message do
       :event,
       :private,
       :inserted_at,
-      :updated_at,
-      :uuid
+      :updated_at
     ])
     |> validate_required([:topic, :extension])
     |> put_timestamp(:updated_at)

lib/realtime/broadcast_changes/handler.ex

@@ -285,7 +285,7 @@ defmodule Realtime.BroadcastChanges.Handler do
             {:noreply, state}
 
           payload ->
-            id = Map.fetch!(to_broadcast, "uuid")
+            id = Map.fetch!(to_broadcast, "id")
 
             to_broadcast =
               %{

lib/realtime/tenants.ex

@@ -284,7 +284,7 @@ defmodule Realtime.Tenants do
     |> Cache.get_tenant_by_external_id()
     |> Tenant.management_changeset(attrs)
     |> Repo.update!()
-    |> tap(fn _ -> Cache.distributed_invalidate_tenant_cache(tenant_id) end)
+    |> tap(fn _ -> Cache.invalidate_tenant_cache(tenant_id) end)
   end
 
   defp broadcast_operation_event(action, external_id) do

lib/realtime/tenants/migrations.ex

@@ -63,7 +63,8 @@ defmodule Realtime.Tenants.Migrations do
     AddPayloadToMessages,
     ChangeMessagesIdType,
     UuidAutoGeneration,
-    MessagesPartitioning
+    MessagesPartitioning,
+    MessagesUsingUuid
   }
 
   @migrations [
@@ -117,7 +118,8 @@ defmodule Realtime.Tenants.Migrations do
     {20_240_919_163_303, AddPayloadToMessages},
     {20_240_919_163_305, ChangeMessagesIdType},
     {20_241_019_105_805, UuidAutoGeneration},
-    {20_241_030_150_047, MessagesPartitioning}
+    {20_241_030_150_047, MessagesPartitioning},
+    {20_241_108_114_728, MessagesUsingUuid}
   ]
   defstruct [:tenant_external_id, :settings]
   @spec run_migrations(map()) :: :ok | {:error, any()}

lib/realtime/tenants/repo/migrations/20241108114728_messages_using_uuid.ex

@@ -0,0 +1,15 @@
+defmodule Realtime.Tenants.Migrations.MessagesUsingUuid do
+  @moduledoc false
+  use Ecto.Migration
+
+  def change do
+    alter table(:messages) do
+      remove(:id)
+      remove(:uuid)
+      add(:id, :uuid, null: false, default: fragment("gen_random_uuid()"))
+    end
+
+    execute("ALTER TABLE realtime.messages ADD PRIMARY KEY (id, inserted_at)")
+    execute("DROP SEQUENCE realtime.messages_id_seq")
+  end
+end

mix.exs

@@ -4,7 +4,7 @@ defmodule Realtime.MixProject do
   def project do
     [
       app: :realtime,
-      version: "2.33.26",
+      version: "2.33.27",
       elixir: "~> 1.16.0",
       elixirc_paths: elixirc_paths(Mix.env()),
       start_permanent: Mix.env() == :prod,

test/realtime/tenants_test.exs

@@ -82,8 +82,8 @@ defmodule Realtime.TenantsTest do
     test "sets private_only flag to true and invalidates cache" do
       %{external_id: external_id} = tenant_fixture(%{private_only: false})
       tenant = Tenants.update_management(external_id, %{private_only: true})
-      assert tenant.private_only == true
-      assert_receive {:invalidate_cache, ^external_id}, 1000
+      assert tenant.private_only
+      assert Tenants.Cache.get_tenant_by_external_id(external_id).private_only
     end
   end
 end