with_query to dry up Queryer::Client

jbr · Feb 16, 2010 · 08e7b88 · 08e7b88
1 parent 789ebba
commit 08e7b88
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/lib/queryer/client.rb b/lib/queryer/client.rb
@@ -8,18 +8,19 @@ def initialize(pool, queryer)
       @queryer = queryer
     end
 
-    def select(query_string)
+    def with_query(query_string)
       with_connection(query_string) do |env|
         env['nk.query'] = @queryer.build_query(env)
-        @queryer.select(env)
+        yield env
       end
     end
+
+    def select(query_string)
+      with_query(query_string) { |env| @queryer.select(env) }
+    end
 
     def execute(query_string)
-      with_connection(query_string) do |env|
-        env['nk.query'] = @queryer.build_query(env)
-        @queryer.execute(env)
-      end
+      with_query(query_string) { |env| @queryer.execute(env) }
     end
 
     def transaction