Flush all in case of sigterm

Author: ashishjha-96

lib/markdoc/application.ex

@@ -31,6 +31,24 @@ defmodule Markdoc.Application do
     Supervisor.start_link(children, opts)
   end
 
+  @impl true
+  def stop(_state) do
+    doc_ids =
+      Registry.select(Markdoc.DocRegistry, [
+        {{:"$1", :_, :_}, [], [:"$1"]}
+      ])
+
+    Enum.each(doc_ids, fn doc_id ->
+      try do
+        Markdoc.DocServer.flush(doc_id)
+      catch
+        _ -> :ok
+      end
+    end)
+
+    :ok
+  end
+
   # Tell Phoenix to update the endpoint configuration
   # whenever the application is updated.
   @impl true

lib/markdoc/doc_server.ex

@@ -100,6 +100,13 @@ defmodule Markdoc.DocServer do
     GenServer.cast(Markdoc.DocRegistry.via_tuple(doc_id), {:save_snapshot, snapshot_binary})
   end
 
+  @doc """
+  Forces a flush to the configured storage backend.
+  """
+  def flush(doc_id) when is_binary(doc_id) do
+    GenServer.call(Markdoc.DocRegistry.via_tuple(doc_id), :flush)
+  end
+
   ## Server Callbacks
 
   @impl true
@@ -277,6 +284,12 @@ defmodule Markdoc.DocServer do
     {:reply, state.history, state}
   end
 
+  @impl true
+  def handle_call(:flush, _from, state) do
+    new_state = maybe_flush(state, :manual)
+    {:reply, :ok, new_state}
+  end
+
   @impl true
   def handle_info(:check_inactivity, state) do
     # Periodic check for inactivity