querier 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b9b2cf6f8f3a3f4aa087c68f85753ddcca380e076a165d0abef3fe750815630a
-  data.tar.gz: b8825c18fae966fbcdc54dd33a799c108efa7c5cebb4c46336c53093cb03bd1d
+  metadata.gz: b665dcde0c37b483e764ddcf1d30221ba9c4b34f478533e6fb911aaf11a81996
+  data.tar.gz: 4bd3bb6e6e3cc430b257ac9b408629440b8249ad642c1db89a6f9ece21d671f3
 SHA512:
-  metadata.gz: 84affb4d1704196d94de6dfbefb66795dc68636ddbfdf0fd6131a7b925313954642032d052d756124bff9e6a04a38b3a0a0917fc50d923c7d85d0cd9cd571a37
-  data.tar.gz: e4ff00606711107491de8ea7f9e7066786136859da62f2fdc3ff1b1f27e7a8ef4e85d7918c1fc9fa8962c840ecf71961f9c8464748a5ad1f39ba862f063d3cf7
+  metadata.gz: 5e0fc946d4e39bc164ffeda2bd762c4e3bddc4f769363d9590e4dc08a6a7379b3f3e27f5516b93baaea2a8c8216ac2808946096a0f48a28b847246d8a73d1da7
+  data.tar.gz: 3af81af98208e77a0f9360de68a5d15cd45e153613aaa5e1654157bcf4ea4ccf77977c80d1e30b22259db96eabff5d13bc6b66a843c4cbbcb210ad114ac905c6

data/lib/querier.rb

