irix 2.4.2 → 2.4.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c95445e49c906a31bbc9e16f2b4b78ca8d023fca73f5d29128d9531ab0112c18
-  data.tar.gz: 79e6e7f4d815726d6624e4d79c0a0d8db2ebed9bda2534645c29b40bbffbf7d4
+  metadata.gz: 66522df660b1ad84d6085e2c85ad9560b6caeca36964d302270bfdcbaba01a16
+  data.tar.gz: f1f81a8dd46e33d1fc118ad77df41afec7ba40bdc5f7fb9e79a35128b2adf5b2
 SHA512:
-  metadata.gz: '0509583f671a0de4396eaec6ae5cc524e0bf4924afb4186c7efa6487dfff53fcc657eccfdf53158a1e81deb3e24b7937a066a4c0e65669b209fda92c1c507133'
-  data.tar.gz: d77b1f4410a5ff9d7ce1268aa30ea7e16556534f5684b3b14ddb241c6d9774e4c19518d92484d34cb8525818b53c3af72954f48eaf7782c1467bf6d3eef2b8d5
+  metadata.gz: 379e9c8f811ba1cf5ce927553551c3f52228217b2c064b475c3e772a2fe8d80b6d6bdc004a121660275094f0cc2fcd2f87d49eed1d9eb1e7b3149cab58d9ce6a
+  data.tar.gz: 6bdb2cda4714b4f14f813edd915bf921954b0c6379921a4ed943e82a9ce3f3e70abecd7121f1d05fef032ba26cad849bc4085864a5e3ea18421e969e24075b8a

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    irix (2.4.2)
+    irix (2.4.3)
       em-synchrony (~> 1.0)
       em-websocket
       eventmachine
@@ -85,7 +85,7 @@ GEM
     parallel (1.19.0)
     parser (2.6.5.0)
       ast (~> 2.4.0)
-    peatio (2.4.2)
+    peatio (2.4.3)
       activemodel (> 5.2, <= 6.0.0)
       amqp
       bunny

data/lib/irix/bitfinex.rb

@@ -2,6 +2,15 @@
 
 module Irix
   class Bitfinex < Peatio::Upstream::Base
+    require 'time'
+
+    MIN_INCREMENT_COUNT_TO_SNAPSHOT = 100
+    MIN_PERIOD_TO_SNAPSHOT = 5
+    MAX_PERIOD_TO_SNAPSHOT = 60
+
+    attr_accessor :snap, :snapshot_time, :increment_count, :sequence_number,
+                  :open_channels, :asks, :bids
+
     def initialize(config)
       super
       @connection = Faraday.new(url: (config['rest']).to_s) do |builder|
@@ -12,15 +21,22 @@ module Irix
           builder.ssl[:verify] = config['verify_ssl']
         end
       end
+      @open_channels = {}
       @ping_set = false
       @rest = (config['rest']).to_s
-      @ws_url = "#{config['websocket']}"
+      @ws_url = (config['websocket']).to_s
     end
 
     def ws_connect
       super
       return if @ping_set
 
+      @ws.on(:open) do |_e|
+        subscribe_trades(@target, @ws)
+        subscribe_orderbook(@target, @ws)
+        logger.info { 'Websocket connected' }
+      end
+
       Fiber.new do
         EM::Synchrony.add_periodic_timer(80) do
           @ws.send('{"event":"ping"}')
@@ -30,6 +46,8 @@ module Irix
     end
 
     def subscribe_trades(market, ws)
