cosmonats 0.7.0 → 0.9.0

data/sig/cosmo/api/counter.rbs

@@ -1,13 +1,12 @@
 module Cosmo
   module API
     class Counter
-      STREAM_NAME: ::String
-      STREAM_CONFIG: Hash[Symbol, untyped]
-
       @namespace: ::String
 
       def self.setup!: () -> void
 
+      def self.stream_config: () -> Hash[Symbol, untyped]
+
       def initialize: (::String namespace) -> void
 
       def increment: (Symbol key, ?by: ::Integer) -> ::Integer

data/sig/cosmo/api/cron/entry.rbs

@@ -19,7 +19,8 @@ module Cosmo
         def schedule_subject: () -> ::String
         def target_subject: () -> ::String
         def dispatch_subject: () -> ::String
-        def schedule_headers: () -> Hash[::String, ::String]
+        def schedule: () -> Hash[Symbol, ::String]
+        def headers: () -> Hash[::String, ::String]
         def job_name: () -> ::String
         def job_payload: () -> ::String
         def as_json: () -> Hash[Symbol, untyped]

data/sig/cosmo/api/stats/busy.rbs

@@ -12,7 +12,7 @@ module Cosmo
 
         def initialize: () -> void
 
-        def with: (untyped message) { () -> void } -> void
+        def with: [T] (untyped message) { () -> T } -> T
 
         def add: (untyped message) -> void
 
@@ -20,6 +20,8 @@ module Cosmo
 
         def list: (?page: ::Integer?, ?limit: ::Integer) -> Array[Hash[Symbol, untyped]]
 
+        def by_stream: () -> Hash[::String, ::Integer]
+
         private
 
         def heartbeat_loop: () -> void

data/sig/cosmo/api/stats/metrics.rbs

@@ -0,0 +1,42 @@
+module Cosmo
+  module API
+    module Stats
+      class Metrics < Counter
+        STREAM_NAME: ::String
+        DESCRIPTION: ::String
+        DAY: Integer
+
+        self.@instance: Metrics
+
+        @lock: Thread::Mutex
+        @buffer: Hash[[::String, ::String], Hash[Symbol, Numeric]]
+
+        def self.instance: () -> Metrics
+
+        def self.stream_config: () -> Hash[Symbol, untyped]
+
+        def initialize: () -> void
+
+        def record: (::String job_class, ?exec_ms: Numeric?, ?wait_ms: Numeric?, ?failed: bool) -> void
+
+        def flush: () -> void
+
+        def summary: (?days: Integer) -> Array[Hash[Symbol, untyped]]
+
+        def daily: (?::String? job_class, ?days: Integer) -> Array[Hash[Symbol, untyped]]
+
+        private
+
+        def buffer: () -> Hash[[::String, ::String], Hash[Symbol, Numeric]]
+
+        def read: (Integer days) -> Array[[[::String, ::String, Symbol], Integer]]
+
+        def dates: (Integer days) -> Array[::String]
+
+        def row: (Hash[Symbol, Numeric] sums) -> Hash[Symbol, untyped]
+
+        def average: (Numeric total, Numeric count) -> Float?
+      end
+    end
+  end
+end

data/sig/cosmo/api/stats/processes.rbs

@@ -4,6 +4,9 @@ module Cosmo
       class Processes < Registry
         TTL: ::Integer
         BUCKET: ::String
+        KINDS: Array[::String]
+
+        def self.kind: (Hash[Symbol, untyped] process) -> ::String
 
         def list: (?page: ::Integer?, ?limit: ::Integer) -> Array[Hash[Symbol, untyped]]
 

data/sig/cosmo/api/stats/totals.rbs

@@ -2,12 +2,13 @@ module Cosmo
   module API
     module Stats
       class Totals < Counter
+        STREAM_NAME: ::String
+        DESCRIPTION: ::String
+
         self.@instance: Totals
 
         def self.instance: () -> Totals
 
-        def with: () { () -> bool } -> void
-
         def processed: () -> ::Integer
 
         def failed: () -> ::Integer

data/sig/cosmo/api/stats.rbs

@@ -16,6 +16,10 @@ module Cosmo
       def self.scheduled: () -> ::Integer
 
       def self.dead: () -> ::Integer
+
+      def self.queues: () -> Array[Hash[Symbol, untyped]]
+
+      def self.queue: (Stream stream, ?Hash[::String, ::Integer] busy) -> Hash[Symbol, untyped]
     end
   end
 end

