X-Git-Url: http://git.cyclocoop.org/?a=blobdiff_plain;f=includes%2Flibs%2Frdbms%2Floadbalancer%2FLoadBalancer.php;h=360be4256e99f287208f9bc2e2efb9b95fc8b13d;hb=f93af41d4528bae0a4b0a1c791682e39686c2bdf;hp=cb6e4f4fa9d6709b6ed029a9475f66f93ffe965a;hpb=4409a6c8ed85d210e6c363a715a8a8a76c18b917;p=lhc%2Fweb%2Fwiklou.git

diff --git a/includes/libs/rdbms/loadbalancer/LoadBalancer.php b/includes/libs/rdbms/loadbalancer/LoadBalancer.php
index cb6e4f4fa9..360be4256e 100644
--- a/includes/libs/rdbms/loadbalancer/LoadBalancer.php
+++ b/includes/libs/rdbms/loadbalancer/LoadBalancer.php
@@ -261,6 +261,14 @@ class LoadBalancer implements ILoadBalancer {
 		if ( isset( $params['chronologyCallback'] ) ) {
 			$this->chronologyCallback = $params['chronologyCallback'];
 		}
+
+		if ( isset( $params['roundStage'] ) ) {
+			if ( $params['roundStage'] === self::STAGE_POSTCOMMIT_CALLBACKS ) {
+				$this->trxRoundStage = self::ROUND_COMMIT_CALLBACKS;
+			} elseif ( $params['roundStage'] === self::STAGE_POSTROLLBACK_CALLBACKS ) {
+				$this->trxRoundStage = self::ROUND_ROLLBACK_CALLBACKS;
+			}
+		}
 	}
 
 	/**
@@ -1565,7 +1573,7 @@ class LoadBalancer implements ILoadBalancer {
 		} );
 	}
 
-	private function flushMasterSnapshots( $fname = __METHOD__ ) {
+	public function flushMasterSnapshots( $fname = __METHOD__ ) {
 		$this->forEachOpenMasterConnection( function ( IDatabase $conn ) use ( $fname ) {
 			$conn->flushSnapshot( $fname );
 		} );