wallet.rb in peatio-nexbit-0.1.3

- old
+ new
@@ -1,60 +1,58 @@
-module Peatio
-  module Nexbit
-    class Wallet < Peatio::Wallet::Abstract
+module Nexbit
+  class Wallet < Peatio::Wallet::Abstract
 
-      def initialize(settings = {})
-        @settings = settings
-      end
+    def initialize(settings = {})
+      @settings = settings
+    end
 
-      def configure(settings = {})
-        # Clean client state during configure.
-        @client = nil
+    def configure(settings = {})
+      # Clean client state during configure.
+      @client = nil
 
-        @settings.merge!(settings.slice(*SUPPORTED_SETTINGS))
+      @settings.merge!(settings.slice(*SUPPORTED_SETTINGS))
 
-        @wallet = @settings.fetch(:wallet) do
-          raise Peatio::Wallet::MissingSettingError, :wallet
-        end.slice(:uri, :address)
+      @wallet = @settings.fetch(:wallet) do
+        raise Peatio::Wallet::MissingSettingError, :wallet
+      end.slice(:uri, :address)
 
-        @currency = @settings.fetch(:currency) do
-          raise Peatio::Wallet::MissingSettingError, :currency
-        end.slice(:id, :base_factor, :options)
-      end
+      @currency = @settings.fetch(:currency) do
+        raise Peatio::Wallet::MissingSettingError, :currency
+      end.slice(:id, :base_factor, :options)
+    end
 
-      def create_address!(_options = {})
-        { address: client.json_rpc(:getnewaddress) }
-      rescue Nexbit::Client::Error => e
-        raise Peatio::Wallet::ClientError, e
-      end
+    def create_address!(_options = {})
+      { address: client.json_rpc(:getnewaddress) }
+    rescue Nexbit::Client::Error => e
+      raise Peatio::Wallet::ClientError, e
+    end
 
-      def create_transaction!(transaction, options = {})
-        txid = client.json_rpc(:sendtoaddress,
-                               [
-                                 transaction.to_address,
-                                 transaction.amount,
-                                 '',
-                                 '',
-                                 options[:subtract_fee].to_s == 'true' # subtract fee from transaction amount.
-                               ])
-        transaction.hash = txid
-        transaction
-      rescue Nexbit::Client::Error => e
-        raise Peatio::Wallet::ClientError, e
-      end
+    def create_transaction!(transaction, options = {})
+      txid = client.json_rpc(:sendtoaddress,
+                             [
+                               transaction.to_address,
+                               transaction.amount,
+                               '',
+                               '',
+                               options[:subtract_fee].to_s == 'true' # subtract fee from transaction amount.
+                             ])
+      transaction.hash = txid
+      transaction
+    rescue Nexbit::Client::Error => e
+      raise Peatio::Wallet::ClientError, e
+    end
 
-      def load_balance!
-        client.json_rpc(:getbalance).to_d
+    def load_balance!
+      client.json_rpc(:getbalance).to_d
 
-      rescue Nexbit::Client::Error => e
-        raise Peatio::Wallet::ClientError, e
-      end
+    rescue Nexbit::Client::Error => e
+      raise Peatio::Wallet::ClientError, e
+    end
 
-      private
+    private
 
-      def client
-        uri = @wallet.fetch(:uri) { raise Peatio::Wallet::MissingSettingError, :uri }
-        @client ||= Client.new(uri)
-      end
+    def client
+      uri = @wallet.fetch(:uri) { raise Peatio::Wallet::MissingSettingError, :uri }
+      @client ||= Client.new(uri, idle_timeout: 1)
     end
   end
 end