data/sig/cosmo/api/stream.rbs

@@ -26,6 +26,8 @@ module Cosmo
 
       def retries: () -> ::Integer
 
+      def consumers: () -> Array[Hash[::String, untyped]]
+
       def each: () { (Job) -> void } -> void
 
       def messages: (?page: ::Integer?, ?limit: ::Integer?) -> Array[Job]

data/sig/cosmo/batch/counters.rbs

@@ -0,0 +1,10 @@
+module Cosmo
+  class Batch
+    class Counters < API::Counter
+      STREAM_NAME: ::String
+      DESCRIPTION: ::String
+
+      def self.stream_config: () -> Hash[Symbol, untyped]
+    end
+  end
+end

data/sig/cosmo/batch.rbs

@@ -3,17 +3,16 @@ module Cosmo
     extend Dispatcher
 
     BUCKET: ::String
-    DEFAULT_EXPIRY: ::String
     EVENTS: ::Array[Symbol]
 
-    self.@counter: API::Counter
+    self.@counter: Batch::Counters
     self.@kv: API::KV
 
     def self.current: () -> Batch?
 
     def self.notify: (::String bid, ::String jid, success: bool) -> void
 
-    def self.counter: () -> API::Counter
+    def self.counter: () -> Batch::Counters
 
     def self.kv: () -> API::KV
 
@@ -36,7 +35,7 @@ module Cosmo
 
     private
 
-    def counter: () -> API::Counter
+    def counter: () -> Batch::Counters
 
     def kv: () -> API::KV
 

data/sig/cosmo/cli.rbs

@@ -20,6 +20,8 @@ module Cosmo
 
     def load_config: (Hash[Symbol, untyped] flags) -> void
 
+    def run_init: () -> void
+
     def boot_application: () -> void
 
     def require_path: (::String? path) -> void

data/sig/cosmo/client.rbs

@@ -2,6 +2,7 @@ module Cosmo
   class Client
     JS_TIMEOUT: ::Integer
     CONNECT_TIMEOUT: ::Integer
+    MAX_RECONNECT_ATTEMPTS: ::Integer
 
     attr_reader name: ::String
 
@@ -54,6 +55,8 @@ module Cosmo
 
     def get_message: (::String | Symbol name, **untyped options) -> NATS::JetStream::API::RawStreamMsg
 
+    def last_messages: (::String stream_name, Array[::String] filters) -> Hash[::String, ::String]
+
     def delete_message: (::String name, ::Integer seq) -> Hash[::String, untyped]
 
     def purge: (::String stream_name, ::String? subject) -> ::Integer?
@@ -66,6 +69,6 @@ module Cosmo
 
     def ephemeral_subscribe: (::String | Array[::String] subject, Hash[Symbol, untyped] config) -> untyped
 
-    def create_kv_with_msg_ttl: (::String name, **untyped opts) -> untyped
+    def create_kv_with_msg_ttl: (::String name, ?replicas: Integer, **untyped opts) -> untyped
   end
 end

data/sig/cosmo/config/settings.rbs

@@ -0,0 +1,31 @@
+module Cosmo
+  class Config < ::Hash[Symbol, untyped]
+    class Scheduled
+      attr_accessor enabled: bool
+
+      def self.new: (enabled: bool) -> Scheduled
+    end
+
+    class Dead
+      attr_accessor enabled: bool
+      attr_accessor max_age: Numeric | ::String
+      attr_accessor max_msgs: Integer
+      attr_accessor max_bytes: Integer
+
+      def self.new: (enabled: bool, max_age: Numeric | ::String, max_msgs: Integer, max_bytes: Integer) -> Dead
+    end
+
+    class Batches
+      attr_accessor expiry: Numeric | ::String
+
+      def self.new: (expiry: Numeric | ::String) -> Batches
+    end
+
+    class Metrics
+      attr_accessor enabled: bool
+      attr_accessor retention: Numeric | ::String
+
+      def self.new: (enabled: bool, retention: Numeric | ::String) -> Metrics
+    end
+  end
+end

data/sig/cosmo/config.rbs

@@ -2,6 +2,8 @@ module Cosmo
   class Config < ::Hash[Symbol, untyped]
     NANO: Integer
     DEFAULT_PATH: ::String
+    DEFAULTS_FILE: ::String
+    SERVICE_STREAMS: Array[Symbol]
 
     self.@instance: Config
     self.@internal: Hash[Symbol, untyped]
