Added comments to queries

CHANGELOG.md

@@ -1,6 +1,7 @@
 ## 1.0.1 [unreleased]
 
 - Added `user_method` option
+- Added comments to queries
 
 ## 1.0.0
 

app/controllers/blazer/queries_controller.rb

@@ -65,7 +65,7 @@ module Blazer
         end
 
         @data_source = Blazer.data_sources[data_source]
-        @rows, @error = @data_source.run_statement(@statement)
+        @rows, @error = @data_source.run_statement(@statement, user: blazer_user, query: @query)
 
         if @query && !@error.to_s.include?("canceling statement due to statement timeout")
           @query.checks.each do |check|

lib/blazer/data_source.rb

@@ -34,13 +34,20 @@ module Blazer
       settings["timeout"]
     end
 
-    def run_statement(statement)
+    def run_statement(statement, options = {})
       rows = []
       error = nil
       begin
         connection_model.transaction do
           connection_model.connection.execute("SET statement_timeout = #{timeout.to_i * 1000}") if timeout && postgresql?
-          result = connection_model.connection.select_all(statement)
+          comment = "blazer"
+          if options[:user].respond_to?(:id)
+            comment << ",user_id:#{options[:user].id}"
+          end
+          if options[:query].respond_to?(:id)
+            comment << ",query_id:#{options[:query].id}"
+          end
+          result = connection_model.connection.select_all("#{statement} /*#{comment}*/")
           result.each do |untyped_row|
             row = {}
             untyped_row.each do |k, v|