pigato 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 59fe6ec27b63289331f1bcc8dfe5bd56a273861e
-  data.tar.gz: d196156fb32ed9b50337a1e6b7d2af5d2aaaa5f2
+  metadata.gz: 084cf3ce0c185586f92f4801285553e68e23085d
+  data.tar.gz: bdffbb4efc8e86d3ffb2ed669fade66987c4e8a9
 SHA512:
-  metadata.gz: c9c083863c48629c7b74f878af92f9b81d5a5aba14f9c585293a53cad91208baec1db9cc6d9064f20c05db2465ff9d2af0d67dea2e4e5c57eaba03238aceb3c0
-  data.tar.gz: 7a6065cba43176980abb835ee529e4199e82e38ea89628cf995c29587aaab01c10b9fa3385dd7b689d7e828df96730e7fe62da2c49389a9cc3ad5ca7a7c65d42
+  metadata.gz: b6a2f6ccb48abb66956e8695079b07e53ca372b7ecb12ae6558935c5e3d10c4dda831a6035b0d055d401199bfec1ea894b84c0d1d8cd11db37896914437e1375
+  data.tar.gz: 427d7dec7fe448f198a72f3c2c0eff322bf9729dd30c5d8724e3bd83d12fb1bb8a81b38bdaf5cf55305c78f2fa4cb8fc5451b68c96ab17bb65761e48ef4e882b

data/examples/echo_worker.rb

@@ -0,0 +1,12 @@
+#!/usr/bin/env ruby
+
+require "rubygems"
+require "#{File.dirname(__FILE__)}/../lib/pigato.rb"
+
+worker = PigatoWorker.new('tcp://localhost:55555', 'echo')
+reply = nil
+
+loop do
+  request = worker.recv reply
+  worker.reply request
+end

data/lib/pigato/version.rb

@@ -1,3 +1,3 @@
 module Pigato
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

data/lib/pigato/worker.rb

@@ -0,0 +1,111 @@
+require "json"
+require "ffi-rzmq"
+
+class PigatoWorker
+  HEARTBEAT_LIVENESS = 3 # 3-5 is reasonable
+
+  def initialize broker, service
+    @broker = broker
+    @service = service
+    @context = ZMQ::Context.new(1)
+    @poller = ZMQ::Poller.new
+    @worker = nil # Socket to broker
+    @heartbeat_at = 0 # When to send HEARTBEAT (relative to time.time(), so in seconds)
+    @liveness = 0 # How many attempts left
+    @timeout = 2500
+    @heartbeat = 2500 # Heartbeat delay, msecs
+    @reconnect = 2500 # Reconnect delay, msecs
+
+    @reply_to = nil
+    @reply_rid = nil
+    @reply_service = nil
+
+    reconnect_to_broker
+  end
+  
+  def reply reply
+    reply = [@reply_to, '', @reply_rid, '0'].concat([reply.to_json])
+    send_to_broker Pigato::W_REPLY, reply, nil
+  end
+
+  def recv reply
+    loop do
+      @reply_rid = nil
+      @reply_to = nil
+      @reply_service = nil
+
+      items = @poller.poll(@timeout) 
+      if items
+        messages = []
+        @worker.recv_strings messages
+
+        @liveness = HEARTBEAT_LIVENESS
+
+        header = messages.shift
+        if header != Pigato::W_WORKER
+          puts "E: Header is not Pigato::WORKER"
+        end
+
+        command = messages.shift
+
+        case command
+        when Pigato::W_REQUEST
+          # We should pop and save as many addresses as there are
+          # up to a null part, but for now, just save one...
+          puts "REQUEST"
+          @reply_to = messages.shift
+          @reply_service = messages.shift
+          messages.shift # empty
+          @reply_rid = messages.shift
+          return messages[0] # We have a request to process
+        when Pigato::W_HEARTBEAT
+          # do nothing
+        when Pigato::W_DISCONNECT
+          reconnect_to_broker
+        else
+        end
+      else
+        @liveness -= 1
+        if @liveness == 0
+          sleep 0.001*@reconnect
+          reconnect_to_broker
+        end
+      end
+
+      if Time.now > @heartbeat_at
+        send_to_broker Pigato::W_HEARTBEAT
+        @heartbeat_at = Time.now + 0.001 * @heartbeat
+      end
+
+    end
+  end
+
+  def reconnect_to_broker
+    if @worker
+      @poller.deregister @worker, ZMQ::DEALER
+      @worker.close
+    end
+
+    @worker = @context.socket ZMQ::DEALER
+    @worker.setsockopt ZMQ::IDENTITY, 'WRK01'
+    @worker.setsockopt ZMQ::LINGER, 0 
+    @worker.connect @broker
+    @poller.register @worker, ZMQ::POLLIN
+    send_to_broker(Pigato::W_READY, @service, [])
+    @liveness = HEARTBEAT_LIVENESS
+    @heartbeat_at = Time.now + 0.001 * @heartbeat
+  end
+
+  def send_to_broker command, message=nil, options=nil
+    if message.nil?
+      message = []
+    elsif not message.is_a?(Array)
+      message = [message]
+    end
+
+    message = [Pigato::W_WORKER, command].concat message
+    message = message.concat(options) if options
+    
+    @worker.send_strings message
+  end
+end

data/lib/pigato.rb

@@ -1,3 +1,4 @@
 require_relative "pigato/version.rb"
 require_relative "pigato/proto.rb"
 require_relative "pigato/client.rb"
+require_relative "pigato/worker.rb"

data/pigato.gemspec

@@ -21,4 +21,5 @@ Gem::Specification.new do |spec|
 
   spec.add_development_dependency "bundler", "~> 1.8"
   spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_runtime_dependency "ffi-rzmq", "~> 2.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pigato
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Paolo Ardoino
@@ -38,6 +38,20 @@ dependencies:
     - - ~>
       - !ruby/object:Gem::Version
         version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: ffi-rzmq
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.0'
 description: PIGATO-RUBY
 email:
 - paolo.ardoino@gmail.com
@@ -53,11 +67,13 @@ files:
 - Rakefile
 - bin/console
 - bin/setup
-- examples/client.rb
+- examples/echo_client.rb
+- examples/echo_worker.rb
 - lib/pigato.rb
 - lib/pigato/client.rb
 - lib/pigato/proto.rb
 - lib/pigato/version.rb
+- lib/pigato/worker.rb
 - pigato.gemspec
 homepage: https://github.com/prdn/pigato
 licenses: