tina4ruby 3.11.13 → 3.11.15

data/lib/tina4/queue_backends/mongo_backend.rb

@@ -1,126 +1,126 @@
-# frozen_string_literal: true
-
-module Tina4
-  module QueueBackends
-    class MongoBackend
-      def initialize(options = {})
-        require "mongo"
-
-        uri = options[:uri] || ENV["TINA4_MONGO_URI"]
-        host = options[:host] || ENV.fetch("TINA4_MONGO_HOST", "localhost")
-        port = (options[:port] || ENV.fetch("TINA4_MONGO_PORT", 27017)).to_i
-        username = options[:username] || ENV["TINA4_MONGO_USERNAME"]
-        password = options[:password] || ENV["TINA4_MONGO_PASSWORD"]
-        db_name = options[:db] || ENV.fetch("TINA4_MONGO_DB", "tina4")
-        @collection_name = options[:collection] || ENV.fetch("TINA4_MONGO_COLLECTION", "tina4_queue")
-
-        if uri
-          @client = Mongo::Client.new(uri)
-        else
-          conn_options = { database: db_name }
-          conn_options[:user] = username if username
-          conn_options[:password] = password if password
-          @client = Mongo::Client.new(["#{host}:#{port}"], conn_options)
-        end
-
-        @db = @client.database
-        create_indexes
-      rescue LoadError
-        raise "MongoDB backend requires the 'mongo' gem. Install with: gem install mongo"
-      end
-
-      def enqueue(message)
-        collection.insert_one(
-          _id: message.id,
-          topic: message.topic,
-          payload: message.payload,
-          created_at: message.created_at.utc,
-          attempts: message.attempts,
-          status: "pending"
-        )
-      end
-
-      def dequeue(topic)
-        doc = collection.find_one_and_update(
-          { topic: topic, status: "pending" },
-          { "$set" => { status: "processing" } },
-          sort: { created_at: 1 },
-          return_document: :after
-        )
-        return nil unless doc
-
-        Tina4::Job.new(
-          topic: doc["topic"],
-          payload: doc["payload"],
-          id: doc["_id"]
-        )
-      end
-
-      def acknowledge(message)
-        collection.delete_one(_id: message.id)
-      end
-
-      def requeue(message)
-        collection.find_one_and_update(
-          { _id: message.id },
-          { "$set" => { status: "pending" }, "$inc" => { attempts: 1 } },
-          upsert: true
-        )
-      end
-
-      def dead_letter(message)
-        collection.find_one_and_update(
-          { _id: message.id },
-          { "$set" => { status: "dead", topic: "#{message.topic}.dead_letter" } },
-          upsert: true
-        )
-      end
-
-      def size(topic)
-        collection.count_documents(topic: topic, status: "pending")
-      end
-
-      def dead_letters(topic, max_retries: 3)
-        collection.find(topic: "#{topic}.dead_letter", status: "dead").map do |doc|
-          Tina4::Job.new(
-            topic: doc["topic"],
-            payload: doc["payload"],
-            id: doc["_id"]
-          )
-        end
-      end
-
-      def purge(topic, status)
-        result = collection.delete_many(topic: topic, status: status.to_s)
-        result.deleted_count
-      end
-
-      def retry_failed(topic, max_retries: 3)
-        result = collection.update_many(
-          { topic: topic, status: "failed", attempts: { "$lt" => max_retries } },
-          { "$set" => { status: "pending" } }
-        )
-        result.modified_count
-      end
-
-      def close
-        @client&.close
-      end
-
-      private
-
-      def collection
-        @db[@collection_name]
-      end
-
-      def create_indexes
-        collection.indexes.create_many([
-          { key: { topic: 1, status: 1, created_at: 1 } },
-          { key: { topic: 1, status: 1, attempts: 1 } }
-        ])
-      rescue Mongo::Error => e
-        Tina4::Log.warning("MongoDB index creation failed: #{e.message}")
-      end
-    end
-  end
-end
+# frozen_string_literal: true
+
+module Tina4
+  module QueueBackends
+    class MongoBackend
+      def initialize(options = {})
+        require "mongo"
+
+        uri = options[:uri] || ENV["TINA4_MONGO_URI"]
+        host = options[:host] || ENV.fetch("TINA4_MONGO_HOST", "localhost")
+        port = (options[:port] || ENV.fetch("TINA4_MONGO_PORT", 27017)).to_i
+        username = options[:username] || ENV["TINA4_MONGO_USERNAME"]
+        password = options[:password] || ENV["TINA4_MONGO_PASSWORD"]
+        db_name = options[:db] || ENV.fetch("TINA4_MONGO_DB", "tina4")
+        @collection_name = options[:collection] || ENV.fetch("TINA4_MONGO_COLLECTION", "tina4_queue")
+
+        if uri
+          @client = Mongo::Client.new(uri)
+        else
+          conn_options = { database: db_name }
+          conn_options[:user] = username if username
+          conn_options[:password] = password if password
+          @client = Mongo::Client.new(["#{host}:#{port}"], conn_options)
+        end
+
+        @db = @client.database
+        create_indexes
+      rescue LoadError
+        raise "MongoDB backend requires the 'mongo' gem. Install with: gem install mongo"
+      end
+
+      def enqueue(message)
+        collection.insert_one(
+          _id: message.id,
+          topic: message.topic,
+          payload: message.payload,
+          created_at: message.created_at.utc,
+          attempts: message.attempts,
+          status: "pending"
+        )
+      end
+
+      def dequeue(topic)
+        doc = collection.find_one_and_update(
+          { topic: topic, status: "pending" },
+          { "$set" => { status: "processing" } },
+          sort: { created_at: 1 },
+          return_document: :after
+        )
+        return nil unless doc
+
+        Tina4::Job.new(
+          topic: doc["topic"],
+          payload: doc["payload"],
+          id: doc["_id"]
+        )
+      end
+
+      def acknowledge(message)
+        collection.delete_one(_id: message.id)
+      end
+
+      def requeue(message)
+        collection.find_one_and_update(
+          { _id: message.id },
+          { "$set" => { status: "pending" }, "$inc" => { attempts: 1 } },
+          upsert: true
+        )
+      end
+
+      def dead_letter(message)
+        collection.find_one_and_update(
+          { _id: message.id },
+          { "$set" => { status: "dead", topic: "#{message.topic}.dead_letter" } },
+          upsert: true
+        )
+      end
+
+      def size(topic)
+        collection.count_documents(topic: topic, status: "pending")
+      end
+
+      def dead_letters(topic, max_retries: 3)
+        collection.find(topic: "#{topic}.dead_letter", status: "dead").map do |doc|
+          Tina4::Job.new(
+            topic: doc["topic"],
+            payload: doc["payload"],
+            id: doc["_id"]
+          )
+        end
+      end
+
+      def purge(topic, status)
+        result = collection.delete_many(topic: topic, status: status.to_s)
+        result.deleted_count
+      end
+
+      def retry_failed(topic, max_retries: 3)
+        result = collection.update_many(
+          { topic: topic, status: "failed", attempts: { "$lt" => max_retries } },
+          { "$set" => { status: "pending" } }
+        )
+        result.modified_count
+      end
+
+      def close
+        @client&.close
+      end
+
+      private
+
+      def collection
+        @db[@collection_name]
+      end
+
+      def create_indexes
+        collection.indexes.create_many([
+          { key: { topic: 1, status: 1, created_at: 1 } },
+          { key: { topic: 1, status: 1, attempts: 1 } }
+        ])
+      rescue Mongo::Error => e
+        Tina4::Log.warning("MongoDB index creation failed: #{e.message}")
+      end
+    end
+  end
+end

