active_statsd 0.1.0 → 0.1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7ba98bd3462d5bb72af512e20ed1a28889a1b8ace4d47478a429086e76007222
-  data.tar.gz: 524bb2108f7b84aa9e13d06c22edbd088a65b8ce2cb0716b3355e191f7cf1ed8
+  metadata.gz: 65a1d95ad8e6bb84ddaa3ebfdbdebf0f2e22461690f7600c8ddffdfd4e285d49
+  data.tar.gz: 6678ff948cb67c06d9d577186de57ebfc06b2d6cbe46ef42a7d35daea5e2e803
 SHA512:
-  metadata.gz: a1269bae74e65c2b55a92720cc962abdb2e8794a145826332b87a0540caefb141faf0bb4c9a4d28a5a6ebca501b0356afab8f7b10125c6d725fe27407dad32e0
-  data.tar.gz: 318929577949d3447672166ed61cbb8c1d81b689a814fa25b05c49b916f852ecb1f2f8707746bfca00a0074b25322c011c3fe8015447c73d4e792d2505e50d58
+  metadata.gz: 32d76aaf7d39d8f9f89c7097da9a6ed184388044ab9f087e0ab5882b4df125f879a18d7dda19f53b12840f50d7daf18285f0753ece6c679d93a18a1865c85cd2
+  data.tar.gz: f4fae172b244f3913fb1a380993848396d91b3b7e625a78d25b4b944d573dd2faf20b320bc8a74025f11eaad620bb2904113a1916f9d6a223b1c0358d6a80466

data/lib/active_statsd/configuration.rb

@@ -1,11 +1,15 @@
+# lib/active_statsd/configuration.rb
 module ActiveStatsD
   class Configuration
-    attr_accessor :host, :port, :namespace
+    attr_accessor :host, :port, :namespace, :aggregation, :forward_host, :forward_port
 
     def initialize
       @host = '127.0.0.1'
       @port = 8125
       @namespace = 'rails_app'
+      @aggregation = true  # Aggregation enabled by default
+      @forward_host = nil  # Set if forwarding
+      @forward_port = nil
     end
   end
 end

data/lib/active_statsd/railtie.rb

@@ -7,12 +7,12 @@ module ActiveStatsD
       ActiveSupport.on_load(:after_initialize) do
         server = ActiveStatsD::Server.new(
           host: ActiveStatsD.configuration.host,
-          port: ActiveStatsD.configuration.port
+          port: ActiveStatsD.configuration.port,
+          aggregation: ActiveStatsD.configuration.aggregation,
+          forward_host: ActiveStatsD.configuration.forward_host,
+          forward_port: ActiveStatsD.configuration.forward_port
         )
-
         server.start
-
-        Rails.logger.info "[ActiveStatsD] Embedded StatsD server started on #{ActiveStatsD.configuration.host}:#{ActiveStatsD.configuration.port}"
       end
     end
 

data/lib/active_statsd/server.rb

@@ -3,18 +3,22 @@ require 'socket'
 
 module ActiveStatsD
   class Server
-    def initialize(host: '127.0.0.1', port: 8125)
+    def initialize(host:, port:, aggregation:, forward_host:, forward_port:)
       @host = host
       @port = port
+      @aggregation = aggregation
+      @forward_host = forward_host
+      @forward_port = forward_port
       @counters = Hash.new(0)
       @mutex = Mutex.new
+      @forward_socket = UDPSocket.new if forwarding_enabled?
     end
 
     def start
-      start_aggregation_thread
+      start_aggregation_thread if aggregation_enabled?
 
       Thread.new do
-        Rails.logger.info "[ActiveStatsD] UDP StatsD listener started on #{@host}:#{@port}"
+        Rails.logger.info "[ActiveStatsD] UDP StatsD listener started on #{@host}:#{@port} (aggregation=#{@aggregation})"
         begin
           Socket.udp_server_loop(@host, @port) do |msg, _|
             Rails.logger.debug "[ActiveStatsD] UDP packet received: #{msg.strip}"
@@ -33,16 +37,20 @@ module ActiveStatsD
 
       return unless metric && type == 'c'
 
-      @mutex.synchronize do
-        @counters[metric] += value
+      if aggregation_enabled?
+        @mutex.synchronize { @counters[metric] += value }
+      end
+
+      if forwarding_enabled?
+        forward_message(message)
       end
+
+      log_message(metric, value, type) unless aggregation_enabled?
     end
 
     def parse_metric(message)
-      # Example message: "rails_app.test.metric:1|c"
       metric_data, type = message.split('|')
       metric, value = metric_data.split(':')
-
       [metric, value.to_i, type]
     rescue
       Rails.logger.error "[ActiveStatsD] Failed to parse metric: #{message}"
@@ -52,7 +60,7 @@ module ActiveStatsD
     def start_aggregation_thread
       Thread.new do
         loop do
-          sleep 10 # flush metrics every 10 seconds
+          sleep 10
           flush_metrics
         end
       end
@@ -69,5 +77,23 @@ module ActiveStatsD
         Rails.logger.info "[ActiveStatsD] Aggregated metric - #{metric}: #{count}"
       end
     end
+
+    def forwarding_enabled?
+      @forward_host && @forward_port
+    end
+
+    def aggregation_enabled?
+      @aggregation
+    end
+
+    def forward_message(message)
+      @forward_socket.send(message, 0, @forward_host, @forward_port)
+    rescue => e
+      Rails.logger.error "[ActiveStatsD] Forwarding error: #{e.message}"
+    end
+
+    def log_message(metric, value, type)
+      Rails.logger.info "[ActiveStatsD] Metric received (no aggregation) - #{metric}:#{value}|#{type}"
+    end
   end
 end

data/lib/active_statsd/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ActiveStatsd
-  VERSION = "0.1.0"
+  VERSION = "0.1.0.1"
 end

data/lib/generators/active_statsd/active_statsd_generator.rb

@@ -0,0 +1,12 @@
+# lib/generators/active_statsd/active_statsd_generator.rb
+require 'rails/generators'
+
+module ActiveStatsd
+  class ActiveStatsdGenerator < Rails::Generators::Base
+    source_root File.expand_path('templates', __dir__)
+
+    def copy_initializer_file
+      copy_file 'active_statsd_initializer.rb', 'config/initializers/active_statsd.rb'
+    end
+  end
+end

data/lib/generators/active_statsd/templates/active_statsd_initializer.rb

@@ -0,0 +1,5 @@
+ActiveStatsD.configure do |config|
+  config.host = '127.0.0.1'
+  config.port = 8125
+  config.namespace = 'my_rails_app'
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: active_statsd
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.0.1
 platform: ruby
 authors:
 - Mike Poage
@@ -74,6 +74,8 @@ files:
 - lib/active_statsd/railtie.rb
 - lib/active_statsd/server.rb
 - lib/active_statsd/version.rb
+- lib/generators/active_statsd/active_statsd_generator.rb
+- lib/generators/active_statsd/templates/active_statsd_initializer.rb
 - sig/active_statsd.rbs
 homepage: https://github.com/RubyBrewsday/active_statsd
 licenses: