peatio-jruby 2.6.2

data/lib/peatio/blockchain/error.rb

@@ -0,0 +1,37 @@
+module Peatio
+  module Blockchain
+    Error = Class.new(StandardError)
+
+    class ClientError < Error
+
+      attr_reader :wrapped_ex
+
+      def initialize(ex_or_string)
+        @wrapped_ex = nil
+
+        if ex_or_string.respond_to?(:backtrace)
+          super(ex_or_string.message)
+          @wrapped_exception = ex_or_string
+        else
+          super(ex_or_string.to_s)
+        end
+      end
+    end
+
+    class MissingSettingError < Error
+      def initialize(key)
+        super "#{key.capitalize} setting is missing"
+      end
+    end
+
+    class UnavailableAddressBalanceError < Error
+      def initialize(address)
+        @address = address
+      end
+
+      def message
+        "Unable to load #{@address} balance"
+      end
+    end
+  end
+end

data/lib/peatio/blockchain/registry.rb

@@ -0,0 +1,16 @@
+require "peatio/adapter_registry"
+
+module Peatio
+  module Blockchain
+
+    VERSION = "1.0.0".freeze
+
+    class << self
+      def registry
+        @registry ||= Registry.new
+      end
+    end
+    class Registry < Peatio::AdapterRegistry
+    end
+  end
+end

data/lib/peatio/command/base.rb

@@ -0,0 +1,11 @@
+module Peatio::Command
+  class Base < Clamp::Command
+    def say(str)
+      puts str
+    end
+
+    def sql_client
+      @sql_client ||= Peatio::Sql::Client.new.connect
+    end
+  end
+end

data/lib/peatio/command/db.rb

@@ -0,0 +1,20 @@
+module Peatio::Command::DB
+  class Create < Peatio::Command::Base
+    def execute
+      client = Peatio::Sql::Client.new
+      database_name = client.config.delete(:database)
+      Peatio::Sql::Schema.new(client.connect).create_database(database_name)
+    end
+  end
+
+  class Migrate < Peatio::Command::Base
+    def execute
+      Peatio::Sql::Schema.new(sql_client).create_tables
+    end
+  end
+
+  class Root < Peatio::Command::Base
+    subcommand "create", "Create database", Peatio::Command::DB::Create
+    subcommand "migrate", "Create tables", Peatio::Command::DB::Migrate
+  end
+end

data/lib/peatio/command/inject.rb

@@ -0,0 +1,13 @@
+module Peatio::Command
+  class Inject < Peatio::Command::Base
+    class PeatioEvents < Peatio::Command::Base
+      option ["-e", "--exchange"], "NAME", "exchange name to inject messages to", default: "peatio.events.ranger"
+      def execute
+        Peatio::Logger.logger.level = :debug
+        Peatio::Injectors::PeatioEvents.new.run!(exchange)
+      end
+    end
+
+    subcommand "peatio_events", "Inject peatio events in mq", PeatioEvents
+  end
+end

data/lib/peatio/command/root.rb

@@ -0,0 +1,14 @@
+require "peatio/command/base"
+require "peatio/command/service"
+require "peatio/command/db"
+require "peatio/command/inject"
+require "peatio/command/security"
+
+module Peatio
+  class Root < Command::Base
+    subcommand "db", "Database related sub-commands", Peatio::Command::DB::Root
+    subcommand "service", "Services management related sub-commands", Peatio::Command::Service::Root
+    subcommand "inject", "Data injectors", Peatio::Command::Inject
+    subcommand "security", "Security management related sub-commands", Peatio::Command::Security
+  end
+end

data/lib/peatio/command/security.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module Peatio::Command
+  class Security < Peatio::Command::Base
+    class KeyGenerator < Peatio::Command::Base
+      option "--print", :flag, "print on screen"
+      option "--path", "FOLDER", "save keypair into folder", default: "secrets"
+
+      def execute
+        keypair = Peatio::Security::KeyGenerator.new
+
+        if print?
+          puts keypair.private, keypair.public
+          puts "-----BASE64 ENCODED-----"
+          puts Base64.urlsafe_encode64(keypair.public)
+        else
+          begin
+            keypair.save(path)
+            puts "Files saved in #{File.join(path, 'rsa-key')}"
+          rescue IOError => e
+            abort("Failed saving files")
+          end
+        end
+      end
+    end
+
+    subcommand "keygen", "Generate a public private rsa key pair", KeyGenerator
+  end
+end

