diff options
Diffstat (limited to 'lib/gitlab/kas/client.rb')
-rw-r--r-- | lib/gitlab/kas/client.rb | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/lib/gitlab/kas/client.rb b/lib/gitlab/kas/client.rb index 842ee98e4da..753a185344e 100644 --- a/lib/gitlab/kas/client.rb +++ b/lib/gitlab/kas/client.rb @@ -7,6 +7,7 @@ module Gitlab JWT_AUDIENCE = 'gitlab-kas' STUB_CLASSES = { + agent_tracker: Gitlab::Agent::AgentTracker::Rpc::AgentTracker::Stub, configuration_project: Gitlab::Agent::ConfigurationProject::Rpc::ConfigurationProject::Stub }.freeze @@ -17,6 +18,15 @@ module Gitlab raise ConfigurationError, 'KAS internal URL is not configured' unless Gitlab::Kas.internal_url.present? end + def get_connected_agents(project:) + request = Gitlab::Agent::AgentTracker::Rpc::GetConnectedAgentsRequest.new(project_id: project.id) + + stub_for(:agent_tracker) + .get_connected_agents(request, metadata: metadata) + .agents + .to_a + end + def list_agent_config_files(project:) request = Gitlab::Agent::ConfigurationProject::Rpc::ListAgentConfigFilesRequest.new( repository: repository(project), @@ -49,7 +59,7 @@ module Gitlab end def kas_endpoint_url - Gitlab::Kas.internal_url.sub(%r{^grpc://|^grpcs://}, '') + Gitlab::Kas.internal_url.sub(%r{^grpcs?://}, '') end def credentials |