relation_to_struct 1.1.0 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 13c9453de8bfec35a498a7819b901efe8d380c11
-  data.tar.gz: 54fe4e8053127c1dc35a17df5dd4e11933c299be
+  metadata.gz: 20d99ebc500ab5e07fd4b78fcb158790fa26b475
+  data.tar.gz: 5af91e2d780a6e0ec03383c3c29599674707f0a5
 SHA512:
-  metadata.gz: 9062091e453d07f865fcd1a5fb9441bf36ae3e6ff488939d66ffdfff43d6ad7892a087d074c6b07c0796f60b80b1dd49e407817810ac4d1720609db47f3601dd
-  data.tar.gz: de7d53e292cb15de12b73f11a461ace4b23cfd7ff35bc678bbca7eb719496c880c848d48082ba5ae0b25d6d98f693cf21737458bc6c3d5359e55bd5d13e5a093
+  metadata.gz: 732739b20458d9223cb161bdc4c3c9928b28a2b1c70536aca9371ddf642264a83b6f78ea1dd8eb78862d22069672a78b4cc12930d9cfbb97f448aa506ecdc759
+  data.tar.gz: 258ea4297e9b7f2538d54a727b9b4eeee01df5f8dfae0041c5ee8ebf2ecc0ba1753866aa0d127e9699aedfc2af95005a74d76b35e97995899e3450b9d444b4c4

data/lib/relation_to_struct/active_record_base_extension.rb

@@ -43,6 +43,20 @@ module RelationToStruct::ActiveRecordBaseExtension
         raise ArgumentError, 'Expected only a single result to be returned'
       end
     end
+
+    def tuple_from_sql(sql, binds=[])
+      result = connection.select_all(sanitize_sql(sql, nil), "Value SQL Load", binds)
+      values = result.cast_values()
+
+      case values.size
+      when 0
+        nil
+      when 1
+        result.columns.size == 1 ? values : values[0]
+      else
+        raise ArgumentError, 'Expected only a single result to be returned'
+      end
+    end
   end
 end
 

data/lib/relation_to_struct/version.rb

@@ -1,3 +1,3 @@
 module RelationToStruct
-  VERSION = "1.1.0"
+  VERSION = "1.2.0"
 end

data/spec/active_record_base_spec.rb

@@ -24,7 +24,7 @@ describe ActiveRecord::Base do
       expect(ActiveRecord::Base.value_from_sql(sql)).to eq(23)
     end
 
-    it 'raises an error when multiple columns are selected' do
+    it 'raises an error when multiple rows are selected' do
       expect do
         sql = "SELECT * FROM (VALUES (1), (2)) t"
         ActiveRecord::Base.value_from_sql(sql)
@@ -56,6 +56,64 @@ describe ActiveRecord::Base do
     end
   end
 
+  describe "#tuple_from_sql" do
+    it 'allows selecting one value with SQL directly' do
+      sql = "SELECT 1"
+      expect(ActiveRecord::Base.tuple_from_sql(sql)).to eq([1])
+    end
+
+    it 'allows selecting multiple values with SQL directly' do
+      sql = "SELECT 1, 23"
+      expect(ActiveRecord::Base.tuple_from_sql(sql)).to eq([1, 23])
+    end
+
+    it 'raises an error when multiple rows are selected' do
+      expect do
+        sql = "SELECT * FROM (VALUES (1, 3), (2, 4)) t"
+        ActiveRecord::Base.tuple_from_sql(sql)
+      end.to raise_error(ArgumentError, 'Expected only a single result to be returned')
+    end
+
+    it 'supports binds' do
+      sql = ["SELECT ?, ?", 5, 6]
+      expect(ActiveRecord::Base.tuple_from_sql(sql)).to eq([5, 6])
+    end
+
+    it 'supports a single array' do
+      if active_record_supports_arrays?
+        Economist.create!(name: 'F.A. Hayek')
+        Economist.create!(name: 'Ludwig von Mises')
+
+        result = ActiveRecord::Base.tuple_from_sql(<<-SQL)
+          SELECT ARRAY_AGG(name ORDER BY id) AS names
+          FROM economists
+        SQL
+        expected_names = ['F.A. Hayek', 'Ludwig von Mises']
+        expect(result).to eq([expected_names])
+      else
+        skip "DB selection doesn't support ARRAY[]"
+      end
+    end
+
+    it 'supports multiple arrays' do
+      if active_record_supports_arrays?
+        Economist.create!(name: 'F.A. Hayek')
+        Economist.create!(name: 'Ludwig von Mises')
+
+        result = ActiveRecord::Base.tuple_from_sql(<<-SQL)
+          SELECT
+            ARRAY_AGG(name ORDER BY id) AS names,
+            ARRAY_AGG(CHAR_LENGTH(name) ORDER BY id) AS lengths
+          FROM economists
+        SQL
+        expected_names = ['F.A. Hayek', 'Ludwig von Mises']
+        expect(result).to eq([expected_names, expected_names.map(&:size)])
+      else
+        skip "DB selection doesn't support ARRAY[]"
+      end
+    end
+  end
+
   describe "#structs_from_sql" do
     it 'ActiveRecord::Base should respond to :structs_from_sql' do
       expect(ActiveRecord::Base.respond_to?(:structs_from_sql)).to eq(true)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: relation_to_struct
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 1.2.0
 platform: ruby
 authors:
 - James Coleman
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-08-19 00:00:00.000000000 Z
+date: 2016-10-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: appraisal