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

diff --git a/includes/libs/rdbms/loadbalancer/LoadBalancer.php b/includes/libs/rdbms/loadbalancer/LoadBalancer.php
index 405ed14160..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;
+			}
+		}
 	}
 
 	/**