Added $tries to Database::transaction

Author: rkrx

src/Database.php

@@ -99,9 +99,10 @@ public function transactionCommit();
 	public function transactionRollback();
 
 	/**
-	 * @param callable $callback
+	 * @param int|callable $tries
+	 * @param callable|null $callback
+	 * @return $this
 	 * @throws \Exception
-	 * @return mixed
 	 */
-	public function transaction($callback);
+	public function transaction($tries = 1, $callback = null);
 }

src/Databases/MySQL.php

@@ -236,19 +236,29 @@ public function transactionRollback() {
 	}
 
 	/**
-	 * @param callable $callback
-	 * @throws \Exception
+	 * @param int|callable $tries
+	 * @param callable|null $callback
 	 * @return $this
+	 * @throws \Exception
 	 */
-	public function transaction($callback) {
-		try {
-			$this->transactionStart();
-			$result = call_user_func($callback, $this);
-			$this->transactionCommit();
-			return $result;
-		} catch (\Exception $e) {
-			$this->transactionRollback();
-			throw $e;
+	public function transaction($tries = 1, $callback = null) {
+		if(is_callable($tries)) {
+			$callback = $tries;
+			$tries = 1;
+		} elseif(!is_callable($callback)) {
+			throw new \Exception("Callable must be a callable");
+		}
+		$e = null;
+		for(; $tries--;) {
+			try {
+				$this->transactionStart();
+				$result = call_user_func($callback, $this);
+				$this->transactionCommit();
+				return $result;
+			} catch (\Exception $e) {
+				$this->transactionRollback();
+			}
 		}
+		throw $e;
 	}
 }