data/lib/tina4/queue_backends/rabbitmq_backend.rb

@@ -1,73 +1,73 @@
-# frozen_string_literal: true
-
-module Tina4
-  module QueueBackends
-    class RabbitmqBackend
-      def initialize(options = {})
-        require "bunny"
-        @connection = Bunny.new(
-          host: options[:host] || "localhost",
-          port: options[:port] || 5672,
-          username: options[:username] || "guest",
-          password: options[:password] || "guest",
-          vhost: options[:vhost] || "/"
-        )
-        @connection.start
-        @channel = @connection.create_channel
-        @queues = {}
-        @exchanges = {}
-      rescue LoadError
-        raise "RabbitMQ backend requires the 'bunny' gem. Install with: gem install bunny"
-      end
-
-      def enqueue(message)
-        queue = get_queue(message.topic)
-        queue.publish(message.to_json, persistent: true)
-      end
-
-      def dequeue(topic)
-        queue = get_queue(topic)
-        delivery_info, _properties, payload = queue.pop
-        return nil unless payload
-
-        data = JSON.parse(payload)
-        msg = Tina4::Job.new(
-          topic: data["topic"],
-          payload: data["payload"],
-          id: data["id"]
-        )
-        @last_delivery_tag = delivery_info.delivery_tag
-        msg
-      end
-
-      def acknowledge(_message)
-        @channel.acknowledge(@last_delivery_tag) if @last_delivery_tag
-      end
-
-      def requeue(message)
-        enqueue(message)
-      end
-
-      def dead_letter(message)
-        dlq = get_queue("#{message.topic}.dead_letter")
-        dlq.publish(message.to_json, persistent: true)
-      end
-
-      def size(topic)
-        queue = get_queue(topic)
-        queue.message_count
-      end
-
-      def close
-        @channel&.close
-        @connection&.close
-      end
-
-      private
-
-      def get_queue(topic)
-        @queues[topic] ||= @channel.queue(topic, durable: true)
-      end
-    end
-  end
-end
+# frozen_string_literal: true
+
+module Tina4
+  module QueueBackends
+    class RabbitmqBackend
+      def initialize(options = {})
+        require "bunny"
+        @connection = Bunny.new(
+          host: options[:host] || "localhost",
+          port: options[:port] || 5672,
+          username: options[:username] || "guest",
+          password: options[:password] || "guest",
+          vhost: options[:vhost] || "/"
+        )
+        @connection.start
+        @channel = @connection.create_channel
+        @queues = {}
+        @exchanges = {}
+      rescue LoadError
+        raise "RabbitMQ backend requires the 'bunny' gem. Install with: gem install bunny"
+      end
+
+      def enqueue(message)
+        queue = get_queue(message.topic)
+        queue.publish(message.to_json, persistent: true)
+      end
+
+      def dequeue(topic)
+        queue = get_queue(topic)
+        delivery_info, _properties, payload = queue.pop
+        return nil unless payload
+
+        data = JSON.parse(payload)
+        msg = Tina4::Job.new(
+          topic: data["topic"],
+          payload: data["payload"],
+          id: data["id"]
+        )
+        @last_delivery_tag = delivery_info.delivery_tag
+        msg
+      end
+
+      def acknowledge(_message)
+        @channel.acknowledge(@last_delivery_tag) if @last_delivery_tag
+      end
+
+      def requeue(message)
+        enqueue(message)
+      end
+
+      def dead_letter(message)
+        dlq = get_queue("#{message.topic}.dead_letter")
+        dlq.publish(message.to_json, persistent: true)
+      end
+
+      def size(topic)
+        queue = get_queue(topic)
+        queue.message_count
+      end
+
+      def close
+        @channel&.close
+        @connection&.close
+      end
+
+      private
+
+      def get_queue(topic)
+        @queues[topic] ||= @channel.queue(topic, durable: true)
+      end
+    end
+  end
+end