rmq.rake in rabbit_carrots-0.1.19

- old
+ new

@@ -3,10 +3,14 @@
 namespace :rabbit_carrots do
   desc 'Listener for Queue'
   task eat: :environment do
     Rails.application.eager_load!
 
+    DatabaseAgonsticNotNullViolation = defined?(ActiveRecord) ? ActiveRecord::NotNullViolation : RabbitCarrots::EventHandlers::Errors::PlaceholderError
+    DatabaseAgonsticConnectionNotEstablished = defined?(ActiveRecord) ? ActiveRecord::ConnectionNotEstablished : Mongo::Error::SocketError
+    DatabaseAgnosticRecordInvalid = defined?(ActiveRecord) ? ActiveRecord::RecordInvalid : Mongoid::Errors::Validations
+    
     channels = RabbitCarrots.configuration.routing_key_mappings.map do |mapping|
       # This will be supplied in initializer. At that time, the Handler will not be available to be loaded and will throw Uninitialized Constant
       { **mapping, handler: mapping[:handler].constantize }
     end
 
@@ -45,14 +49,14 @@
       Rails.logger.info "Nacked message: #{payload}"
       channel.nack(delivery_info.delivery_tag, false, false)
     rescue RabbitCarrots::EventHandlers::Errors::NackAndRequeueMessage => _e
       Rails.logger.info "Nacked and Requeued message: #{payload}"
       channel.nack(delivery_info.delivery_tag, false, true)
-    rescue ActiveRecord::NotNullViolation, ActiveRecord::RecordInvalid => e
+    rescue DatabaseAgonsticNotNullViolation, DatabaseAgnosticRecordInvalid => e
       # on null constraint violation, we want to ack the message
       Rails.logger.error "Null constraint or Invalid violation: #{payload}. Error: #{e.message}"
       channel.ack(delivery_info.delivery_tag, false)
-    rescue ActiveRecord::ConnectionNotEstablished => e
+    rescue DatabaseAgonsticConnectionNotEstablished => e
       # on connection not established, we want to requeue the message and sleep for 3 seconds
       Rails.logger.error "Error connection not established to the database: #{payload}. Error: #{e.message}"
       # delay for 3 seconds before requeuing
       sleep 3
       channel.nack(delivery_info.delivery_tag, false, true)