Add PGHero to the demo app

.github/workflows/test.yml

@@ -28,8 +28,8 @@ jobs:
       - name: Run linter
         run: bin/lint --nofix
 
-  development_environment:
-    name: Development Environment
+  development_demo:
+    name: Tests for Development and Demo
     runs-on: ubuntu-latest
     timeout-minutes: 10
     env:
@@ -39,6 +39,7 @@ jobs:
       BUNDLE_RETRY: 3
       BUNDLE_PATH: vendor/bundle
       DISABLE_SPRING: 1
+      TEST_DEMO: 1
     services:
       postgres:
         image: postgres:15

Gemfile

@@ -26,6 +26,9 @@ gem 'nokogiri'
 gem 'pg', platforms: [:mri, :mingw, :x64_mingw]
 gem 'rack', '~> 2.2'
 gem 'rails'
+gem "sprockets-rails"
+# PGHero for the demo app
+gem 'pghero'
 
 platforms :ruby do
   gem "bootsnap"

Gemfile.lock

@@ -288,6 +288,8 @@ GEM
       ast (~> 2.4.1)
       racc
     pg (1.5.4)
+    pghero (3.4.1)
+      activerecord (>= 6)
     prettier_print (1.2.1)
     prism (0.24.0)
     protocol-hpack (1.4.2)
@@ -452,6 +454,13 @@ GEM
       thor (>= 0.19.2)
     spoon (0.0.6)
       ffi
+    sprockets (4.2.1)
+      concurrent-ruby (~> 1.0)
+      rack (>= 2.2.4, < 4)
+    sprockets-rails (3.4.2)
+      actionpack (>= 5.2)
+      activesupport (>= 5.2)
+      sprockets (>= 3.0.0)
     stackprof (0.2.25)
     stringio (3.1.0)
     syntax_tree (6.2.0)
@@ -528,6 +537,7 @@ DEPENDENCIES
   net-smtp
   nokogiri
   pg
+  pghero
   pry-byebug
   pry-rails
   puma
@@ -545,6 +555,7 @@ DEPENDENCIES
   sorbet
   sorbet-runtime
   spoom
+  sprockets-rails
   stackprof
   tapioca
   timecop

demo/app/jobs/pg_hero_maintenance_job.rb

@@ -0,0 +1,15 @@
+class PgHeroMaintenanceJob < ApplicationJob
+  include GoodJob::ActiveJobExtensions::Concurrency
+
+  good_job_control_concurrency_with(
+    key: "pg_hero_maintenance",
+    total_limit: 1
+  )
+
+  discard_on StandardError
+
+  def perform
+    PgHero.capture_query_stats
+    PgHero.clean_query_stats
+  end
+end

demo/config/initializers/good_job.rb

@@ -82,6 +82,11 @@
       complex_schedule: {
         cron: -> (last_ran) { last_ran ? last_ran + 17.hours : Time.now},
         class: "OtherJob",
+      },
+      pg_hero_maintenance: {
+        cron: "*/10 * * * *", # Every 10 minutes
+        class: "PgHeroMaintenanceJob",
+        description: "Runs PG Hero maintenance",
       }
     }
   end

demo/config/routes.rb

@@ -3,4 +3,6 @@
   mount GoodJob::Engine => "/good_job"
 
   get :create_job, to: 'application#create_job'
+
+  mount PgHero::Engine, at: "pghero"
 end

demo/db/migrate/20240321162554_create_pghero_query_stats.rb

@@ -0,0 +1,17 @@
+class CreatePgheroQueryStats < ActiveRecord::Migration[7.1]
+  def change
+    enable_extension "pg_stat_statements"
+
+    create_table :pghero_query_stats do |t|
+      t.text :database
+      t.text :user
+      t.text :query
+      t.integer :query_hash, limit: 8
+      t.float :total_time
+      t.integer :calls, limit: 8
+      t.timestamp :captured_at
+    end
+
+    add_index :pghero_query_stats, [:database, :captured_at]
+  end
+end

demo/db/schema.rb

@@ -10,8 +10,9 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema.define(version: 2024_01_14_221613) do
+ActiveRecord::Schema.define(version: 2024_03_21_162554) do
   # These are extensions that must be enabled in order to support this database
+  enable_extension "pg_stat_statements"
   enable_extension "pgcrypto"
   enable_extension "plpgsql"
 
@@ -94,4 +95,15 @@
     t.index ["scheduled_at"], name: "index_good_jobs_on_scheduled_at", where: "(finished_at IS NULL)"
   end
 
+  create_table "pghero_query_stats", force: :cascade do |t|
+    t.text "database"
+    t.text "user"
+    t.text "query"
+    t.bigint "query_hash"
+    t.float "total_time"
+    t.bigint "calls"
+    t.datetime "captured_at", precision: nil
+    t.index ["database", "captured_at"], name: "index_pghero_query_stats_on_database_and_captured_at"
+  end
+
 end