Add StatementLog support for REST2

bestpractical · Jan 30, 2024 · c1447d9 · c1447d9
1 parent 57099ec
commit c1447d9
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/lib/RT/Interface/Web.pm b/lib/RT/Interface/Web.pm
@@ -1426,8 +1426,10 @@ sub LogRecordedSQLStatements {
     $RT::Handle->AddRequestToHistory({
         %{ $args{RequestData} },
         Queries => \@log,
-    });
+    }) if $args{RequestData};
 
+    my $current_user = $args{CurrentUser} || $HTML::Mason::Commands::session{'CurrentUser'};
+    my $current_user_name = $current_user ? $current_user->Name : '';
     for my $stmt (@log) {
         my ( $time, $sql, $bind, $duration ) = @{$stmt};
         my @bind;
@@ -1440,7 +1442,7 @@ sub LogRecordedSQLStatements {
         }
         $RT::Logger->log(
             level   => $log_sql_statements,
-            message => ($HTML::Mason::Commands::session{'CurrentUser'} ? $HTML::Mason::Commands::session{'CurrentUser'}->Name : '')
+            message => $current_user_name
                 . " - "
                 . "SQL("
                 . sprintf( "%.6f", $duration )

diff --git a/lib/RT/REST2/Dispatcher.pm b/lib/RT/REST2/Dispatcher.pm
@@ -94,6 +94,8 @@ sub to_psgi_app {
 
         RT->SetCurrentInterface('REST2');
         RT::ConnectToDatabase();
+        RT::Interface::Web::MaybeEnableSQLStatementLog();
+
         my $dispatch = $self->_dispatcher->dispatch($env->{PATH_INFO});
 
         return [404, ['Content-Type' => 'text/plain'], 'Not Found']

diff --git a/lib/RT/REST2/Middleware/Log.pm b/lib/RT/REST2/Middleware/Log.pm
@@ -62,7 +62,9 @@ sub call {
         RT->Logger->log(%$what);
     };
 
-    return $self->app->($env);
+    my $ret = $self->app->($env);
+    RT::Interface::Web::LogRecordedSQLStatements( CurrentUser => $env->{'rt.current_user'} );
+    return $ret;
 }
 
 1;