mirror of
https://github.com/penpot/penpot.git
synced 2025-03-17 02:01:32 -05:00
138 lines
4.6 KiB
Clojure
138 lines
4.6 KiB
Clojure
;; This Source Code Form is subject to the terms of the Mozilla Public
|
|
;; License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
;; file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
;;
|
|
;; This Source Code Form is "Incompatible With Secondary Licenses", as
|
|
;; defined by the Mozilla Public License, v. 2.0.
|
|
;;
|
|
;; Copyright (c) 2020 UXBOX Labs SL
|
|
|
|
(ns app.tasks.telemetry
|
|
"A task that is reponsible to collect anonymous statistical
|
|
information about the current instance and send it to the telemetry
|
|
server."
|
|
(:require
|
|
[app.common.exceptions :as ex]
|
|
[app.common.spec :as us]
|
|
[app.config :as cfg]
|
|
[app.db :as db]
|
|
[app.util.http :as http]
|
|
[app.util.json :as json]
|
|
[clojure.spec.alpha :as s]
|
|
[integrant.core :as ig]))
|
|
|
|
(declare handler)
|
|
(declare acquire-lock)
|
|
(declare release-all-locks)
|
|
(declare retrieve-stats)
|
|
|
|
(s/def ::version ::us/string)
|
|
(s/def ::uri ::us/string)
|
|
(s/def ::instance-id ::us/uuid)
|
|
(s/def ::sprops
|
|
(s/keys :req-un [::instance-id]))
|
|
|
|
|
|
(defmethod ig/pre-init-spec ::handler [_]
|
|
(s/keys :req-un [::db/pool ::version ::uri ::sprops]))
|
|
|
|
(defmethod ig/init-key ::handler
|
|
[_ {:keys [pool] :as cfg}]
|
|
(fn [_]
|
|
(db/with-atomic [conn pool]
|
|
(try
|
|
(acquire-lock conn)
|
|
(handler (assoc cfg :conn conn))
|
|
(finally
|
|
(release-all-locks conn))))))
|
|
|
|
(defn- acquire-lock
|
|
[conn]
|
|
(db/exec-one! conn ["select pg_advisory_lock(87562985867332);"]))
|
|
|
|
(defn- release-all-locks
|
|
[conn]
|
|
(db/exec-one! conn ["select pg_advisory_unlock_all();"]))
|
|
|
|
(defn- handler
|
|
[{:keys [sprops] :as cfg}]
|
|
(let [instance-id (:instance-id sprops)
|
|
data (retrieve-stats cfg)
|
|
data (assoc data :instance-id instance-id)
|
|
response (http/send! {:method :post
|
|
:uri (:uri cfg)
|
|
:headers {"content-type" "application/json"}
|
|
:body (json/encode-str data)})]
|
|
|
|
(when (not= 200 (:status response))
|
|
(ex/raise :type :internal
|
|
:code :invalid-response-from-google
|
|
:context {:status (:status response)
|
|
:body (:body response)}))))
|
|
|
|
(defn retrieve-num-teams
|
|
[conn]
|
|
(-> (db/exec-one! conn ["select count(*) as count from team;"]) :count))
|
|
|
|
(defn retrieve-num-projects
|
|
[conn]
|
|
(-> (db/exec-one! conn ["select count(*) as count from project;"]) :count))
|
|
|
|
(defn retrieve-num-files
|
|
[conn]
|
|
(-> (db/exec-one! conn ["select count(*) as count from project;"]) :count))
|
|
|
|
(def sql:team-averages
|
|
"with projects_by_team as (
|
|
select t.id, count(p.id) as num_projects
|
|
from team as t
|
|
left join project as p on (p.team_id = t.id)
|
|
group by 1
|
|
), files_by_project as (
|
|
select p.id, count(f.id) as num_files
|
|
from project as p
|
|
left join file as f on (f.project_id = p.id)
|
|
group by 1
|
|
), comment_threads_by_file as (
|
|
select f.id, count(ct.id) as num_comment_threads
|
|
from file as f
|
|
left join comment_thread as ct on (ct.file_id = f.id)
|
|
group by 1
|
|
), users_by_team as (
|
|
select t.id, count(tp.profile_id) as num_users
|
|
from team as t
|
|
left join team_profile_rel as tp on(tp.team_id = t.id)
|
|
where t.is_default = false
|
|
group by 1
|
|
)
|
|
select (select avg(num_projects)::integer from projects_by_team) as avg_projects_on_team,
|
|
(select max(num_projects)::integer from projects_by_team) as max_projects_on_team,
|
|
(select avg(num_files)::integer from files_by_project) as avg_files_on_project,
|
|
(select max(num_files)::integer from files_by_project) as max_files_on_project,
|
|
(select avg(num_comment_threads)::integer from comment_threads_by_file) as avg_comment_threads_on_file,
|
|
(select max(num_comment_threads)::integer from comment_threads_by_file) as max_comment_threads_on_file,
|
|
(select avg(num_users)::integer from users_by_team) as avg_users_on_team,
|
|
(select max(num_users)::integer from users_by_team) as max_users_on_team;")
|
|
|
|
(defn retrieve-team-averages
|
|
[conn]
|
|
(->> [sql:team-averages]
|
|
(db/exec-one! conn)))
|
|
|
|
(defn retrieve-jvm-stats
|
|
[]
|
|
(let [^Runtime runtime (Runtime/getRuntime)]
|
|
{:jvm-heap-current (.totalMemory runtime)
|
|
:jvm-heap-max (.maxMemory runtime)
|
|
:jvm-cpus (.availableProcessors runtime)}))
|
|
|
|
(defn- retrieve-stats
|
|
[{:keys [conn version]}]
|
|
(merge
|
|
{:version version
|
|
:with-taiga (:telemetry-with-taiga cfg/config false)
|
|
:total-teams (retrieve-num-teams conn)
|
|
:total-projects (retrieve-num-projects conn)
|
|
:total-files (retrieve-num-files conn)}
|
|
(retrieve-team-averages conn)
|
|
(retrieve-jvm-stats)))
|