nsq-ruby-fastly 2.4.0

data/lib/nsq/frames/message.rb

@@ -0,0 +1,33 @@
+require_relative 'frame'
+
+module Nsq
+  class Message < Frame
+
+    attr_reader :attempts
+    attr_reader :id
+    attr_reader :body
+
+    def initialize(data, connection)
+      super
+      @timestamp_in_nanoseconds, @attempts, @id, @body = @data.unpack('Q>S>a16a*')
+      @body.force_encoding('UTF-8')
+    end
+
+    def finish
+      connection.fin(id)
+    end
+
+    def requeue(timeout = 0)
+      connection.req(id, timeout)
+    end
+
+    def touch
+      connection.touch(id)
+    end
+
+    def timestamp
+      Time.at(@timestamp_in_nanoseconds / 1_000_000_000.0)
+    end
+
+  end
+end

data/lib/nsq/frames/response.rb

@@ -0,0 +1,6 @@
+require_relative 'frame'
+
+module Nsq
+  class Response < Frame
+  end
+end

data/lib/nsq/logger.rb

@@ -0,0 +1,38 @@
+require 'logger'
+module Nsq
+  @@logger = Logger.new(nil)
+
+
+  def self.logger
+    @@logger
+  end
+
+
+  def self.logger=(new_logger)
+    @@logger = new_logger
+  end
+
+
+  module AttributeLogger
+    def self.included(klass)
+      klass.send :class_variable_set, :@@log_attributes, []
+    end
+
+    %w(fatal error warn info debug).map{|m| m.to_sym}.each do |level|
+      define_method level do |msg|
+        Nsq.logger.send(level, "#{prefix} #{msg}")
+      end
+    end
+
+
+    private
+    def prefix
+      attrs = self.class.send(:class_variable_get, :@@log_attributes)
+      if attrs.count > 0
+        "[#{attrs.map{|a| "#{a.to_s}: #{self.send(a)}"}.join(' ')}] "
+      else
+        ''
+      end
+    end
+  end
+end

data/lib/nsq/producer.rb

@@ -0,0 +1,94 @@
+require_relative 'client_base'
+
+module Nsq
+  class Producer < ClientBase
+    attr_reader :topic
+
+    def initialize(opts = {})
+      @connections = {}
+      @topic = opts[:topic]
+      @discovery_interval = opts[:discovery_interval] || 60
+      @ssl_context = opts[:ssl_context]
+      @tls_options = opts[:tls_options]
+      @tls_v1 = opts[:tls_v1]
+
+      nsqlookupds = []
+      if opts[:nsqlookupd]
+        nsqlookupds = [opts[:nsqlookupd]].flatten
+        discover_repeatedly(
+          nsqlookupds: nsqlookupds,
+          interval: @discovery_interval
+        )
+
+      elsif opts[:nsqd]
+        nsqds = [opts[:nsqd]].flatten
+        nsqds.each{|d| add_connection(d)}
+
+      else
+        add_connection('127.0.0.1:4150')
+      end
+    end
+
+    def write(*raw_messages)
+      if !@topic
+        raise 'No topic specified. Either specify a topic when instantiating the Producer or use write_to_topic.'
+      end
+
+      write_to_topic(@topic, *raw_messages)
+    end
+
+    # Arg 'delay' in seconds
+    def deferred_write(delay, *raw_messages)
+      if !@topic
+        raise 'No topic specified. Either specify a topic when instantiating the Producer or use write_to_topic.'
+      end
+      if delay < 0.0
+        raise "Delay can't be negative, use a positive float."
+      end
+
+      deferred_write_to_topic(@topic, delay, *raw_messages)
+    end
+
+    def write_to_topic(topic, *raw_messages)
+      # return error if message(s) not provided
+      raise ArgumentError, 'message not provided' if raw_messages.empty?
+
+      # stringify the messages
+      messages = raw_messages.map(&:to_s)
+
+      # get a suitable connection to write to
+      connection = connection_for_write
+
+      if messages.length > 1
+        connection.mpub(topic, messages)
+      else
+        connection.pub(topic, messages.first)
+      end
+    end
+
+    def deferred_write_to_topic(topic, delay, *raw_messages)
+      raise ArgumentError, 'message not provided' if raw_messages.empty?
+      messages = raw_messages.map(&:to_s)
+      connection = connection_for_write
+      messages.each do |msg|
+        connection.dpub(topic, (delay * 1000).to_i, msg)
+      end
+    end
+
+    private
+    def connection_for_write
+      # Choose a random Connection that's currently connected
+      # Or, if there's nothing connected, just take any random one
+      connections_currently_connected = connections.select{|_,c| c.connected?}
+      connection = connections_currently_connected.values.sample || connections.values.sample
+
+      # Raise an exception if there's no connection available
+      unless connection
+        raise 'No connections available'
+      end
+
+      connection
+    end
+
+  end
+end

data/lib/nsq.rb

@@ -0,0 +1,13 @@
+require_relative 'version'
+
+require_relative 'nsq/logger'
+
+require_relative 'nsq/exceptions'
+
+require_relative 'nsq/frames/frame'
+require_relative 'nsq/frames/error'
+require_relative 'nsq/frames/response'
+require_relative 'nsq/frames/message'
+
+require_relative 'nsq/consumer'
+require_relative 'nsq/producer'

data/lib/version.rb

@@ -0,0 +1,9 @@
+module Nsq
+  module Version
+    MAJOR = 2
+    MINOR = 3
+    PATCH = 1
+    BUILD = nil
+    STRING = [MAJOR, MINOR, PATCH, BUILD].compact.join('.')
+  end
+end

metadata

@@ -0,0 +1,117 @@
+--- !ruby/object:Gem::Specification
+name: nsq-ruby-fastly
+version: !ruby/object:Gem::Version
+  version: 2.4.0
+platform: ruby
+authors:
+- Wistia, Fastly
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2022-02-23 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: jeweler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: nsq-cluster
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+description: ''
+email: team-internaleng@fastly.com
+executables: []
+extensions: []
+extra_rdoc_files:
+- CHANGELOG.md
+- LICENSE.txt
+- README.md
+files:
+- CHANGELOG.md
+- LICENSE.txt
+- README.md
+- lib/nsq.rb
+- lib/nsq/client_base.rb
+- lib/nsq/connection.rb
+- lib/nsq/consumer.rb
+- lib/nsq/discovery.rb
+- lib/nsq/exceptions.rb
+- lib/nsq/frames/error.rb
+- lib/nsq/frames/frame.rb
+- lib/nsq/frames/message.rb
+- lib/nsq/frames/response.rb
+- lib/nsq/logger.rb
+- lib/nsq/producer.rb
+- lib/version.rb
+homepage: http://github.com/fastly/nsq-ruby
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.1.6
+signing_key:
+specification_version: 4
+summary: Ruby client library for NSQ
+test_files: []