data/lib/peatio/command/service.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+module Peatio::Command::Service
+  class Start < Peatio::Command::Base
+    class Ranger < Peatio::Command::Base
+      option ["-e", "--exchange"], "NAME", "exchange name to inject messages to", default: "peatio.events.ranger"
+      option "--[no-]stats", :flag, "display periodically connections statistics", default: true
+      option "--stats-period", "SECONDS", "period of displaying stats in seconds", default: 30
+      def execute
+        raise ArgumentError, "JWT_PUBLIC_KEY was not specified." if ENV["JWT_PUBLIC_KEY"].to_s.empty?
+
+        key_decoded = Base64.urlsafe_decode64(ENV["JWT_PUBLIC_KEY"])
+
+        jwt_public_key = OpenSSL::PKey.read(key_decoded)
+        if jwt_public_key.private?
+          raise ArgumentError, "JWT_PUBLIC_KEY was set to private key, however it should be public."
+        end
+
+        raise "stats period missing" if stats? && !stats_period
+
+        Prometheus::Client.config.data_store = Prometheus::Client::DataStores::SingleThreaded.new()
+        registry = Prometheus::Client.registry
+
+        opts = {
+          display_stats: stats?,
+          stats_period:  stats_period.to_f,
+          metrics_port:  8082,
+          registry:      registry
+        }
+        ::Peatio::Ranger.run!(jwt_public_key, exchange, opts)
+      end
+    end
+
+    subcommand "ranger", "Start ranger process", Ranger
+  end
+
+  class Root < Peatio::Command::Base
+    subcommand "start", "Start a service", Start
+  end
+end

data/lib/peatio/error.rb

@@ -0,0 +1,18 @@
+class Peatio::Error < ::StandardError
+  @@default_code = 2000
+
+  attr :code, :text
+
+  def initialize(opts = {})
+    @code = opts[:code] || @@default_code
+    @text = opts[:text] || ""
+
+    @message = {error: {code: @code, message: @text}}
+
+    if @text != ""
+      super("#{@code}: #{text}")
+    else
+      super("#{@code}")
+    end
+  end
+end

data/lib/peatio/executor.rb

@@ -0,0 +1,64 @@
+require "mysql2"
+require "benchmark"
+
+client = Mysql2::Client.new(
+    host: "172.19.0.3",
+    username: "root",
+    password: "changeme",
+    port: 3306,
+    database: "peatio_development")
+
+queries = [
+    "INSERT INTO `trades` (`ask_id`, `ask_member_id`, `bid_id`, `bid_member_id`, `price`, `volume`, `funds`, `market_id`, `trend`, `created_at`, `updated_at`) VALUES (18711, 81, 18708, 82, 0.99999999, 50.0, 49.9999995, 'eurusd', 0, NOW(), NOW())",
+    "UPDATE `accounts` SET `accounts`.`locked` = 3571.09999702 WHERE `accounts`.`id` = 164",
+    "UPDATE `accounts` SET `accounts`.`balance` = 999995119.5335 WHERE `accounts`.`id` = 163",
+    "UPDATE `accounts` SET `accounts`.`locked` = 4257.0 WHERE `accounts`.`id` = 161",
+    "UPDATE `accounts` SET `accounts`.`balance` = 999995825.720262199325 WHERE `accounts`.`id` = 162",
+    "UPDATE `orders` SET `volume` = 20.0, `locked` = 19.9999998, `funds_received` = 53.0, `trades_count` = 2, `updated_at` = '2018-06-25 23:44:37' WHERE `orders`.`id` = 18708",
+    "UPDATE `orders` SET `volume` = 0.0, `locked` = 0.0, `funds_received` = 78.59999924, `trades_count` = 2, `state` = 200, `updated_at` = '2018-06-25 23:44:37' WHERE `orders`.`id` = 18711"
+]
+
+puts Benchmark.measure {
+    1_000.times {
+
+        client.query("begin")
+        begin
+
+            100.times {
+                queries.each do |q|
+                    client.query q
+                end
+            }
+
+        rescue Mysql2::Error => e
+            puts "+++++++ DB ERROR - ROLLING BACK ++++++++"
+            puts e
+            client.query("rollback")
+            exit
+        end
+        client.query("commit") #commit the changes to the DB
+
+    }
+}
+
+client.close
+
+__END__
+
+require 'mysql2/em'
+
+EM.run do
+   client = Mysql2::EM::Client.new(
+        :host => '172.19.0.3',
+        :username => 'root',
+        :password => 'changeme',
+        :port => 3306,
+        :database => 'peatio_development')
+
+
+    defer1 = client.query "SELECT sleep(3) as first_query"
+    defer1.callback do |result|
+        puts "Result: #{result.to_a.inspect}"
+    end
+
+end

data/lib/peatio/injectors/peatio_events.rb

