Allow configure more than one publisher (#15)

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ## main
 
+- [PR#15](https://github.com/DmitryTsepelev/io_monitor/pull/15) Allow configure more than one publisher ([@DmitryTsepelev])
 - [PR#9](https://github.com/DmitryTsepelev/io_monitor/pull/9) Restrict minimum Rails version to 6.1, adjust test matrix, and related changes ([@Envek])
 
 ## 0.2.0 (2022-05-29)

README.md

@@ -34,7 +34,7 @@ Change configuration in an initializer if you need:
 
 ```ruby
 IoMonitor.configure do |config|
-  config.publish = :notifications # defaults to :logs
+  config.publish = [:logs, :notifications] # defaults to :logs
   config.warn_threshold = 0.8 # defaults to 0
   config.adapters = [:active_record, :net_http, :redis] # defaults to [:active_record]
 end

lib/io_monitor/configuration.rb

@@ -5,22 +5,15 @@ class Configuration
     DEFAULT_WARN_THRESHOLD = 0.0
 
     def initialize
-      @publisher = LogsPublisher.new
+      @publishers = [LogsPublisher.new]
       @adapters = [ActiveRecordAdapter.new]
       @warn_threshold = DEFAULT_WARN_THRESHOLD
     end
 
-    attr_reader :publisher, :adapters, :warn_threshold
+    attr_reader :publishers, :adapters, :warn_threshold
 
-    def publish=(value)
-      if value.is_a?(BasePublisher)
-        @publisher = value
-      elsif (publisher_type = resolve(IoMonitor::PUBLISHERS, value))
-        @publisher = publisher_type.new
-      else
-        supported = IoMonitor::PUBLISHERS.map(&:kind)
-        raise ArgumentError, "Only the following publishers are supported: #{supported}."
-      end
+    def publish=(values)
+      @publishers = [*values].map { |value| value_to_publisher(value) }
     end
 
     def adapters=(value)
@@ -50,5 +43,16 @@ def warn_threshold=(value)
     def resolve(list, kind)
       list.find { |p| p.kind == kind }
     end
+
+    def value_to_publisher(value)
+      if value.is_a?(BasePublisher)
+        value
+      elsif (publisher_type = resolve(IoMonitor::PUBLISHERS, value))
+        publisher_type.new
+      else
+        supported = IoMonitor::PUBLISHERS.map(&:kind)
+        raise ArgumentError, "Only the following publishers are supported: #{supported}."
+      end
+    end
   end
 end

lib/io_monitor/patches/action_controller_base_patch.rb

@@ -4,7 +4,7 @@ module IoMonitor
   module ActionControllerBasePatch
     def log_process_action(payload)
       super.tap do |messages|
-        next unless IoMonitor.config.publisher.is_a?(LogsPublisher)
+        next unless IoMonitor.config.publishers.any? { |publisher| publisher.is_a?(LogsPublisher) }
 
         data = payload[IoMonitor::NAMESPACE]
         next unless data

lib/io_monitor/railtie.rb

@@ -15,7 +15,7 @@ class Railtie < Rails::Railtie
         payload = args.last[IoMonitor::NAMESPACE]
         next unless payload
 
-        IoMonitor.config.publisher.process_action(payload)
+        IoMonitor.config.publishers.each { |publisher| publisher.process_action(payload) }
       end
     end
   end

spec/io_monitor/configuration_spec.rb

@@ -6,7 +6,7 @@
   describe ".publish" do
     it "resolves publisher by kind" do
       config.publish = :notifications
-      expect(config.publisher).to be_a(IoMonitor::NotificationsPublisher)
+      expect(config.publishers.first).to be_a(IoMonitor::NotificationsPublisher)
     end
 
     context "when kind is unknown" do
@@ -18,11 +18,19 @@
     it "allows custom publishers" do
       custom = Class.new(IoMonitor::BasePublisher) {}
       config.publish = custom.new
-      expect(config.publisher).to be_a(custom)
+      expect(config.publishers.first).to be_a(custom)
     end
 
     it "equals to :logs by default" do
-      expect(config.publisher).to be_a(IoMonitor::LogsPublisher)
+      expect(config.publishers.first).to be_a(IoMonitor::LogsPublisher)
+    end
+
+    context "when multiple publishers are set" do
+      it "assigns all publishers" do
+        config.publish = [:notifications, :logs]
+        expect(config.publishers.first).to be_a(IoMonitor::NotificationsPublisher)
+        expect(config.publishers.last).to be_a(IoMonitor::LogsPublisher)
+      end
     end
   end