# frozen_string_literal: true require 'cyclone_lariat/repo/inbox_messages' require 'cyclone_lariat/core' require 'luna_park/errors' require 'cyclone_lariat/messages/builder' require 'json' module CycloneLariat class Middleware attr_reader :config def initialize(errors_notifier: nil, message_notifier: nil, before_save: nil, repo: Repo::InboxMessages, **options) @config = CycloneLariat::Options.wrap(options).merge!(CycloneLariat.config) @events_repo = repo.new(**@config.to_h) @message_notifier = message_notifier @errors_notifier = errors_notifier @before_save = before_save end def call(_worker_instance, queue, _sqs_msg, body, &block) msg = receive_message!(body) message_notifier&.info 'Receive message', message: msg, queue: queue return if msg.is_a? String catch_standard_error(queue, msg) do event = Messages::Builder.new(raw_message: msg).call store_in_dataset(event) do catch_business_error(event, &block) end end end private attr_reader :errors_notifier, :message_notifier, :events_repo, :before_save def receive_message!(body) body[:Message] ? JSON.parse(body[:Message], symbolize_names: true) : body rescue JSON::ParserError => e errors_notifier&.error(e, message: body[:Message]) body[:Message] end def store_in_dataset(event) return yield if events_repo.disabled? existed = events_repo.find(uuid: event.uuid) return true if existed&.processed? return yield if existed event.clone.tap do |e| before_save.call(e) if before_save events_repo.create(e) end yield events_repo.processed!(uuid: event.uuid, error: event.client_error) end def catch_business_error(event) yield rescue LunaPark::Errors::Business => e errors_notifier&.error(e, event: event) event.client_error = e end def catch_standard_error(queue, msg) yield rescue Exception => e errors_notifier&.error(e, queue: queue, message: msg) raise e end end end