Check if deployment group belongs to app server

This commit is contained in:
Alexandre de Souza 2025-05-08 16:27:11 -03:00
parent c1dc817122
commit afdc6f080c
No known key found for this signature in database
GPG key ID: E39228FFBA346545

View file

@ -166,6 +166,14 @@ defmodule Livebook.Hubs.TeamClient do
GenServer.call(registry_name(id), :authorization_groups_enabled?)
end
@doc """
Returns if the Team client is an app server for given deployment group.
"""
@spec current_app_server_deployment_group?(String.t(), String.t()) :: boolean()
def current_app_server_deployment_group?(id, deployment_group_id) do
GenServer.call(registry_name(id), {:current_deployment_group?, deployment_group_id})
end
@doc """
Returns if the Team client is connected.
"""
@ -346,6 +354,10 @@ defmodule Livebook.Hubs.TeamClient do
end
end
def handle_call({:current_deployment_group?, id}, _caller, state) do
{:reply, state.deployment_group_id == id, state}
end
@impl true
def handle_info(:connected, state) do
Hubs.Broadcasts.hub_connected(state.hub.id)