Disable ETS persistence for Agent and CLI hubs

Author: aleDsz

lib/livebook.ex

@@ -157,6 +157,10 @@ defmodule Livebook do
       config :livebook, teams_url: url, warn_on_live_teams_server: false
     end
 
+    if System.get_env("LIVEBOOK_TEAMS_AUTH") do
+      config :livebook, :persist_storage, false
+    end
+
     if Livebook.Config.boolean!("LIVEBOOK_SHUTDOWN_ENABLED", false) do
       config :livebook, :shutdown_callback, {System, :stop, []}
     end

lib/livebook/storage.ex

@@ -186,10 +186,17 @@ defmodule Livebook.Storage do
     # in case it is persisting to disk. terminate/2 is still a no-op.
     Process.flag(:trap_exit, true)
 
-    table = load_or_create_table()
-    :persistent_term.put(__MODULE__, table)
+    persist_storage? = Application.get_env(:livebook, :persist_storage, true)
+
+    table =
+      if persist_storage? do
+        load_or_create_table()
+      else
+        :ets.new(__MODULE__, [:protected, :duplicate_bag])
+      end
 
-    {:ok, %{table: table}}
+    :persistent_term.put(__MODULE__, table)
+    {:ok, %{table: table, persist?: persist_storage?}}
   end
 
   @impl true
@@ -220,6 +227,10 @@ defmodule Livebook.Storage do
   end
 
   @impl true
+  def handle_continue(:save_to_file, %{persist?: false} = state) do
+    {:noreply, state}
+  end
+
   def handle_continue(:save_to_file, %{table: table} = state) do
     file_path = String.to_charlist(config_file_path())
     :ok = :ets.tab2file(table, file_path)

lib/livebook_cli.ex

@@ -21,6 +21,7 @@ defmodule LivebookCLI do
     extract_priv!()
 
     :ok = Application.load(:livebook)
+    Application.put_env(:livebook, :persist_storage, false)
 
     if unix?() do
       Application.put_env(:elixir, :ansi_enabled, true)

test/test_helper.exs

@@ -38,6 +38,9 @@ Application.put_env(:livebook, Livebook.Runtime.Embedded,
   load_packages: {Livebook.Runtime.Embedded.Packages, :list, []}
 )
 
+# Disable ETS disk persistence
+Application.put_env(:livebook, :persist_storage, false)
+
 # Disable autosaving
 Livebook.Storage.insert(:settings, "global", autosave_path: nil)