usage_data.rb 2.63 KB
Newer Older
1 2
module Gitlab
  class UsageData
3 4
    include Gitlab::CurrentSettings

5
    class << self
6 7
      def data(force_refresh: false)
        Rails.cache.fetch('usage_data', force: force_refresh, expires_in: 2.weeks) { uncached_data }
8 9 10 11 12 13
      end

      def uncached_data
        license_usage_data.merge(system_usage_data)
      end

14 15
      def to_json(force_refresh: false)
        data(force_refresh: force_refresh).to_json
16 17 18 19 20 21 22
      end

      def system_usage_data
        {
          counts: {
            boards: Board.count,
            ci_builds: ::Ci::Build.count,
23 24
            ci_internal_pipelines: ::Ci::Pipeline.internal.count,
            ci_external_pipelines: ::Ci::Pipeline.external.count,
25 26
            ci_runners: ::Ci::Runner.count,
            ci_triggers: ::Ci::Trigger.count,
27
            ci_pipeline_schedules: ::Ci::PipelineSchedule.count,
28 29
            deploy_keys: DeployKey.count,
            deployments: Deployment.count,
30 31
            environments: ::Environment.count,
            in_review_folder: ::Environment.in_review_folder.count,
32 33 34 35 36 37 38 39 40 41
            groups: Group.count,
            issues: Issue.count,
            keys: Key.count,
            labels: Label.count,
            lfs_objects: LfsObject.count,
            merge_requests: MergeRequest.count,
            milestones: Milestone.count,
            notes: Note.count,
            pages_domains: PagesDomain.count,
            projects: Project.count,
42
            projects_imported_from_github: Project.where(import_type: 'github').count,
43 44 45 46
            protected_branches: ProtectedBranch.count,
            releases: Release.count,
            snippets: Snippet.count,
            todos: Todo.count,
47
            uploads: Upload.count,
48
            web_hooks: WebHook.count
49
          }.merge(services_usage)
50 51 52 53
        }
      end

      def license_usage_data
Sean McGivern's avatar
Sean McGivern committed
54 55
        usage_data = {
          uuid: current_application_settings.uuid,
Sean McGivern's avatar
Sean McGivern committed
56
          hostname: Gitlab.config.gitlab.host,
Sean McGivern's avatar
Sean McGivern committed
57 58 59 60 61 62
          version: Gitlab::VERSION,
          active_user_count: User.active.count,
          recorded_at: Time.now,
          mattermost_enabled: Gitlab.config.mattermost.enabled,
          edition: 'CE'
        }
63 64 65

        usage_data
      end
66 67 68 69 70 71 72 73 74 75 76 77

      def services_usage
        types = {
          JiraService: :projects_jira_active,
          SlackService: :projects_slack_notifications_active,
          SlackSlashCommandsService: :projects_slack_slash_active,
          PrometheusService: :projects_prometheus_active
        }

        results = Service.unscoped.where(type: types.keys, active: true).group(:type).count
        results.each_with_object({}) { |(key, value), response| response[types[key.to_sym]] = value  }
      end
78 79 80
    end
  end
end