@@ -1,83 +1,91 @@
-class Querier
-  PARAM_NAME_INDEX = 0
-  PARAM_VALUE_INDEX = 1
-
-  attr_reader :query_execution_count, :query_template, :query_params
-
-  def initialize(**template_query_params)
-    @query_execution_count = 0
-    @query_params = template_query_params.dup
-  end
-
-  def execute
-    @query_execution_count += 1
-    @execution_cached_result = ActiveRecord::Base.connection.select_all(fill_query_params(query_template: @query_template,
-                                                                                          query_params: @query_params)).map(&:symbolize_keys!)
-  end
-
-  def cached_result(format: :hash)
-    raise 'query not executed yet' if @query_execution_count.eql?(0)
-
-    case format
-    when :hash
-      @execution_cached_result
-    when :open_struct
-      hash_to_open_struct(dataset: @execution_cached_result)
-    else
-      raise 'invalid value type'
-    end
-  end
-
-  def structured_results
-    hash_to_open_struct(dataset: execute)
-  end
-
-  def to_sql
-    fill_query_params(query_template: @query_template, query_params: @query_params)
-  end
-
-  def to_file
-    file_name = "querier #{Time.now.strftime '[%d-%m-%Y]-[%Hh %Mm %Ss]'}.sql"
-    File.open("tmp/#{file_name}", 'w') { |f| f << to_sql }
-  end
-
-  def field_group_and_count(field_name:, sort_element_index: nil, reverse_sort: true)
-    count_result = cached_results(format: :open_struct).group_by(&field_name).map { |k, v| [k, v.count] }
-
-    unless sort_element_index.nil?
-      count_result = count_result.sort_by { |el| el[sort_element_index] }
-      count_result.reverse! if reverse_sort.eql? true
-    end
-
-    count_result
-  end
-
-  private
-
-  def hash_to_open_struct(dataset:)
-    dataset.map { |record| OpenStruct.new(record.symbolize_keys!) }
-  end
-
-  def get_param_value(raw_query_param:, quotefy_param: true)
-    # where's String#quote when we need it?
-    raw_query_param.instance_of?(String) && quotefy_param ? "'#{raw_query_param.to_s}'" : raw_query_param.to_s
-  end
-
-  def fill_query_params(query_template:, query_params:)
-    query = query_template.dup
-
-    query_params.each do |query_param|
-      query_param_name = query_param[PARAM_NAME_INDEX].to_s
-
-      query.gsub!(/\${#{query_param_name}}/,
-                  get_param_value(raw_query_param: query_param[PARAM_VALUE_INDEX], 
-                                  quotefy_param: true))
-
-      query.gsub!(/\${#{query_param_name}\/no_quote}/,
-                  get_param_value(raw_query_param: query_param[PARAM_VALUE_INDEX],
-                                  quotefy_param: false))
-    end
-
-    query
-  end
-end
+class Querier
+  PARAM_NAME_INDEX = 0
+  PARAM_VALUE_INDEX = 1
+
+  @active_record_class = ActiveRecord::Base
+  # based on rubocop's tips at: https://www.rubydoc.info/gems/rubocop/RuboCop/Cop/Style/ClassVars
+  # solution here: https://www.ruby-lang.org/en/documentation/faq/8/
+  class << self
+    attr_accessor :active_record_class
+  end
+
+  attr_reader :query_execution_count, :query_template, :query_params
+
+  def initialize(**template_query_params)
+    @active_record_class = self.class.active_record_class || self.class.superclass.active_record_class
+    @query_execution_count = 0
+    @query_params = template_query_params.dup
+  end
+
+  def execute
+    @query_execution_count += 1
+    @execution_cached_result = @active_record_class.connection.select_all(fill_query_params(query_template: @query_template,
+                                                                                          query_params: @query_params)).map(&:symbolize_keys!)
+  end
+
+  def cached_result(format: :hash)
+    raise 'query not executed yet' if @query_execution_count.eql?(0)
+
+    case format
+    when :hash
+      @execution_cached_result
+    when :open_struct
+      hash_to_open_struct(dataset: @execution_cached_result)
+    else
+      raise 'invalid value type'
+    end
+  end
+
+  def structured_results
+    hash_to_open_struct(dataset: execute)
+  end
+
+  def to_sql
+    fill_query_params(query_template: @query_template, query_params: @query_params)
+  end
+
+  def to_file
+    file_name = "querier #{Time.now.strftime '[%d-%m-%Y]-[%Hh %Mm %Ss]'}.sql"
+    File.open("tmp/#{file_name}", 'w') { |f| f << to_sql }
+  end
+
+  def field_group_and_count(field_name:, sort_element_index: nil, reverse_sort: true)
+    count_result = cached_results(format: :open_struct).group_by(&field_name).map { |k, v| [k, v.count] }
+
+    unless sort_element_index.nil?
+      count_result = count_result.sort_by { |el| el[sort_element_index] }
+      count_result.reverse! if reverse_sort.eql? true
+    end
+
+    count_result
+  end
+
+  private
+
+  def hash_to_open_struct(dataset:)
+    dataset.map { |record| OpenStruct.new(record.symbolize_keys!) }
+  end
+
+  def get_param_value(raw_query_param:, quotefy_param: true)
+    # where's String#quote when we need it?
+    raw_query_param.instance_of?(String) && quotefy_param ? "'#{raw_query_param.to_s}'" : raw_query_param.to_s
+  end
+
+  def fill_query_params(query_template:, query_params:)
+    query = query_template.dup
+
+    query_params.each do |query_param|
+      query_param_name = query_param[PARAM_NAME_INDEX].to_s
+
+      query.gsub!(/\${#{query_param_name}}/,
+                  get_param_value(raw_query_param: query_param[PARAM_VALUE_INDEX], 
+                                  quotefy_param: true))
+
+      query.gsub!(/\${#{query_param_name}\/no_quote}/,
+                  get_param_value(raw_query_param: query_param[PARAM_VALUE_INDEX],
+                                  quotefy_param: false))
+    end
+
+    query
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: querier
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Gedean Dias
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-12-28 00:00:00.000000000 Z
+date: 2021-04-05 00:00:00.000000000 Z
 dependencies: []
 description: Active Record queries with variable number of params
 email: gedean.dias@gmail.com
@@ -37,7 +37,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.3
+rubygems_version: 3.2.15
 signing_key:
 specification_version: 4
 summary: Active Record Querier