istox 0.1.103 → 0.1.104

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2a4fdf1c4ef6e43e08853ee0921fe24c7a5055954f09e8994947d319f95e127f
-  data.tar.gz: fb11c641a84a1f285ef85ea5ad532b3896d3db6971f65ce4c83ec0970c43ed9d
+  metadata.gz: 017e8b356e88bb9934ee5e01dce3ef6ce1ccdbf937bd5e8463ccd138c074ccee
+  data.tar.gz: 606737426dc9eaf9ca1b3d1f879b0dda17f46dacc89817cacf80990ce8a9fad2
 SHA512:
-  metadata.gz: 4bec68b057c3f448e08f4958b7a51964cbda8b99a5689028107928e2b5a385914e78ff703d74ac43542e4f1c93b615be1c3a0a07a130c9fa4264df223be67fe9
-  data.tar.gz: 9f5d910541029a58e96bb1ef1955671b0ba431bb68acdd59de4e8da7a957485a1e68ad92e79ef278f80dc13f640a9e5c81b148e2a62f4eec4e79ea29ca007b58
+  metadata.gz: 38bd1dcc56ce327d8c994f5743310cf3e34d3f70639513dcdfdcb0bd2a9b6a3c5cdef19e5b3b26c4ac43d8ee5742146cfb798b73be17fd4f9838bb14c7b5013c
+  data.tar.gz: d338cf9fe623a1a99481581c57142a5637899509f319407d489662ce8c511318d5a5747a8f091072f17a796bc51038382162278b89b5a72b0cd9c1a739e506e2

data/.rubocop.yml

@@ -74,7 +74,7 @@ Metrics/ParameterLists:
   CountKeywordArgs: true
 
 Metrics/PerceivedComplexity:
-  Max: 12
+  Max: 20
 
 Style/FrozenStringLiteralComment:
   Enabled: false

data/Gemfile.lock

@@ -1,13 +1,15 @@
 PATH
   remote: .
   specs:
-    istox (0.1.102)
+    istox (0.1.103)
+      awesome_print
       binding_of_caller
       bunny (>= 2.12.0)
       graphlient
       gruf
       istox_gruf
       listen (~> 3.0.5)
+      ougai
       paranoia (~> 2.2)
       vault (~> 0.1)
 
@@ -57,6 +59,7 @@ GEM
       tzinfo (~> 1.1)
     amq-protocol (2.3.0)
     arel (9.0.0)
+    awesome_print (1.8.0)
     aws-eventstream (1.0.3)
     aws-sigv4 (1.1.0)
       aws-eventstream (~> 1.0, >= 1.0.2)
@@ -66,7 +69,7 @@ GEM
     bullet (5.7.6)
       activesupport (>= 3.0.0)
       uniform_notifier (~> 1.11.0)
-    bunny (2.14.2)
+    bunny (2.14.3)
       amq-protocol (~> 2.3, >= 2.3.0)
     concurrent-ruby (1.1.4)
     crass (1.0.4)
@@ -81,29 +84,29 @@ GEM
       railties (>= 3.0.0)
     faker (1.7.3)
       i18n (~> 0.5)
-    faraday (0.15.4)
+    faraday (0.17.3)
       multipart-post (>= 1.2, < 3)
     faraday_middleware (0.13.1)
       faraday (>= 0.7.4, < 1.0)
-    ffi (1.11.1)
+    ffi (1.11.3)
     globalid (0.4.2)
       activesupport (>= 4.2.0)
-    google-protobuf (3.9.1-universal-darwin)
+    google-protobuf (3.11.2)
     googleapis-common-protos-types (1.0.4)
       google-protobuf (~> 3.0)
-    graphlient (0.3.6)
+    graphlient (0.3.7)
       faraday
       faraday_middleware
       graphql-client
-    graphql (1.9.12)
-    graphql-client (0.15.0)
+    graphql (1.9.17)
+    graphql-client (0.16.0)
       activesupport (>= 3.0)
       graphql (~> 1.8)
-    grpc (1.23.0-universal-darwin)
+    grpc (1.26.0)
       google-protobuf (~> 3.8)
       googleapis-common-protos-types (~> 1.0)
-    grpc-tools (1.23.0)
-    gruf (2.7.0)
+    grpc-tools (1.26.0)
+    gruf (2.7.1)
       activesupport (> 4)
       concurrent-ruby (> 1)
       grpc (~> 1.10)
