lib/cyclone_lariat/middleware.rb in cyclone_lariat-0.2.3 vs lib/cyclone_lariat/middleware.rb in cyclone_lariat-0.3.0

@@ -1,24 +1,24 @@
 # frozen_string_literal: true
 
-require_relative 'events_repo'
+require_relative 'messages_repo'
 require 'luna_park/errors'
 require 'json'
 
 module CycloneLariat
   class Middleware
-    def initialize(dataset: nil, errors_notifier: nil, message_notifier: nil, repo: EventsRepo)
+    def initialize(dataset: nil, errors_notifier: nil, message_notifier: nil, repo: MessagesRepo)
       @events_repo      = repo.new(dataset) if dataset
       @message_notifier = message_notifier
       @errors_notifier  = errors_notifier
     end
 
     def call(_worker_instance, queue, _sqs_msg, body, &block)
       log_received_message queue, body
 
       catch_standard_error(queue, body) do
         return true unless check(body[:Message])
-        
+
         event = Event.wrap(JSON.parse(body[:Message]))
 
         catch_business_error(event) do
           store_in_dataset(event, &block)
         end