Allow to configure bucket size for histograms (#6)

Author: ryan-dyer-sp

Gemfile

@@ -36,3 +36,5 @@ gem "rspec", "~> 3.0"
 gem "rubocop", "~> 1.30"
 
 gem "debug"
+
+gem "anyway_config", ">= 1.3", "< 3.0"

Gemfile.lock

@@ -3,6 +3,7 @@ PATH
   specs:
     yabeda-activerecord (0.1.1)
       activerecord (>= 6.0)
+      anyway_config (>= 1.3, < 3.0)
       yabeda (~> 0.6)
 
 GEM
@@ -83,6 +84,7 @@ PLATFORMS
 
 DEPENDENCIES
   activerecord (~> 7.0)
+  anyway_config (>= 1.3, < 3.0)
   debug
   rake (~> 13.0)
   rspec (~> 3.0)

README.md

@@ -22,6 +22,14 @@ Launch/restart your application and that's it: metrics are being collected.
 
 To expose metrics Don't forget to also add one of [Yabeda adapters](https://github.com/yabeda-rb/yabeda#available-monitoring-system-adapters) to your Gemfile. 
 
+## Configuration
+
+Configuration is handled by [anyway_config] gem. With it you can load settings from environment variables (upcased and prefixed with `YABEDA_ACTIVERECORD_`), YAML files, and other sources. See [anyway_config] docs for details.
+
+| Config key | Type  | Default | Description                                 |
+| -----------| ----- | ------- | ------------------------------------------- |
+| `buckets`  | array | []      | Set buckets to be used by histogram metrics |
+
 ## Metrics
 
 ### Query performance 

lib/yabeda/active_record.rb

@@ -2,6 +2,7 @@
 
 require "yabeda"
 require "active_record"
+require "yabeda/active_record/config"
 
 require_relative "active_record/version"
 
@@ -14,14 +15,18 @@ class Error < StandardError; end
       0.005, 0.01, 0.025, 0.05, 0.1, 0.25, 0.5, 1, 2.5, 5, 10, # standard (from Prometheus)
       30, 60, 120, 300, 1800, 3600, 21_600, # Well, sometime queries can execute way too long
     ].freeze
-
+    
+    
     # rubocop: disable Layout/LineLength
     Yabeda.configure do
+      config = ::Yabeda::ActiveRecord::Config.new
+      buckets = config.buckets || LONG_RUNNING_QUERY_RUNTIME_BUCKETS
+
       group :activerecord do
         counter   :queries_total, tags: %i[config kind cached async],
                                   comment: "Total number of SQL queries issued by application via ActiveRecord"
         histogram :query_duration, tags: %i[config kind cached async],
-                                   unit: :seconds, buckets: LONG_RUNNING_QUERY_RUNTIME_BUCKETS,
+                                   unit: :seconds, buckets: buckets,
                                    comment: "Duration of SQL queries generated by ActiveRecord"
 
         gauge :connection_pool_size,

lib/yabeda/active_record/config.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+require 'anyway'
+
+module Yabeda
+  module ActiveRecord
+    # yabeda-activerecord configuration
+    class Config < ::Anyway::Config
+      config_name :yabeda_activerecord
+
+      attr_config :buckets
+    end
+  end
+end

yabeda-activerecord.gemspec

@@ -33,4 +33,5 @@ Gem::Specification.new do |spec|
 
   spec.add_dependency "activerecord", ">= 6.0"
   spec.add_dependency "yabeda", "~> 0.6"
+  spec.add_dependency "anyway_config", ">= 1.3", "< 3.0"
 end