@@ -136,6 +139,9 @@ GEM
     nio4r (2.3.1)
     nokogiri (1.10.1)
       mini_portile2 (~> 2.4.0)
+    oj (3.10.0)
+    ougai (1.8.2)
+      oj (~> 3.4)
     paranoia (2.4.2)
       activerecord (>= 4.0, < 6.1)
     rack (2.0.6)
@@ -167,7 +173,7 @@ GEM
       thor (>= 0.19.0, < 2.0)
     rake (10.5.0)
     rb-fsevent (0.10.3)
-    rb-inotify (0.10.0)
+    rb-inotify (0.10.1)
       ffi (~> 1.0)
     rspec (3.8.0)
       rspec-core (~> 3.8.0)
@@ -204,7 +210,7 @@ GEM
     tzinfo (1.2.5)
       thread_safe (~> 0.1)
     uniform_notifier (1.11.0)
-    vault (0.12.0)
+    vault (0.13.0)
       aws-sigv4
     websocket-driver (0.7.0)
       websocket-extensions (>= 0.1.0)

data/istox.gemspec

@@ -39,6 +39,8 @@ Gem::Specification.new do |spec|
   spec.add_dependency "graphlient"
   spec.add_dependency 'listen', '~> 3.0.5'
   spec.add_dependency "binding_of_caller"
+  spec.add_dependency "ougai"
+  spec.add_dependency "awesome_print"
   spec.add_development_dependency "bundler", "~> 1.16"
   spec.add_development_dependency "rake", "~> 10.0"
   spec.add_development_dependency "rspec", "~> 3.0"

data/lib/istox.rb

@@ -12,6 +12,7 @@ module Istox
     end
   end
 
+  require 'istox/helpers/logger'
   require 'istox/interfaces/chainhub/transaction'
   require 'istox/helpers/publisher'
   require 'istox/helpers/bunny_boot'

data/lib/istox/consumers/blockchain_status_handler.rb

@@ -1,3 +1,5 @@
+require 'istox/helpers/logger'
+
 module Istox
   class BlockchainStatusHandler
     class << self
@@ -14,7 +16,7 @@ module Istox
       def chainhub_response_event(transaction)
         receipt = ::Istox::BlockchainReceipt.where(txid: transaction.uuid).first
         if receipt.blank?
-          puts 'Transaction doesn\'t belong to this backend, skipping...'
+          log.info 'Transaction doesn\'t belong to this backend, skipping...'
           return
         end
 
@@ -26,7 +28,7 @@ module Istox
         resource = begin
           class_eval("::#{receipt.resource_name}", __FILE__, __LINE__).find(receipt.resource_id)
                    rescue NameError
-                     puts 'Class not found, skipping...'
+                     log.warn "Transaction bind to #{receipt.resource_name} with id #{receipt.resource_id} class not found, skipping..."
                      return
         end
 
@@ -49,36 +51,42 @@ module Istox
       private
 
       def publish_to_frontend(model, success, receipt, sid)
