lib/qstash/message.rb in qstash-rb-0.1.1 vs lib/qstash/message.rb in qstash-rb-0.1.2

@@ -1,20 +1,40 @@
 # frozen_string_literal: true
 
 require "qstash/message/publish"
 require "qstash/message/enqueue"
+require "qstash/message/batch"
+require "qstash/message/get"
+require "qstash/message/cancel"
+require "qstash/message/bulk_cancel"
 
 module Qstash
   module Message
     class << self
-      def publish(destination:, body:, headers: {})
-        Publish.call(destination: destination, body: body, headers: headers)
+      def batch(messages:, headers: {})
+        Batch.call(messages: messages, headers: headers)
       end
 
+      def bulk_cancel(message_ids, headers: {})
+        BulkCancel.call(message_ids, headers: headers)
+      end
+
+      def cancel(message_id, headers: {})
+        Cancel.call(message_id, headers: headers)
+      end
+
       def enqueue(queue_name:, destination:, body:, headers: {})
         Enqueue.call(queue_name: queue_name, destination: destination, body: body, headers: headers)
       end
+
+      def get(message_id, headers: {})
+        Get.call(message_id, headers: headers)
+      end
+
+      def publish(destination:, body:, headers: {})
+        Publish.call(destination: destination, body: body, headers: headers)
+      end
     end
 
-    private_constant :Publish, :Enqueue
+    private_constant :Publish, :Enqueue, :Batch, :Get, :Cancel, :BulkCancel
   end
 end