@@ -12,8 +14,14 @@ module Cosmo
 
     def self.to_ns: (Numeric seconds) -> ::Integer
 
+    def self.read: (::String path) -> Hash[Symbol, untyped]
+
     def self.parse_file: (::String path) -> Hash[Symbol, untyped]
 
+    def self.build: (Hash[Symbol, untyped] user) -> Hash[Symbol, untyped]
+
+    def self.validate!: (Hash[Symbol, untyped] config) -> void
+
     def self.normalize!: (Hash[Symbol, untyped] config) -> void
 
     def self.deliver_policy: (Symbol | ::String | Time start_position) -> Hash[Symbol, ::String]
@@ -24,15 +32,61 @@ module Cosmo
 
     def self.dig: (*Symbol keys) -> untyped
 
-    def self.to_h: () -> Hash[Symbol, untyped]
+    def self.load: (?::String? path, ?overrides: Hash[Symbol, untyped]?) -> void
+
+    def self.replicas: () -> Integer
+
+    def self.scheduled: () -> Scheduled
+
+    def self.dead: () -> Dead
+
+    def self.batches: () -> Batches
+
+    def self.metrics: () -> Metrics
+
+    def self.server_middleware: () ?{ (Middleware::Chain chain) -> void } -> Middleware::Chain
+
+    def self.client_middleware: () ?{ (Middleware::Chain chain) -> void } -> Middleware::Chain
+
+    def self.error_handlers: () -> Array[untyped]
+
+    @server_middleware: Middleware::Chain
+
+    def load: (?::String? path, ?overrides: Hash[Symbol, untyped]?) -> void
+
+    def server_middleware: () ?{ (Middleware::Chain chain) -> void } -> Middleware::Chain
+
+    @client_middleware: Middleware::Chain
+
+    def client_middleware: () ?{ (Middleware::Chain chain) -> void } -> Middleware::Chain
+
+    @error_handlers: Array[untyped]
+
+    def error_handlers: () -> Array[untyped]
+
+    @replicas: Integer?
+    @scheduled: Scheduled
+    @dead: Dead
+    @batches: Batches
+
+    attr_writer replicas: Integer
+
+    def replicas: () -> Integer
+
+    def scheduled: () -> Scheduled
+
+    def dead: () -> Dead
+
+    def batches: () -> Batches
 
-    def self.set: (*untyped) -> untyped
+    @metrics: Metrics
 
-    def self.load: (?::String? path) -> void
+    def metrics: () -> Metrics
 
+    def logger: () -> untyped
 
-    def set: (*untyped) -> untyped
+    def logger=: (untyped logger) -> void
 
-    def load: (?::String? path) -> void
+    def log_level=: (Symbol | ::String | Integer level) -> void
   end
 end

data/sig/cosmo/heartbeat.rbs

@@ -17,6 +17,8 @@ module Cosmo
 
     def beat: () -> void
 
+    def flush: () -> void
+
     def identity: () -> ::String
 
     def info: () -> Hash[Symbol, untyped]

data/sig/cosmo/job/data.rbs

@@ -17,16 +17,18 @@ module Cosmo
 
       def stream: (?target: bool) -> Symbol
 
-      def subject: (?target: bool) -> Array[::String]
+      def subject: () -> ::String
+
+      def headers: () -> Hash[::String, untyped]
 
       def as_json: () -> { jid: ::String, class: ::String, args: Array[untyped], retry: Integer, dead: bool }
 
       def to_json: (*untyped _args) -> ::String?
 
-      def to_args: () -> [::String, ::String, Hash[Symbol, untyped]]
-
       private
 
+      def subject_for: (Symbol | ::String stream) -> ::String
+
       def validate!: () -> void
 
       def retries: () -> Integer

data/sig/cosmo/job/enqueuer.rbs

@@ -0,0 +1,11 @@
+module Cosmo
+  module Job
+    module Enqueuer
+      def self.enqueue: (::String class_name, Array[untyped] args, Hash[Symbol, untyped] options, ?batch: Batch?) -> ::String?
+
+      def self.pipe: (::String class_name, Data data) -> bool
+
+      def self.publish: (Data data, Hash[Symbol, untyped] payload) -> untyped
+    end
+  end
+end

data/sig/cosmo/job/processor.rbs

@@ -53,13 +53,11 @@ module Cosmo
 
       def reject_message: (untyped message, Exception error, ?Hash[Symbol, untyped]? data) -> void
 