-        ::Istox::Publisher.publish(
-          'amq.topic',
-          model.class.name.gsub('::', '').underscore.to_s,
-          data: {
-            id: model.id,
-            extra: model.handle_extra
+        ::Istox::Publisher.manual_publish(
+          exchange: 'amq.topic',
+          exchange_type: 'direct',
+          routing_key: model.class.name.gsub('::', '').underscore.to_s,
+          message: {
+            data: {
+              id: model.id,
+              extra: model.handle_extra
+            }
           }
         )
 
-        ::Istox::Publisher.publish(
-          'amq.topic',
-          sid,
-          data: {
-            success: success,
-            purpose: receipt.activity,
-            message: receipt.message
+        ::Istox::Publisher.manual_publish(
+          exchange: 'amq.topic',
+          exchange_type: 'direct',
+          routing_key: sid,
+          message: {
+            data: {
+              success: success,
+              purpose: receipt.activity,
+              message: receipt.message
+            }
           }
         )
       rescue StandardError => e
-        Rails.logger.warning "Unable to publish to frontend for receipt #{receipt.inspect},
+        log.warn "Unable to publish to frontend for receipt #{receipt.inspect},
         but will silently ignore the error"
-        Rails.logger.warning e
+        log.warn e
       end
 
       def mark_resource_handled(receipt)
         receipt.update(resource_handled: true)
       rescue StandardError => e
-        Rails.logger.warning "Unable to update resource_handled for receipt #{receipt.inspect},
+        log.warn "Unable to update resource_handled for receipt #{receipt.inspect},
         but will silently ignore the error"
-        Rails.logger.warning e
+        log.warn e
       end
 
       def resource_handled?(receipt)

data/lib/istox/helpers/bunny_boot.rb

@@ -1,77 +1,124 @@
 require 'bunny'
+require 'istox/helpers/logger'
+
 module Istox
   class BunnyBoot
     class << self
+      # path to amqp.yml
       def initialize!(amqp_config_path)
-        @@amqp_config_path = amqp_config_path
-        ::Istox::Publisher.initialize!(amqp_config_path)
-        connect
-      end
+        @amqp_config_path = amqp_config_path
 
-      def start_subscribe
-        subscribe_to_queues
+        data
       end
 
-      def connect
-        connection = Bunny.new(data[:connect].symbolize_keys)
-        connection.start
-        $channel = connection.create_channel
-      end
+      # optionally can pass in consumer_key for single subscription / consumer_keys for multiple subcriptions
+      # consumer_key must be defined in amqp.yml
+      # if nothing pass in it will auto subscribe to all available consumers defined in amqp.yml queues key
+      def start_subscribe(consumer_key: nil, consumer_keys: nil)
+        subscribing_consumer_keys = consumer_keys.present? ? consumer_keys : []
+
+        if subscribing_consumer_keys.empty? && consumer_key.nil?
+          subscribing_consumer_keys = data['queues'].keys
+        elsif subscribing_consumer_keys.empty? && consumer_key.present?
+          subscribing_consumer_keys = [consumer_key]
+        end
 
-      protected
-
-      def subscribe_to_queues
-        data&.queues&.each do |data|
-          config = OpenStruct.new(data[1])
-          klass = Object.const_get('::' + "#{data[0]}_consumer".camelize)
-          exchange_type = if config.exchange.ends_with?('.fanout')
-                            :fanout
-                          else
-                            :direct
-                          end
-          options = {}
-          options[:routing_key] = config.name if exchange_type != :fanout
-          queue = $channel.queue(
-            config.name,
-            durable: config.durable
-          ).bind(exchange(config.exchange, exchange_type), options)
-
-          manual_ack = true
-          manual_ack = config.manual_ack unless config.manual_ack.nil?
-
-          queue.subscribe manual_ack: manual_ack do |delivery_info, metadata, payload|
-            Rails.logger.debug("Processing in consumer: #{klass}")
-            klass.new.process(
-              Hashie::Mash.new(JSON.parse(payload)),
-              metadata,
-              delivery_info
-            )
-            $channel.ack(delivery_info.delivery_tag) if manual_ack
-          rescue StandardError => e
-            Rails.logger.error(e)
-            $channel.nack(delivery_info.delivery_tag, false, true)
-          end
+        subscribing_consumer_keys.each do |key|
+          subscribe_to_consumer(key.to_s)
         end
       end
 
-      private
+      #############################################################################################
+      ## INTERNAL METHODS START
+      ############################################################################################
 
-      def read_file
-        YAML.safe_load(ERB.new(File.read(@@amqp_config_path)).result)
+      def channel
+        return @channel if @channel.present?
+
+        @channel = connection.create_channel(nil, data['channel_pool_size'] || 1)
+        @channel.prefetch(data['channel_prefetch'] || 1)
+
+        @channel
       end
 
-      def data
-        OpenStruct.new(read_file)
+      # calling this method will spawn another channel, please use the normal channel method if you like to use the same channel
+      def create_new_channel(channel_config)
+        channel = connection.create_channel(nil, channel_config['channel_pool_size'] || data['channel_pool_size'] || 1)
+        channel.prefetch(channel_config['channel_prefetch'] || data['channel_prefetch'] || 1)
+
+        channel
       end
 
-      def exchange(host_exchange, exchange_type)
-        case exchange_type
-        when :fanout
-          $channel.fanout(host_exchange, durable: true)
+      def subscribe_to_consumer(consumer_key)
+        queue_config = data['queues'][consumer_key]
+
+        raise "Unable to find consumer with key #{consumer_key}, have you forgotten to define it in amqp.yml?" if queue_config.nil?
+
+        klass = Object.const_get(
+          '::' + (queue_config['ruby_class'].nil? ? "#{consumer_key.to_s.underscore}_consumer" : queue_config['ruby_class']).camelize
+        )
+
+        exchange = nil
+        exchange_config = exchange_config!(consumer_key)
+
+        active_channel = queue_config['channel'].present? ? create_new_channel(queue_config['channel']) : channel
+
+        queue_options = {}
+
+        exchange_durable = exchange_config['durable'].nil? ? true : exchange_config['durable']
+
+        case exchange_config['type']
+        when 'fanout'
+          exchange = active_channel.fanout(queue_config['exchange'], durable: exchange_durable)
+        when 'direct'
+          exchange = active_channel.direct(queue_config['exchange'], durable: exchange_durable)
+          queue_options[:routing_key] = queue_config['queue_name']
         else
-          $channel.direct(host_exchange, durable: true)
+          raise "Exchange type #{exchange_config.type} is not valid/supported."
+        end
+
+        queue = active_channel.queue(queue_config['queue_name'],
+                                     durable: queue_config['durable'].nil? ? exchange_durable : queue_config['durable']).bind(exchange, queue_options)
+
+        manual_ack = queue_config['manual_ack'].nil? ? true : queue_config['manual_ack']
+
+        queue.subscribe manual_ack: manual_ack do |delivery_info, metadata, payload|
+          processing_paylod = JSON.parse(payload)
+          log.info "Processing in consumer: #{klass}, paylod: #{processing_paylod.inspect}"
+
+          klass.new.process(::Istox::CommonHelper.to_open_struct(processing_paylod, metadata, delivery_info))
+          active_channel.ack(delivery_info.delivery_tag) if manual_ack
+        rescue StandardError => e
+          log.error(e)
+          active_channel.nack(delivery_info.delivery_tag, false, true) if manual_ack
         end
       end
+
+      def data
+        raise 'Unable to find amqp configuration file, have you forgotten to initialise Istox::BunnyBoot?' if @amqp_config_path.empty?
+
+        @data ||= JSON.parse(YAML.load_file(@amqp_config_path).to_json)
+      end
+
+      def connection
+        return @connection if @connection.present?
+
+        raise 'RabbitMQ connection configuration not found, have you forgotten to define connect key in amqp.yml?' if data['connect'].nil?
+
+        @connection = Bunny.new(data['connect'].symbolize_keys)
+        @connection.start
+
+        @connection
+      end
+
+      def exchange_config!(consumer_key)
+        exchange_name = data['queues'][consumer_key]['exchange']
+        exchange_config = data['exchanges'][exchange_name]
+
+        raise "Exchange #{exchange_name} config not found, have you forgotten to define the exchange in amqp.yml?" if exchange_config.nil?
+
+        exchange_config
+      end
     end
   end
 end

data/lib/istox/helpers/graphql_client.rb

@@ -1,4 +1,5 @@
 require 'graphlient'
+require 'istox/helpers/logger'
 
 module Istox
   class GraphqlClient
@@ -13,12 +14,12 @@ module Istox
 
         raise "Please make sure you have initialised graphql cient for host #{host_type}" unless client
 
-        Rails.logger.debug "Querying Graphql host: #{host_type}, query: #{query}, variables: #{variables.inspect}"
+        log.info "Querying Graphql host: #{host_type}, query: #{query}, variables: #{variables.inspect}"
         result = client.query(query, variables)
 
         return_values = ::Istox::CommonHelper.to_open_struct(result&.data)
 
-        Rails.logger.debug "Graphql result: #{return_values.inspect}"
+        log.info "Graphql result: #{return_values.inspect}"
 
         return_values
       end

data/lib/istox/helpers/grpc_client.rb

@@ -1,3 +1,5 @@
+require 'istox/helpers/logger'
+
 module Istox
   class GrpcClient
     class << self
@@ -18,14 +20,14 @@ module Istox
       private
 
       def execute(host_type, service, method, **keyword_args)
-        Rails.logger.debug "Calling method in grpc method: #{method.inspect}, params: #{keyword_args.inspect}"
+        log.info "Calling method in grpc method: #{method.inspect}, params: #{keyword_args.inspect}"
         t1 = Time.now
 
         result = get_host(host_type, service).call(method, keyword_args)
-        Rails.logger.debug "Time taken for grpc execution: #{Time.now - t1} seconds"
+        log.info "Time taken for grpc execution: #{Time.now - t1} seconds"
 
         return_values = ::Istox::CommonHelper.to_open_struct(result.message)
-        Rails.logger.debug "Result: #{return_values}"
+        log.info "Result: #{return_values}"
 
         return_values
       end
@@ -49,10 +51,10 @@ module Istox
         host_url = @@hosts[host_type]
         raise StandardError, 'Unable to find host, have you forgotten to add host to grpc client?' unless host_url
 
-        Rails.logger.debug 'Reinitiating to grpc host at ' + host_url
+        log.info 'Reinitiating to grpc host at ' + host_url
         t1 = Time.now
         @@services[get_key(host_type, service)] = ::Gruf::Client.new(service: service, options: { hostname: host_url })
-        Rails.logger.debug "Time taken for reinitiating grpc host: #{Time.now - t1} seconds"
+        log.info "Time taken for reinitiating grpc host: #{Time.now - t1} seconds"
       end
 
       def get_key(host_type, service)

data/lib/istox/helpers/gruf_listener_hook.rb

@@ -1,21 +1,21 @@
 module Istox
-    class GrufListenerHook < ::Gruf::Hooks::Base
-        def before_server_start(server:)
-            # do my thing before the server starts  
+  class GrufListenerHook < ::Gruf::Hooks::Base
+    def before_server_start(server:)
+      # do my thing before the server starts
 
-            @listener = ::Listen.to('app/rpc') do |modified, added, removed|
-                puts "modified absolute path: #{modified}"
-                puts "added absolute path: #{added}"
-                puts "removed absolute path: #{removed}"
-                
-                server.init_restart
-            end
-            @listener.start # not blocking
-        end
-        
-        def after_server_stop(server:)
-            # do my thing after the server stops
-            @listener.stop
-        end
+      @listener = ::Listen.to('app/rpc') do |modified, added, removed|
+        puts "modified absolute path: #{modified}"
+        puts "added absolute path: #{added}"
+        puts "removed absolute path: #{removed}"
+
+        server.init_restart
+      end
+      @listener.start # not blocking
+    end
+
+    def after_server_stop(server:)
+      # do my thing after the server stops
+      @listener.stop
     end
-end
+  end
+end

data/lib/istox/helpers/logger.rb

@@ -0,0 +1,18 @@
+require 'ougai'
+
+module Istox
+  class Logger
+    def self.logger
+      return @logger if @logger.present?
+
+      @logger = ::Ougai::Logger.new(STDOUT)
+      @logger.formatter = ::Ougai::Formatters::Readable.new unless Rails.env.production?
+
+      @logger
+    end
+  end
+end
+
+def log
+  ::Istox::Logger.logger
+end

data/lib/istox/helpers/message_service.rb

@@ -1,46 +1,51 @@
 module Istox
-    class MessageService
-        class << self
-            def send_sms(number, subject, content)
-            publish('sms',                 data: {
-                        to: number,
-                        subject: subject,
-                        body: content
-                    },
-                                            type: __method__.to_s.upcase!)
-            end
+  class MessageService
+    class << self
+      def send_sms(number, subject, content)
+        publish('sms', data: {
+                  to: number,
+                  subject: subject,
+                  body: content
+                },
+                       type: __method__.to_s.upcase!)
+      end
 
-            def send_sms_template(number, template)
-            publish('sms',                 data: {
-                        to: number,
-                        template: template
-                    },
-                                            type: __method__.to_s.upcase!)
-            end
+      def send_sms_template(number, template)
+        publish('sms', data: {
+                  to: number,
+                  template: template
+                },
+                       type: __method__.to_s.upcase!)
+      end
 
-            def send_email(email, subject, content)
-            publish('email',                 data: {
-                        to: email,
-                        subject: subject,
-                        type: 'text/plain',
-                        body: content
-                    },
-                                            type: __method__.to_s.upcase!)
-            end
+      def send_email(email, subject, content)
+        publish('email', data: {
+                  to: email,
+                  subject: subject,
+                  type: 'text/plain',
+                  body: content
+                },
+                         type: __method__.to_s.upcase!)
+      end
 
-            def send_email_template(email, template)
-            publish('email',                 data: {
-                        to: email,
-                        template: template
-                    },
-                                            type: __method__.to_s.upcase!)
-            end
+      def send_email_template(email, template)
+        publish('email', data: {
+                  to: email,
+                  template: template
+                },
+                         type: __method__.to_s.upcase!)
+      end
 
-            private
+      private
 
-            def publish(type, payload)
-                ::Istox::Publisher.publish('message', "#{type}.message", payload)
-            end
-        end
+      def publish(type, payload)
+        ::Istox::Publisher.manual_publish(
+          exchange: 'message',
+          exchange_type: 'direct',
+          routing_key: "#{type}.message",
+          message: payload
+        )
+      end
     end
-end
+  end
+end

data/lib/istox/helpers/publisher.rb

@@ -1,44 +1,102 @@
+require 'istox/helpers/logger'
+
 module Istox
+  # Publisher is relying on BunnyBoot to publish message, please make sure BunnyBoot is initalised properly first during runtime.
   class Publisher
     class << self
-      def initialize!(amqp_config_path)
-        @@amqp_config_path = amqp_config_path
-      end
-
-      def publish(exchange, routing_key, message, persistent: true)
-        Rails.logger.debug 'Before publish retrieve data..'
-        durable = data.fetch('publish')[exchange]['durable']
-        Rails.logger.debug "Publishing message: #{message} with exchange: #{exchange} routing key: #{routing_key}"
-        enqueue do
-          $channel
-            .direct(exchange, durable: durable)
-            .publish(message.to_json, routing_key: routing_key, durable: durable, persistent: persistent)
+      # routing_key can be nil if publishing to fanout exchange
+      # exchange can be nil if routing_key is unique in "publish" section, this function will be able to identify which exchange it is using
+      # else both routing_key and exchange should be provided
+      # check http://reference.rubybunny.info/Bunny/Exchange.html#publish-instance_method for available publish options
+      def publish(exchange: nil, routing_key: nil, message:, options: {})
+        raise 'Exchange and routing key cannot be nil at the same time.' if exchange.nil? && routing_key.nil?
+
+        queue_publish_config = nil
+        if routing_key.present?
+          exchange = search_exchange_of_routing_key!(routing_key) if exchange.nil?
+
+          puts "EXCYANGE IS #{exchange}"
+
+          queue_publish_config = queue_publish_config!(routing_key: routing_key, exchange: exchange)
         end
+
+        exchange_config = exchange_config!(exchange)
+        exchange_durable = exchange_config['durable'].nil? ? true : exchange_config['durable']
+
+        queue_durable = queue_publish_config.present? ? queue_publish_config['durable'] : nil
+        queue_durable = exchange_config['durable'].nil? ? true : exchange_config['durable'] if queue_durable.nil?
+
+        options[:durable] = queue_durable
+        options[:persistent] = options[:persistent].nil? ? true : options[:persistent]
+
+        send_message(exchange: exchange, routing_key: routing_key, message: message,
+                     exchange_type: exchange_config['type'], exchange_durable: exchange_durable, options: options)
       end
 
-      def publish_to_fanout(exchange, message, persistent: true)
-        Rails.logger.debug 'Before publish retrieve data..'
-        durable = data.fetch('publish')[exchange]['durable']
-        enqueue do
-          $channel
-            .fanout(exchange, durable: durable)
-            .publish(message.to_json, durable: durable, persistent: persistent)
-        end
+      # publish without defining exchange and queue in amqp.yml
+      def manual_publish(exchange:, routing_key: nil, message:, exchange_type: 'direct', exchange_durable: true, options: {})
+        options[:durable] = options[:durable].nil? ? true : options[:durable]
+        options[:persistent] = options[:persistent].nil? ? true : options[:persistent]
+
+        send_message(exchange: exchange, routing_key: routing_key, message: message,
+                     exchange_type: exchange_type, exchange_durable: exchange_durable, options: options)
       end
 
       private
 
-      def enqueue
-        initialize_channel! unless !!$channel
-        yield
-      end
+      def send_message(exchange:, routing_key: nil, message:, exchange_type:, exchange_durable:, options: {})
+        log.info "Publishing RABBITMQ message to exchange: #{exchange}, routing key: #{routing_key}, data: #{message}, options: #{options.inspect}"
 
-      def initialize_channel!
-        ::Istox::BunnyBoot.connect
+        case exchange_type
+        when 'direct'
+          raise 'Routing key cannot be empty for exchange type direct' if routing_key.blank?
+
+          options[:routing_key] = routing_key
+          channel.direct(exchange, durable: exchange_durable).publish(message.to_json, options)
+        when 'fanout'
+          channel.fanout(exchange, durable: exchange_durable).publish(message.to_json, options)
+        else
+          raise "Exchange type #{exchange_config.type} is not valid/supported."
+        end
       end
 
       def data
-        YAML.safe_load(ERB.new(File.read(@@amqp_config_path)).result)
+        ::Istox::BunnyBoot.data
+      end
+
+      def channel
+        ::Istox::BunnyBoot.channel
+      end
+
+      def queue_publish_config!(routing_key:, exchange:)
+        queue_publish_config = data['publish'][exchange][routing_key]
+        unless data['publish'][exchange].key?(routing_key)
+          raise "Cannot find publish queue config for exchange #{exchange} and routing key #{routing_key},
+          have you forgotten to define it in exchange section in amqp.yml"
+        end
+
+        queue_publish_config
+      end
+
+      def search_exchange_of_routing_key!(routing_key)
+        data['publish'].each do |k, v|
+          v.each do |k2, _v2|
+            return k if k2 == routing_key
+          end
+        end
+
+        raise "Cannot find exchange of routing key #{routing_key}, have you forgotten to define it in exchange section in amqp.yml"
+      end
+
+      def exchange_config!(exchange_name)
+        exchange_config = data['exchanges'][exchange_name]
+
+        if exchange_config.nil?
+          raise "Exchange config of #{exchange_name} cannot be found, have you
+          forgotten to define it in exchange section in amqp.yml"
+        end
+
+        exchange_config
       end
     end
   end

data/lib/istox/helpers/vault.rb

@@ -1,3 +1,4 @@
+require 'istox/helpers/logger'
 require 'vault'
 
 module Istox
@@ -5,7 +6,7 @@ module Istox
     module TOTP
       class << self
         def create(sid, host)
-          Rails.logger.debug { "Generate vault TOTP for key #{totp_key(sid).inspect}" }
+          log.debug { "Generate vault TOTP for key #{totp_key(sid).inspect}" }
 
           write_data(totp_key(sid),
                      generate: true,
@@ -16,17 +17,17 @@ module Istox
         end
 
         def generate_code(sid)
-          Rails.logger.debug { "Generate 6-digit OTP code: key #{totp_code_key(sid)}" }
+          log.debug { "Generate 6-digit OTP code: key #{totp_code_key(sid)}" }
           read_data(totp_code_key(sid)).data[:code]
         end
 
         def validate?(sid, code)
-          Rails.logger.debug { "Validate TOTP code: key #{totp_code_key(sid)}, code: #{code}" }
+          log.debug { "Validate TOTP code: key #{totp_code_key(sid)}, code: #{code}" }
           result = write_data(totp_code_key(sid), code: code).data[:valid]
 
           unless result
             code = read_data(totp_code_key(sid)).data[:code]
-            Rails.logger.debug { "Code is not valid, it should be #{code}" }
+            log.debug { "Code is not valid, it should be #{code}" }
           end
 
           result
@@ -67,7 +68,7 @@ module Istox
             config.ssl_verify = false
             config.timeout = 60
           else
-            Rails.logger.info 'Vault token not found, OTP will not be able to use'
+            log.info 'Vault token not found, OTP will not be able to use'
           end
         end
       end
@@ -79,9 +80,6 @@ module Istox
 
       def validate_otp(sid, otp)
         ::Istox::Vault::TOTP.validate?(sid, otp)
-      rescue StandardError => e
-        raise e
-        false
       end
     end
   end

data/lib/istox/version.rb

@@ -1,3 +1,3 @@
 module Istox
-  VERSION = '0.1.103'.freeze
+  VERSION = '0.1.104'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: istox
 version: !ruby/object:Gem::Version
-  version: 0.1.103
+  version: 0.1.104
 platform: ruby
 authors:
 - Siong Leng
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2019-12-02 00:00:00.000000000 Z
+date: 2020-01-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bunny
@@ -122,6 +122,34 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: ougai
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: awesome_print
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -297,6 +325,7 @@ files:
 - lib/istox/helpers/graphql_client.rb
 - lib/istox/helpers/grpc_client.rb
 - lib/istox/helpers/gruf_listener_hook.rb
+- lib/istox/helpers/logger.rb
 - lib/istox/helpers/message_service.rb
 - lib/istox/helpers/my_open_struct.rb
 - lib/istox/helpers/order_book_price_time.rb