Optimize Model.as_set calls to avoid queries in the static_cache plugin

Author: jeremyevans

CHANGELOG

@@ -1,5 +1,7 @@
 === master
 
+* Optimize Model.as_set calls to avoid queries in the static_cache plugin (jeremyevans)
+
 * Support filtering by associations with sets in the pg_array_associations plugin (jeremyevans)
 
 * Support sets in the to_dot extension (jeremyevans)

lib/sequel/plugins/static_cache.rb

@@ -176,6 +176,19 @@ def as_hash(key_column = nil, value_column = nil, opts = OPTS)
           h
         end
 
+        # Use the cache instead of a query to get the results.
+        def as_set(column)
+          set = Set.new
+
+          if column.is_a?(Array)
+            @all.each{|r| set.add(r.values.values_at(*column))}
+          else
+            @all.each{|r| set.add(r[column])}
+          end
+
+          set
+        end
+
         # Alias of as_hash for backwards compatibility.
         def to_hash(*a)
           as_hash(*a)

spec/extensions/static_cache_spec.rb

@@ -132,6 +132,13 @@
       proc{@c.map(:id){}}.must_raise(Sequel::Error)
     end
 
+    it "should have as_set not send a query" do
+      @c.as_set(:id).must_equal Set[1, 2]
+      @db.sqls.must_equal []
+      @c.as_set([:id,:id]).must_equal Set[[1,1], [2,2]]
+      @db.sqls.must_equal []
+    end
+
     it "should have other enumerable methods work without sending a query" do
       a = @c.sort_by{|o| o.id}
       a.first.must_equal @c1