+      return unless @config['trade_proxy']
+
       sub = {
         event: 'subscribe',
         channel: 'trades',
@@ -42,9 +60,40 @@ module Irix
       end
     end
 
+    def subscribe_orderbook(market, ws)
+      return unless @config['orderbook_proxy']
+
+      @sequence_number = 0
+      @increment_count = 0
+      @bids = []
+      @asks = []
+      @snap = { 'asks' => [], 'bids' => [] }
+      sub = {
+        event: 'subscribe',
+        channel: 'book',
+        symbol: market.upcase,
+        len: 25
+      }
+      Rails.logger.info 'Open event' + sub.to_s
+      EM.next_tick do
+        ws.send(JSON.generate(sub))
+      end
+      Fiber.new do
+        EM::Synchrony.add_periodic_timer(0.2) do
+          publish_increment
+        end
+      end.resume
+    end
+
     def ws_read_public_message(msg)
       if msg.is_a?(Array)
-        detect_trade(msg)
+        if msg[1] == 'hb'
+          @ws.send('{"event":"ping"}')
+        elsif @open_channels[msg[0]] == 'trades'
+          detect_trade(msg)
+        elsif @open_channels[msg[0]] == 'book'
+          detect_order(msg)
+        end
       elsif msg.is_a?(Hash)
         message_event(msg)
       end
@@ -65,10 +114,85 @@ module Irix
       end
     end
 
+    # [
+    #   CHANNEL_ID,
+    #   [
+    #     PRICE,
+    #     COUNT,
+    #     AMOUNT
+    #   ]
+    # ]
+    def detect_order(msg)
+      if msg[1][0].is_a?(Array)
+        msg[1].each do |point|
+          if point[2] > 0
+            @snap['bids'] << [point[0].to_s, point[2].to_s]
+          else
+            @snap['asks'] << [point[0].to_s, point[2].abs.to_s]
+          end
+        end
+        publish_snapshot
+      else
+        if @increment_count < MIN_INCREMENT_COUNT_TO_SNAPSHOT && @snapshot_time <= Time.now - MAX_PERIOD_TO_SNAPSHOT
+          publish_snapshot
+          @increment_count = 0
+        elsif @increment_count >= MIN_INCREMENT_COUNT_TO_SNAPSHOT && @snapshot_time < Time.now - MIN_PERIOD_TO_SNAPSHOT
+          publish_snapshot
+          @increment_count = 0
+        end
+
+        fill_increment(msg[1])
+      end
+    end
+
+    def fill_increment(order)
+      side = order[2].positive? ? 'bid' : 'ask'
+      price = order[0].to_s
+      if order[1].zero?
+        amount = 0
+        @snap["#{side}s"].delete_if { |point| point[0] == price }
+      else
+        amount = order[2].abs.to_s
+        @snap["#{side}s"].delete_if { |point| point[0] == price }
+        @snap["#{side}s"] << [price.to_s, amount.to_s]
+      end
+      if side == 'bid'
+        @bids.delete_if { |point| point[0] == price }
+        @bids << [price.to_s, amount.to_s]
+      elsif side == 'ask'
+        @asks.delete_if { |point| point[0] == price }
+        @asks << [price.to_s, amount.to_s]
+      end
+      @increment_count += 1
+    end
+
+    def publish_increment
+      inc = {}
+      inc['bids'] = @bids.sort.reverse if @bids.present?
+      inc['asks'] = @asks.sort if @asks.present?
+      if inc.present?
+        @sequence_number += 1
+        @peatio_mq.enqueue_event('public', @market, 'ob-inc',
+                                 'bids' => inc['bids'], 'asks' => inc['asks'],
+                                 'sequence' => @sequence_number)
+      end
+      @bids = []
+      @asks = []
+    end
+
+    def publish_snapshot
+      @snapshot_time = Time.now
+      @peatio_mq.enqueue_event('public', @market, 'ob-snap',
+                               'bids' => @snap['bids'].sort.reverse,
+                               'asks' => @snap['asks'].sort,
+                               'sequence' => @sequence_number)
+    end
+
     def message_event(msg)
       case msg['event']
       when 'subscribed'
         Rails.logger.info "Event: #{msg}"
+        @open_channels[msg['chanId']] = msg['channel']
       when 'error'
         Rails.logger.info "Event: #{msg} ignored"
       end

data/lib/irix/version.rb

@@ -1,3 +1,3 @@
 module Irix
-  VERSION = "2.4.2"
+  VERSION = "2.4.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: irix
 version: !ruby/object:Gem::Version
-  version: 2.4.2
+  version: 2.4.3
 platform: ruby
 authors:
 - Naichuk M.
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-03-24 00:00:00.000000000 Z
+date: 2020-04-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: em-synchrony