Contents

class GooglePubsubEnhancer::Middleware::Publisher

  def initialize(app, opts={})
    @app = app
    @short_topic_name = opts[:short_topic_name] || raise
    @full_topic_name = GooglePubsubEnhancer.name_by('topics',@short_topic_name)
    @messages_key = opts[:messages] || raise
    @logger = opts[:logger] || Logger.new(STDOUT)
    @google_cloud_pubsub ||= Google::Cloud::Pubsub.new
  end

  def call(env)
    begin
      @logger.debug("#{env[@messages_key].length} messages published")
      @google_cloud_pubsub.publish(@full_topic_name) do |publisher|
        [*env[@messages_key]].each do |m|
          publisher.publish(m, {recordId: Digest::MD5.hexdigest(m)} )
        end
      end
    rescue => ex
      @logger.error("Retry publisher: #{ex}")
      retry
    end
    @app.call(env)
  end

end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
google-pubsub-enhancer-0.6.2	lib/google_pubsub_enhancer/middleware/publisher.rb
google-pubsub-enhancer-0.6.1	lib/google_pubsub_enhancer/middleware/publisher.rb
google-pubsub-enhancer-0.6.0	lib/google_pubsub_enhancer/middleware/publisher.rb
google-pubsub-enhancer-0.5.7	lib/google_pubsub_enhancer/middleware/publisher.rb