lib/gratan/driver.rb in gratan-0.3.1 vs lib/gratan/driver.rb in gratan-0.3.2

@@ -213,9 +213,15 @@
     if @options[:override_sql_mode]
       query('SET SQL_MODE = ""')
     end
   end
 
+  def set_wait_timeout
+    if @options[:wait_timeout]
+      query("SET @@wait_timeout = #{@options[:wait_timeout]}")
+    end
+  end
+
   private
 
   def query(sql)
     log(:debug, sql, :dry_run => false)
     @client.query(sql)