peatio1.9 0.4.4

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,11 @@
+module Peatio::Command
+  class Inject < Peatio::Command::Base
+    class PeatioEvents < Peatio::Command::Base
+      def execute
+        Peatio::Injectors::PeatioEvents.new.run!
+      end
+    end
+
+    subcommand "peatio_events", "Inject peatio events in mq", PeatioEvents
+  end
+end

data/lib/peatio/command/root.rb

@@ -0,0 +1,16 @@
+require "peatio/command/base"
+require "peatio/command/service"
+require "peatio/command/db"
+require "peatio/command/amqp"
+require "peatio/command/inject"
+require "peatio/command/security"
+
+module Peatio
+  class Root < Command::Base
+    subcommand "amqp", "AMQP related sub-commands", Peatio::Command::AMQP::Root
+    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,30 @@
+require 'base64'
+
+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,26 @@
+module Peatio::Command::Service
+  class Start < Peatio::Command::Base
+    class Ranger < Peatio::Command::Base
+      def execute
+        if ENV["JWT_PUBLIC_KEY"].nil?
+          raise ArgumentError, "JWT_PUBLIC_KEY was not specified."
+        end
+
+        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
+
+        ::Peatio::Ranger.run!(jwt_public_key)
+      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,209 @@
+module Peatio::Injectors
+  class PeatioEvents
+    attr_accessor :market, :seller_uid, :buyer_uid, :logger
+
+    def run!
+      require "time"
+      @logger = Peatio::Logger.logger
+      @market = "eurusd"
+      @seller_uid = 21
+      @buyer_uid = 42
+      @messages = create_messages
+      @exchange_name = "peatio.events.market"
+
+      EventMachine.run do
+        Peatio::MQ::Client.new
+        Peatio::MQ::Client.connect!
+        Peatio::MQ::Client.create_channel!
+        inject_message
+      end
+    end
+
+    def inject_message
+      if message = @messages.shift
+        type, id, event, data = message
+        Peatio::MQ::Events.publish(type, id, event, data) {
+          inject_message
+        }
+      else
+        Peatio::MQ::Client.disconnect { EventMachine.stop }
+      end
+    end
+
+    def create_messages
+      [
+        private_trade,
+        order_created,
+        order_canceled,
+        order_completed,
+        order_updated,
+        trade_completed,
+      ]
+    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 private_trade
+      [
+        "private",
+        "debug_user",
+        "trade",
+        {
+          trade: "some-data",
+        },
+      ]
+    end
+
+    def order_created
+      [
+        "public",
+        market,
+        "order_created",
+        {
+          market: "#{market}",
+          type: "buy",
+          trader_uid: buyer_uid,
+          income_unit: "btc",
+          income_fee_type: "relative",
+          income_fee_value: "0.0015",
+          outcome_unit: "usd",
+          outcome_fee_type: "relative",
+          outcome_fee_value: "0.0",
+          initial_income_amount: "14.0",
+          current_income_amount: "14.0",
+          initial_outcome_amount: "0.42",
+          current_outcome_amount: "0.42",
+          strategy: "limit",
+          price: "0.03",
+          state: "open",
+          trades_count: 0,
+          created_at: created_at.iso8601,
+        },
+      ]
+    end
+
+    def order_canceled
+      [
+        "public",
+        market,
+        "order_canceled",
+        {
+          market: "#{market}",
+          type: "sell",
+          trader_uid: seller_uid,
+          income_unit: "usd",
+          income_fee_type: "relative",
+          income_fee_value: "0.0015",
+          outcome_unit: "btc",
+          outcome_fee_type: "relative",
+          outcome_fee_value: "0.0",
+          initial_income_amount: "3.0",
+          current_income_amount: "3.0",
+          initial_outcome_amount: "100.0",
+          current_outcome_amount: "100.0",
+          strategy: "limit",
+          price: "0.03",
+          state: "canceled",
+          trades_count: 0,
+          created_at: created_at.iso8601,
+          canceled_at: canceled_at.iso8601,
+        },
+      ]
+    end
+
+    def order_completed
+      [
+        "public",
+        market,
+        "order_completed", {
+          market: "#{market}",
+          type: "sell",
+          trader_uid: seller_uid,
+          income_unit: "usd",
+          income_fee_type: "relative",
+          income_fee_value: "0.0015",
+          outcome_unit: "btc",
+          outcome_fee_type: "relative",
+          outcome_fee_value: "0.0",
+          initial_income_amount: "3.0",
+          current_income_amount: "0.0",
+          previous_income_amount: "3.0",
+          initial_outcome_amount: "100.0",
+          current_outcome_amount: "0.0",
+          previous_outcome_amount: "100.0",
+          strategy: "limit",
+          price: "0.03",
+          state: "completed",
+          trades_count: 1,
+          created_at: created_at.iso8601,
+          completed_at: completed_at.iso8601,
+        },
+      ]
+    end
+
+    def order_updated
+      [
+        "public",
+        market,
+        "order_updated", {
+          market: "#{market}",
+          type: "sell",
+          trader_uid: seller_uid,
+          income_unit: "usd",
+          income_fee_type: "relative",
+          income_fee_value: "0.0015",
+          outcome_unit: "btc",
+          outcome_fee_type: "relative",
+          outcome_fee_value: "0.0",
+          initial_income_amount: "3.0",
+          current_income_amount: "2.4",
+          previous_income_amount: "3.0",
+          initial_outcome_amount: "100.0",
+          current_outcome_amount: "80.0",
+          previous_outcome_amount: "100.0",
+          strategy: "limit",
+          price: "0.03",
+          state: "open",
+          trades_count: 1,
+          created_at: created_at.iso8601,
+          updated_at: updated_at.iso8601,
+        },
+      ]
+    end
+
+    def trade_completed
+      [
+        "public",
+        market,
+        "trade_completed", {
+          market: "#{market}",
+          price: "0.03",
+          buyer_uid: buyer_uid,
+          buyer_income_unit: "btc",
+          buyer_income_amount: "14.0",
+          buyer_income_fee: "0.021",
+          buyer_outcome_unit: "usd",
+          buyer_outcome_amount: "0.42",
+          buyer_outcome_fee: "0.0",
+          seller_uid: seller_uid,
+          seller_income_unit: "usd",
+          seller_income_amount: "0.42",
+          seller_income_fee: "0.00063",
+          seller_outcome_unit: "btc",
+          seller_outcome_amount: "14.0",
+          seller_outcome_fee: "0.0",
+          completed_at: completed_at.iso8601,
+        },
+      ]
+    end
+  end
+end

data/lib/peatio/logger.rb

@@ -0,0 +1,37 @@
+module Peatio
+  class Logger
+    class << self
+      def logger
+        @logger ||= ::Logger.new(STDERR, level: level)
+      end
+
+      def level
+        (ENV["LOG_LEVEL"] || "debug").downcase.to_sym
+      end
+
+      def debug(progname = nil, &block)
+        logger.debug(progname, &block)
+      end
+
+      def info(progname = nil, &block)
+        logger.info(progname, &block)
+      end
+
+      def warn(progname = nil, &block)
+        logger.warn(progname, &block)
+      end
+
+      def error(progname = nil, &block)
+        logger.error(progname, &block)
+      end
+
+      def fatal(progname = nil, &block)
+        logger.fatal(progname, &block)
+      end
+
+      def unknown(progname = nil, &block)
+        logger.unknown(progname, &block)
+      end
+    end
+  end
+end