Integrate yabeda-rails and metrics

Author: anaumov

Gemfile

@@ -17,6 +17,9 @@ gem "stimulus-rails"
 # Build JSON APIs with ease [https://github.com/rails/jbuilder]
 gem "jbuilder"
 
+gem "yabeda-rails"
+gem "yabeda-prometheus"
+
 # Use Active Model has_secure_password [https://guides.rubyonrails.org/active_model_basics.html#securepassword]
 # gem "bcrypt", "~> 3.1.7"
 

Gemfile.lock

@@ -74,6 +74,8 @@ GEM
       uri (>= 0.13.1)
     addressable (2.8.7)
       public_suffix (>= 2.0.2, < 7.0)
+    anyway_config (2.7.2)
+      ruby-next-core (~> 1.0)
     ast (2.4.3)
     base64 (0.3.0)
     bcrypt_pbkdf (1.1.1)
@@ -104,6 +106,7 @@ GEM
       reline (>= 0.3.8)
     dotenv (3.1.8)
     drb (2.2.3)
+    dry-initializer (3.2.0)
     ed25519 (1.4.0)
     erb (5.0.3)
     erubi (1.13.1)
@@ -186,6 +189,8 @@ GEM
       prettyprint
     prettyprint (0.2.0)
     prism (1.4.0)
+    prometheus-client (4.2.5)
+      base64
     propshaft (1.3.1)
       actionpack (>= 7.0.0)
       activesupport (>= 7.0.0)
@@ -274,6 +279,7 @@ GEM
       rubocop (>= 1.72)
       rubocop-performance (>= 1.24)
       rubocop-rails (>= 2.30)
+    ruby-next-core (1.1.2)
     ruby-progressbar (1.13.0)
     rubyzip (3.1.1)
     securerandom (0.4.1)
@@ -342,6 +348,19 @@ GEM
     websocket-extensions (0.1.5)
     xpath (3.2.0)
       nokogiri (~> 1.8)
+    yabeda (0.14.0)
+      anyway_config (>= 1.0, < 3)
+      concurrent-ruby
+      dry-initializer
+    yabeda-prometheus (0.9.1)
+      prometheus-client (>= 3.0, < 5.0)
+      rack
+      yabeda (~> 0.10)
+    yabeda-rails (0.10.0)
+      activesupport
+      anyway_config (>= 1.3, < 3)
+      railties
+      yabeda (~> 0.8)
     zeitwerk (2.7.3)
 
 PLATFORMS
@@ -371,6 +390,8 @@ DEPENDENCIES
   turbo-rails
   tzinfo-data
   web-console
+  yabeda-prometheus
+  yabeda-rails
 
 BUNDLED WITH
    2.4.19

app/controllers/hello_controller.rb

@@ -0,0 +1,17 @@
+class HelloController < ApplicationController
+  def index
+    Yabeda.top_shop.index_page_opened.increment
+    render plain: "Главная страница"
+  end
+
+  def refresh_online
+    users_online_count = rand(100..1000)
+    Yabeda.top_shop.user_online.set({}, users_online_count)
+    render plain: "#{users_online_count} users online now"
+  end
+
+  def seed_everything
+    GenerateEverythingJob.perform_now
+    render plain: "Done!"
+  end
+end

app/jobs/generate_everything_job.rb

@@ -14,10 +14,10 @@ def seed_users
   end
 
   def seed_couriers
-    rand(0..1).times { User.create(kind: :courier) }
+    rand(1..2).times { User.create(kind: :courier) }
   end
 
   def seed_orders
-    rand(10..20).times { Order.create(amount: rand(100..5000)) }
+    rand(40..90).times { Order.create(amount: rand(100..5000)) }
   end
 end

app/models/order.rb

@@ -1,2 +1,10 @@
 class Order < ApplicationRecord
+  after_create :track_order
+
+  private
+
+  def track_order
+    Yabeda.top_shop.orders_placed.increment
+    Yabeda.top_shop.money_earned.increment({}, by: amount)
+  end
 end

app/models/user.rb

@@ -1,2 +1,9 @@
 class User < ApplicationRecord
+  after_create :track_user
+
+  private
+
+  def track_user
+    Yabeda.top_shop.users_registered.increment({ kind: })
+  end
 end

config/initializers/yabeda.rb

@@ -0,0 +1,10 @@
+Yabeda.configure do
+  group :top_shop do
+    counter :index_page_opened, comment: "Index page visits"
+    counter :users_registered, tags: %i[kind], comment: "Users registered"
+    counter :orders_placed, comment: "Orders placed"
+    counter :money_earned, comment: "Total GMV"
+
+    gauge :user_online, comment: "Users online"
+  end
+end

config/routes.rb

@@ -1,4 +1,5 @@
 Rails.application.routes.draw do
+  mount Yabeda::Prometheus::Exporter, at: "/metrics"
   # Define your application routes per the DSL in https://guides.rubyonrails.org/routing.html
 
   # Reveal health status on /up that returns 200 if the app boots with no exceptions, otherwise 500.
@@ -9,6 +10,9 @@
   # get "manifest" => "rails/pwa#manifest", as: :pwa_manifest
   # get "service-worker" => "rails/pwa#service_worker", as: :pwa_service_worker
 
+  get "seed" => "hello#seed_everything"
+  get "refresh_online" => "hello#refresh_online"
+
   # Defines the root path route ("/")
-  # root "posts#index"
+  root "hello#index"
 end