Parse query_info to a hash

Author: sodabrew

ext/mysql2/client.c

@@ -1143,7 +1143,7 @@ void init_mysql2_client() {
   rb_define_method(cMysql2Client, "store_result", rb_mysql_client_store_result, 0);
   rb_define_method(cMysql2Client, "reconnect=", set_reconnect, 1);
   rb_define_method(cMysql2Client, "warning_count", rb_mysql_client_warning_count, 0);
-  rb_define_method(cMysql2Client, "query_info", rb_mysql_info, 0);
+  rb_define_method(cMysql2Client, "query_info_string", rb_mysql_info, 0);
 #ifdef HAVE_RUBY_ENCODING_H
   rb_define_method(cMysql2Client, "encoding", rb_mysql_client_encoding, 0);
 #endif

lib/mysql2/client.rb

@@ -61,6 +61,14 @@ def self.default_query_options
       @@default_query_options
     end
 
+    def query_info
+      info = query_info_string
+      return {} unless info
+      info_hash = {}
+      info.split.each_slice(2) { |s| info_hash[s[0].downcase.delete(':').to_sym] = s[1].to_i }
+      info_hash
+    end
+
     private
       def self.local_offset
         ::Time.local(2010).utc_offset.to_r / 86400

spec/mysql2/client_spec.rb

@@ -140,7 +140,8 @@ def connect *args
         @client.query('select 1')
       end
       it "should 0" do
-        @client.query_info.should be_nil
+        @client.query_info.should be_empty
+        @client.query_info_string.should be_nil
       end
     end
     context "when has some info" do
@@ -159,7 +160,8 @@ def connect *args
         @client.query("INSERT INTO infoTest (blah) VALUES (1234),(4535)")
       end
       it "should retrieve it" do
-        @client.query_info.should eq 'Records: 2  Duplicates: 0  Warnings: 0'
+        @client.query_info.should == {:records => 2, :duplicates => 0, :warnings => 0}
+        @client.query_info_string.should eq 'Records: 2  Duplicates: 0  Warnings: 0'
       end
     end
   end