chore: set max heap size for handlers (#280)

VERSION

@@ -1 +1 @@
-1.1.20
+1.1.21

lib/supavisor/client_handler.ex

@@ -35,6 +35,7 @@ defmodule Supavisor.ClientHandler do
 
   def init(ref, trans, opts) do
     Process.flag(:trap_exit, true)
+    H.set_max_heap_size(150)
 
     {:ok, sock} = :ranch.handshake(ref)
     :ok = trans.setopts(sock, active: true)

lib/supavisor/db_handler.ex

@@ -35,6 +35,8 @@ defmodule Supavisor.DbHandler do
   @impl true
   def init(args) do
     Process.flag(:trap_exit, true)
+    H.set_max_heap_size(150)
+
     {_, tenant} = args.tenant
     Logger.metadata(project: tenant, user: args.user, mode: args.mode)
 

lib/supavisor/helpers.ex

@@ -330,4 +330,17 @@ defmodule Supavisor.Helpers do
       kind, reason -> {:error, {:badrpc, {kind, reason}}}
     end
   end
+
+  @doc """
+  Sets the maximum heap size for the current process. The `max_heap_size` parameter is in megabytes.
+
+  ## Parameters
+
+  - `max_heap_size`: The maximum heap size in megabytes.
+  """
+  @spec set_max_heap_size(pos_integer()) :: map()
+  def set_max_heap_size(max_heap_size) do
+    max_heap_words = div(max_heap_size * 1024 * 1024, :erlang.system_info(:wordsize))
+    Process.flag(:max_heap_size, %{size: max_heap_words})
+  end
 end