Support mysql2 in SQL queries.

filiptepper · May 1, 2011 · f74d056 · f74d056
1 parent 6dcd889
commit f74d056
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 7 deletions.
diff --git a/lib/rack/bug/views/panels/execute_sql.html.erb b/lib/rack/bug/views/panels/execute_sql.html.erb
@@ -13,8 +13,14 @@
 <table class="sortable">
   <thead>
     <tr>
-      <% result.fetch_fields.each do |field| %>
-        <th><%= field.name.upcase %></th>
+      <% if defined?(Mysql2) %>
+        <% result.fields.each do |field| %>
+          <th><%= field.upcase %></th>
+        <% end %>
+      <% else %>
+        <% result.fetch_fields.each do |field| %>
+          <th><%= field.name.upcase %></th>
+        <% end %>
       <% end %>
     </tr>
   </thead>

diff --git a/lib/rack/bug/views/panels/explain_sql.html.erb b/lib/rack/bug/views/panels/explain_sql.html.erb
@@ -13,10 +13,16 @@
 <table class="sortable">
   <thead>
     <tr>
-      <% result.fetch_fields.each do |field| %>
-        <th><%= field.name.upcase %></th>
+      <% if defined?(Mysql2) %>
+        <% result.fields.each do |field| %>
+          <th><%= field.upcase %></th>
+        <% end %>
+      <% else %>
+        <% result.fetch_fields.each do |field| %>
+          <th><%= field.name.upcase %></th>
+        <% end %>
       <% end %>
-    </tr>
+   </tr>
   </thead>
   <tbody>
     <% i = 1 %>

diff --git a/lib/rack/bug/views/panels/profile_sql.html.erb b/lib/rack/bug/views/panels/profile_sql.html.erb
@@ -13,8 +13,14 @@
 <table class="sortable">
   <thead>
     <tr>
-      <% result.fetch_fields.each do |field| %>
-        <th><%= field.name.upcase %></th>
+      <% if defined?(Mysql2) %>
+        <% result.fields.each do |field| %>
+          <th><%= field.upcase %></th>
+        <% end %>
+      <% else %>
+        <% result.fetch_fields.each do |field| %>
+          <th><%= field.name.upcase %></th>
+        <% end %>
       <% end %>
     </tr>
   </thead>