-      def move_message: (untyped message, ?Hash[Symbol, untyped]? data, ?Exception? exception) -> void
-
-      def with_stats: (untyped message) { () -> untyped } -> void
+      def error_context: (untyped message, Hash[Symbol, untyped]? data, ?source: Symbol) -> Hash[Symbol, untyped]
 
-      def acquire_concurrency_slot: (untyped worker_class, untyped message, Hash[Symbol, untyped] data) -> (::String | false)
+      def move_message: (untyped message, ?Hash[Symbol, untyped]? data, ?Exception? exception) -> void
 
-      def perform_job: (untyped job_instance, data: Hash[Symbol, untyped], message: untyped, ?duration: Float?) -> untyped
+      def perform_job: (untyped job_instance, data: Hash[Symbol, untyped], message: untyped) -> untyped
     end
   end
 end

data/sig/cosmo/job/requeue.rbs

@@ -0,0 +1,9 @@
+module Cosmo
+  module Job
+    class Requeue < StandardError
+      attr_reader delay: Numeric
+
+      def initialize: (?Numeric delay) -> void
+    end
+  end
+end

data/sig/cosmo/job.rbs

@@ -31,8 +31,6 @@ module Cosmo
       private
 
       def client: () -> Client
-
-      def publish: (Data data, Batch? batch) -> void
     end
 
     attr_accessor jid: ::String

data/sig/cosmo/logger.rbs

@@ -45,5 +45,11 @@ module Cosmo
     def self.without: (*Symbol) -> nil
 
     def self.instance: () -> untyped
+
+    def self.instance=: (untyped logger) -> void
+
+    def self.level=: (Symbol | ::String | Integer level) -> void
+
+    def self.coerce_level: (Symbol | ::String | Integer level) -> Integer
   end
 end

data/sig/cosmo/middleware/chain.rbs

@@ -0,0 +1,51 @@
+module Cosmo
+  module Middleware
+    class Chain
+      include Enumerable[Entry]
+
+      class Entry
+        attr_accessor klass: Class
+        attr_accessor args: Array[untyped]
+        attr_accessor kwargs: Hash[Symbol, untyped]
+
+        def self.new: (Class klass, Array[untyped] args, Hash[Symbol, untyped] kwargs) -> Entry
+
+        def build: () -> untyped
+      end
+
+      @entries: Array[Entry]
+
+      def initialize: () ?{ (Chain chain) -> void } -> void
+
+      def each: () { (Entry entry) -> void } -> Array[Entry]
+
+      def add: (Class klass, *untyped args, **untyped kwargs) -> Chain
+
+      def prepend: (Class klass, *untyped args, **untyped kwargs) -> Chain
+
+      def insert_before: (Class existing, Class klass, *untyped args, **untyped kwargs) -> Chain
+
+      def insert_after: (Class existing, Class klass, *untyped args, **untyped kwargs) -> Chain
+
+      def remove: (Class klass) -> Chain
+
+      def exists?: (Class klass) -> bool
+
+      alias include? exists?
+
+      def empty?: () -> bool
+
+      def clear: () -> Chain
+
+      def invoke: (*untyped args) { () -> untyped } -> untyped
+
+      private
+
+      def traverse: (Array[untyped] chain, Integer position, Array[untyped] args) { () -> untyped } -> untyped
+
+      def index: (Class klass) -> Integer?
+
+      def take: (Class klass, Array[untyped] args, Hash[Symbol, untyped] kwargs) -> Entry
+    end
+  end
+end

data/sig/cosmo/middleware/sentry.rbs

@@ -0,0 +1,14 @@
+module Cosmo
+  module Middleware
+    class Sentry
+      NAME_PREFIX: ::String
+      OP_NAME: ::String
+      SPAN_ORIGIN: ::String
+      STATUS_OK: ::Integer
+      STATUS_FAIL: ::Integer
+      ERROR_HANDLER: ^(Exception error, Hash[Symbol, untyped] context) -> void
+
+      def call: (untyped job, Hash[Symbol, untyped] data, untyped message) { () -> untyped } -> untyped
+    end
+  end
+end

data/sig/cosmo/middleware.rbs

