lib/ahoy.rb in ahoy_matey-1.5.4 vs lib/ahoy.rb in ahoy_matey-1.5.5

@@ -25,10 +25,12 @@
 require "ahoy/stores/active_record_token_store"
 require "ahoy/stores/log_store"
 require "ahoy/stores/fluentd_store"
 require "ahoy/stores/mongoid_store"
 require "ahoy/stores/kafka_store"
+require "ahoy/stores/nats_store"
+require "ahoy/stores/nsq_store"
 require "ahoy/stores/kinesis_firehose_store"
 require "ahoy/stores/bunny_store"
 require "ahoy/engine" if defined?(Rails)
 require "ahoy/warden" if defined?(Warden)
 
@@ -114,9 +116,10 @@
 end
 
 if defined?(Rails)
   ActiveSupport.on_load(:action_controller) do
     ActionController::Base.send :include, Ahoy::Controller
+    ActionController::API.send :include, Ahoy::Controller if defined?(ActionController::API)
   end
 
   ActiveSupport.on_load(:active_record) do
     ActiveRecord::Base.send(:extend, Ahoy::Model)
   end