@@ -0,0 +1,240 @@
+# frozen_string_literal: true
+
+module Peatio::Injectors
+  class PeatioEvents
+    attr_accessor :market, :market_name, :base_unit, :quote_unit, :seller_uid, :buyer_uid, :logger
+
+    def run!(exchange_name)
+      require "time"
+      @logger = Peatio::Logger.logger
+      @market = "eurusd"
+      @market_name = "EUR/USD"
+      @base_unit = "eur"
+      @quote_unit = "usd"
+      @seller_uid = 21
+      @buyer_uid = 42
+      @messages = create_messages
+      @opts = {
+        ex_name: exchange_name
+      }
+      EventMachine.run do
+        inject_message()
+      end
+    end
+
+    def inject_message()
+      if message = @messages.shift
+        type, id, event, data = message
+        Peatio::Ranger::Events.publish(type, id, event, data, @opts)
+        EM.next_tick do
+          inject_message()
+        end
+      else
+        Peatio::MQ::Client.disconnect
+        EventMachine.stop
+      end
+    end
+
+    def create_messages
+      [
+        public_tickers,
+        public_orderbook,
+        private_order,
+        private_trade_user1,
+        private_trade_user2,
+        public_trade,
+        public_orderbook_increment1,
+        public_orderbook_snapshot1,
+        public_orderbook_increment2,
+        public_orderbook_increment3,
+      ]
+    end
+
+    def created_at
+      Time.now - 600
+    end
+
+    def updated_at
+      Time.now
+    end
+
+    alias completed_at updated_at
+    alias canceled_at updated_at
+
+    def public_orderbook
+      [
+        "public",
+        market,
+        "update",
+        {
+          "asks": [
+            ["1020.0", "0.005"],
+            ["1026.0", "0.03"]
+          ],
+          "bids": [
+            ["1000.0", "0.25"],
+            ["999.0", "0.005"],
+            ["994.0", "0.005"],
+            ["1.0", "11.0"]
+          ]
+        }
+      ]
+    end
+
+    def public_orderbook_snapshot1
+      [
+        "public",
+        market,
+        "ob-snap",
+        {
+          "asks": [
+            ["1020.0", "0.005"],
+            ["1026.0", "0.03"]
+          ],
+          "bids": [
+            ["1000.0", "0.25"],
+            ["999.0", "0.005"],
+            ["994.0", "0.005"],
+            ["1.0", "11.0"]
+          ]
+        }
+      ]
+    end
+
+    def public_orderbook_increment1
+      [
+        "public",
+        market,
+        "ob-inc",
+        {
+          "asks": [
+            ["1020.0", "0.015"],
+          ],
+        }
+      ]
+    end
+
+    def public_orderbook_increment2
+      [
+        "public",
+        market,
+        "ob-inc",
+        {
+          "bids": [
+            ["1000.0", "0"],
+          ],
+        }
+      ]
+    end
+
+    def public_orderbook_increment3
+      [
+        "public",
+        market,
+        "ob-inc",
+        {
+          "bids": [
+            ["999.0", "0.001"],
+          ],
+        }
+      ]
+    end
+
+    def public_tickers
+      [
+        "public",
+        "global",
+        "tickers",
+        {
+          market => {
+            "name":       market_name,
+            "base_unit":  base_unit,
+            "quote_unit": quote_unit,
+            "low":        "1000.0",
+            "high":       "10000.0",
+            "last":       "1000.0",
+            "open":       1000.0,
+            "volume":     "0.0",
+            "sell":       "1020.0",
+            "buy":        "1000.0",
+            "at":         Time.now.to_i
+          }
+        }
+      ]
+    end
+
+    def private_order
+      [
+        "private",
+        "IDABC0000001",
+        "order",
+        {
+          "id":            22,
+          "at":            created_at.to_i,
+          "market":        market,
+          "kind":          "bid",
+          "price":         "1026.0",
+          "state":         "wait",
+          "volume":        "0.001",
+          "origin_volume": "0.001"
+        }
+      ]
+    end
+
+    def private_trade_user1
+      [
+        "private",
+        "IDABC0000001",
+        "trade",
+        {
+          "id":     7,
+          "kind":   "ask",
+          "at":     created_at.to_i,
+          "price":  "1020.0",
+          "volume": "0.001",
+          "ask_id": 15,
+          "bid_id": 22,
+          "market": market
+        }
+      ]
+    end
+
+    def private_trade_user2
+      [
+        "private",
+        "IDABC0000002",
+        "trade",
+        {
+          "id":     7,
+          "kind":   "bid",
+          "at":     created_at.to_i,
+          "price":  "1020.0",
+          "volume": "0.001",
+          "ask_id": 15,
+          "bid_id": 22,
+          "market": market
+        }
+      ]
+    end
+
+    def public_trade
+      [
+        "public",
+        market,
+        "trades",
+        {
+          "trades": [
+            {
+              "tid":        7,
+              "taker_type": "buy",
+              "date":       created_at.to_i,
+              "price":      "1020.0",
+              "amount":
+                            "0.001"
+            }
+          ]
+        }
+      ]
+    end
+  end
+end