@@ -0,0 +1,35 @@
+module Cosmo
+  module Middleware
+    class Limit
+      def call: (untyped job, Hash[Symbol, untyped] data, untyped message) { () -> untyped } -> untyped
+
+      private
+
+      def acquire: (untyped job_class, Hash[Symbol, untyped] data) -> ::String?
+
+      def timeout: (Numeric? duration) { () -> untyped } -> untyped
+    end
+
+    class Busy
+      def call: (untyped job, Hash[Symbol, untyped] data, untyped message) { () -> untyped } -> untyped
+    end
+
+    class Metrics
+      def call: (untyped job, Hash[Symbol, untyped] data, untyped message) { () -> untyped } -> untyped
+
+      private
+
+      def job_name: (untyped job, Hash[Symbol, untyped] data) -> ::String
+
+      def clock: () -> Float
+    end
+
+    class Totals
+      def call: (untyped job, Hash[Symbol, untyped] data, untyped message) { () -> untyped } -> untyped
+
+      private
+
+      def increment: (Symbol key) -> void
+    end
+  end
+end

data/sig/cosmo/processor.rbs

@@ -60,5 +60,7 @@ module Cosmo
     def stopwatch: () -> Utils::Stopwatch
 
     def consumer_state: () -> untyped
+
+    def backoff_key: (Hash[Symbol, untyped] config) -> String
   end
 end

data/sig/cosmo/publisher.rbs

@@ -4,18 +4,10 @@ module Cosmo
 
     def self.instance: () -> Publisher
 
-    def self.publish: (::String subject, untyped data, ?serializer: Module?, **untyped options) -> bool
-
-    def self.publish_job: (Job::Data data) -> ::String
-
-    def self.publish_batch: (::String subject, Array[untyped] batch, **untyped options) -> void
+    def self.publish: (::String subject, untyped data, ?serializer: Module?, **untyped options) -> untyped
 
     def initialize: () -> void
 
     def publish: (::String subject, untyped data, ?serializer: Module?, **untyped options) -> untyped
-
-    def publish_job: (Job::Data data) -> ::String
-
-    def publish_batch: (::String subject, Array[untyped] batch, **untyped options) -> void
   end
 end

data/sig/cosmo/services.rbs

@@ -0,0 +1,18 @@
+module Cosmo
+  module Services
+    SCHEDULED: ::String
+    DEAD: ::String
+    METADATA: Hash[::String, ::String]
+    RETENTION_CHANGE: Integer
+
+    def self.scheduled_stream: () -> Hash[Symbol, untyped]?
+
+    def self.scheduled_consumer: () -> Hash[Symbol, untyped]?
+
+    def self.dead_stream: () -> Hash[Symbol, untyped]?
+
+    def self.setup!: () -> Array[::String]
+
+    def self.setup_stream: (::String name, Hash[Symbol, untyped] config) -> void
+  end
+end

data/sig/cosmo/stream/processor.rbs

@@ -13,6 +13,8 @@ module Cosmo
 
       def fetch_subjects: (Hash[Symbol, untyped] config) -> untyped
 
+      def backoff_key: (Hash[Symbol, untyped] config) -> String
+
       def fetch_timeout: (Hash[Symbol, untyped] config) -> Float
 
       def static_config: -> Array[Hash[Symbol, untyped]]

data/sig/cosmo/utils/hash.rbs

@@ -7,15 +7,14 @@ module Cosmo
 
       def self.dup: [T] (T hash) -> T
 
-      def self.set: (::Hash[untyped, untyped] hash, *untyped keys, value: untyped) -> untyped
-
       def symbolize_keys!: (untyped obj) -> (::Hash[Symbol, untyped] | Array[untyped] | untyped)
 
       def stringify_keys: (untyped obj) -> (::Hash[String, untyped] | Array[untyped] | untyped)
 
+      def deep_merge: (::Hash[untyped, untyped] base, ::Hash[untyped, untyped] other) -> ::Hash[untyped, untyped]
+
       def dup: [T] (T hash) -> T
 
-      def set: (::Hash[untyped, untyped] hash, *untyped keys, value: untyped) -> untyped
     end
   end
 end

data/sig/cosmo.rbs

@@ -1,6 +1,10 @@
 module Cosmo
   VERSION: ::String
 
+  def self.configure: () { (Config config) -> void } -> void
+
+  def self.handle_error: (Exception error, Hash[Symbol, untyped] context) -> void
+
   class Error < StandardError
   end
 
@@ -10,10 +14,17 @@ module Cosmo
   class NotImplementedError < Error
   end
 
+  class ConfigError < Error
+  end
+
   class ConfigNotFoundError < Error
     def initialize: (::String config_file) -> void
   end
 
+  class SchedulingDisabledError < Error
+    def initialize: () -> void
+  end
+
   class StreamNotFoundError < Error
     def initialize: (::String stream_name) -> void
   end