From ef3e963c0c8532dcc2d2c132138e7e220db17b9f Mon Sep 17 00:00:00 2001
From: Tom Hughes <tom@compton.nu>
Date: Fri, 15 Jan 2021 17:46:55 +0000
Subject: [PATCH] Add promscale for long term storage of prometheus data

---
 cookbooks/prometheus/metadata.rb              |   1 +
 cookbooks/prometheus/recipes/server.rb        | 107 ++++++++++++++++++
 .../templates/default/prometheus.yml.erb      |   6 +
 .../serverspec/promscale_spec.rb              |  13 +++
 4 files changed, 127 insertions(+)
 create mode 100644 test/integration/prometheus-server/serverspec/promscale_spec.rb

diff --git a/cookbooks/prometheus/metadata.rb b/cookbooks/prometheus/metadata.rb
index e6173a037..2ac7f9bf7 100644
--- a/cookbooks/prometheus/metadata.rb
+++ b/cookbooks/prometheus/metadata.rb
@@ -10,3 +10,4 @@ depends           "apache"
 depends           "apt"
 depends           "git"
 depends           "networking"
+depends           "timescaledb"
diff --git a/cookbooks/prometheus/recipes/server.rb b/cookbooks/prometheus/recipes/server.rb
index f3ff4d9ca..6e112a000 100644
--- a/cookbooks/prometheus/recipes/server.rb
+++ b/cookbooks/prometheus/recipes/server.rb
@@ -20,6 +20,7 @@
 include_recipe "apache"
 include_recipe "apt"
 include_recipe "networking"
+include_recipe "timescaledb"
 
 passwords = data_bag_item("prometheus", "passwords")
 tokens = data_bag_item("prometheus", "tokens")
@@ -33,6 +34,112 @@ end
 
 package "prometheus"
 
+promscale_version = "0.1.4"
+
+database_cluster = node[:timescaledb][:cluster]
+
+postgresql_user "prometheus" do
+  cluster database_cluster
+  createrole true
+end
+
+postgresql_database "promscale" do
+  cluster database_cluster
+  owner "prometheus"
+end
+
+postgresql_extension "timescaledb" do
+  cluster database_cluster
+  database "promscale"
+end
+
+directory "/opt/promscale" do
+  owner "root"
+  group "root"
+  mode "755"
+end
+
+package %w[
+  make
+  gcc
+  clang-9
+  llvm-9
+  cargo
+]
+
+git "/opt/promscale/extension" do
+  action :sync
+  repository "https://github.com/timescale/promscale_extension.git"
+  revision "0.1.1"
+  user "root"
+  group "root"
+end
+
+execute "/opt/promscale/extension/Makefile" do
+  action :nothing
+  command "make install"
+  cwd "/opt/promscale/extension"
+  user "root"
+  group "root"
+  subscribes :run, "git[/opt/promscale/extension]", :immediate
+  notifies :restart, "service[postgresql]", :immediate
+end
+
+directory "/opt/promscale/bin" do
+  owner "root"
+  group "root"
+  mode "755"
+end
+
+remote_file "/opt/promscale/bin/promscale" do
+  action :create
+  source "https://github.com/timescale/promscale/releases/download/#{promscale_version}/promscale_#{promscale_version}_Linux_x86_64"
+  owner "root"
+  group "root"
+  mode "755"
+end
+
+systemd_service "promscale" do
+  description "Promscale Connector"
+  type "simple"
+  user "prometheus"
+  exec_start "/opt/promscale/bin/promscale --db-host /run/postgresql --db-port 5432 --db-user prometheus --db-name promscale --db-connections-max 400"
+#  exec_start lazy { "/opt/promscale/bin/promscale --db-host /run/postgresql --db-port #{node[:postgresql][:clusters][database_cluster][:port]} --db-user prometheus --db-name promscale --db-max-connections 400" }
+  private_tmp true
+  protect_system "strict"
+  protect_home true
+  no_new_privileges true
+end
+
+service "promscale" do
+  action [:enable, :start]
+  subscribes :restart, "remote_file[/opt/promscale/bin/promscale]"
+  subscribes :restart, "systemd_service[promscale]"
+end
+
+postgresql_extension "promscale" do
+  cluster database_cluster
+  database "promscale"
+  notifies :restart, "service[promscale]"
+end
+
+systemd_service "promscale-maintenance" do
+  description "Promscale Maintenace"
+  type "simple"
+  user "prometheus"
+  exec_start "/usr/bin/psql --command='CALL prom_api.execute_maintenance()' promscale"
+  private_tmp true
+  protect_system "strict"
+  protect_home true
+  no_new_privileges true
+end
+
+systemd_timer "promscale-maintenance" do
+  description "Promscale Maintenace"
+  on_active_sec 1800
+  on_unit_inactive_sec 1800
+end
+
 jobs = {}
 
 search(:node, "recipes:prometheus\\:\\:default").sort_by(&:name).each do |client|
diff --git a/cookbooks/prometheus/templates/default/prometheus.yml.erb b/cookbooks/prometheus/templates/default/prometheus.yml.erb
index 46d241905..591c1e0ec 100644
--- a/cookbooks/prometheus/templates/default/prometheus.yml.erb
+++ b/cookbooks/prometheus/templates/default/prometheus.yml.erb
@@ -21,3 +21,9 @@ scrape_configs:
           instance: <%= target[:name].split(".").first %>
 <% end -%>
 <% end -%>
+
+remote_write:
+  - url: "http://localhost:9201/write"
+
+remote_read:
+  - url: "http://localhost:9201/read"
diff --git a/test/integration/prometheus-server/serverspec/promscale_spec.rb b/test/integration/prometheus-server/serverspec/promscale_spec.rb
new file mode 100644
index 000000000..aaedf15f6
--- /dev/null
+++ b/test/integration/prometheus-server/serverspec/promscale_spec.rb
@@ -0,0 +1,13 @@
+require "serverspec"
+
+# Required by serverspec
+set :backend, :exec
+
+describe service("promscale") do
+  it { should be_enabled }
+  it { should be_running }
+end
+
+describe port(9201) do
+  it { should be_listening.with("tcp6") }
+end
-- 
2.39.5