use restore

nazarpunk · Apr 15, 2021 · 7744704 · 7744704
1 parent 464b365
commit 7744704
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/Call.php b/src/Call.php
@@ -13,16 +13,18 @@ class Call {
 	private string        $connection;
 	private static string $connection_default;
 
-	private string $use = '';
+	private string $use_before = '';
+	private string $use_after  = '';
 
 	public function __construct(string $procedure) {
 		$this->connection = static::$connection_default;
 		$this->procedure  = $procedure;
 	}
 
 	//<editor-fold desc="use">
-	public function use(string $database): Call {
-		$this->use = $database;
+	public function use(string $database, string $restore = ''): Call {
+		$this->use_before = $database;
+		$this->use_after  = $restore;
 		return $this;
 	}
 	//</editor-fold>
@@ -325,10 +327,11 @@ public function execute(
 		}
 
 		// use
-		$use = strlen($this->use) ? "use `$this->use`;" : '';
+		$use_before = strlen($this->use_before) ? "use `$this->use_before`;" : '';
+		$use_after  = strlen($this->use_after) ? "use `$this->use_after`;" : '';
 
 		// result
-		$query = $use . implode('', $variable) . "call `$this->procedure` (" . implode(',', $argument) . ');';
+		$query = $use_before . implode('', $variable) . "call `$this->procedure` (" . implode(',', $argument) . ');' . $use_after;
 
 		$this->mysqli->multi_query($query);
 		$results = [];