debian-mirror-gitlab/app/controllers/projects/logs_controller.rb

102 lines
2.5 KiB
Ruby
Raw Normal View History

2020-04-08 14:13:33 +05:30
# frozen_string_literal: true
module Projects
class LogsController < Projects::ApplicationController
2020-07-28 23:09:34 +05:30
include ::Gitlab::Utils::StrongMemoize
2020-04-08 14:13:33 +05:30
before_action :authorize_read_pod_logs!
before_action :ensure_deployments, only: %i(k8s elasticsearch)
2021-01-03 14:25:43 +05:30
feature_category :logging
2020-04-08 14:13:33 +05:30
def index
2020-07-28 23:09:34 +05:30
if environment || cluster
2020-04-08 14:13:33 +05:30
render :index
2020-07-28 23:09:34 +05:30
else
render :empty_logs
2020-04-08 14:13:33 +05:30
end
end
def k8s
render_logs(::PodLogs::KubernetesService, k8s_params)
end
def elasticsearch
render_logs(::PodLogs::ElasticsearchService, elasticsearch_params)
end
private
def render_logs(service, permitted_params)
::Gitlab::UsageCounters::PodLogs.increment(project.id)
::Gitlab::PollingInterval.set_header(response, interval: 3_000)
result = service.new(cluster, namespace, params: permitted_params).execute
if result.nil?
head :accepted
elsif result[:status] == :success
render json: result
else
render status: :bad_request, json: result
end
end
2020-07-28 23:09:34 +05:30
# cluster is selected either via environment or directly by id
def cluster_params
params.permit(:environment_name, :cluster_id)
2020-04-08 14:13:33 +05:30
end
def k8s_params
params.permit(:container_name, :pod_name)
end
def elasticsearch_params
2020-04-22 19:07:51 +05:30
params.permit(:container_name, :pod_name, :search, :start_time, :end_time, :cursor)
2020-04-08 14:13:33 +05:30
end
def environment
2020-07-28 23:09:34 +05:30
strong_memoize(:environment) do
if cluster_params.key?(:environment_name)
EnvironmentsFinder.new(project, current_user, name: cluster_params[:environment_name]).find.first
else
project.default_environment
end
end
2020-04-08 14:13:33 +05:30
end
def cluster
2020-07-28 23:09:34 +05:30
strong_memoize(:cluster) do
if gitlab_managed_apps_logs?
clusters = ClusterAncestorsFinder.new(project, current_user).execute
clusters.find { |cluster| cluster.id == cluster_params[:cluster_id].to_i }
else
environment&.deployment_platform&.cluster
end
end
2020-04-08 14:13:33 +05:30
end
def namespace
2020-07-28 23:09:34 +05:30
if gitlab_managed_apps_logs?
Gitlab::Kubernetes::Helm::NAMESPACE
else
environment.deployment_namespace
end
2020-04-08 14:13:33 +05:30
end
def ensure_deployments
2020-07-28 23:09:34 +05:30
return if gitlab_managed_apps_logs?
2020-04-08 14:13:33 +05:30
return if cluster && namespace.present?
render status: :bad_request, json: {
status: :error,
message: _('Environment does not have deployments')
}
end
2020-07-28 23:09:34 +05:30
def gitlab_managed_apps_logs?
cluster_params.key?(:cluster_id)
end
2020-04-08 14:13:33 +05:30
end
end