Avoid post-send DBPerformance log warnings for page rollback

Change-Id: I3d4eafc7d9d04647955822d5100d8cf253d80c8a

diff --git a/includes/actions/RollbackAction.php b/includes/actions/RollbackAction.php
index c962e20..dec3e61 100644 (file)
--- a/includes/actions/RollbackAction.php
+++ b/includes/actions/RollbackAction.php
@@ -73,6 +73,9 @@ class RollbackAction extends FormlessAction {
                $trxLimits = $this->context->getConfig()->get( 'TrxProfilerLimits' );
                $trxProfiler = Profiler::instance()->getTransactionProfiler();
                $trxProfiler->setExpectations( $trxLimits['POST'], __METHOD__ );
+               DeferredUpdates::addCallableUpdate( function () use ( $trxProfiler, $trxLimits ) {
+                       $trxProfiler->setExpectations( $trxLimits['PostSend-POST'], __METHOD__ );
+               } );
 
                $data = null;
                $errors = $this->page->doRollback(

diff --git a/includes/api/ApiRollback.php b/includes/api/ApiRollback.php
index db57f7e..10ba478 100644 (file)
--- a/includes/api/ApiRollback.php
+++ b/includes/api/ApiRollback.php
@@ -59,6 +59,9 @@ class ApiRollback extends ApiBase {
                $trxLimits = $this->getConfig()->get( 'TrxProfilerLimits' );
                $trxProfiler = Profiler::instance()->getTransactionProfiler();
                $trxProfiler->setExpectations( $trxLimits['POST'], __METHOD__ );
+               DeferredUpdates::addCallableUpdate( function () use ( $trxProfiler, $trxLimits ) {
+                       $trxProfiler->setExpectations( $trxLimits['PostSend-POST'], __METHOD__ );
+               } );
 
                $retval = $pageObj->doRollback(
                        $this->getRbUser( $params ),