Merge pull request #67 from mrtcode/fix-concurrent-request-limiter

Register a shutdown function for concurrent request finishing
zotero · Jun 8, 2023 · 2cf66c3 · 2cf66c3
2 parents 17e67f4 + d421978
commit 2cf66c3
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/controllers/ApiController.php b/controllers/ApiController.php
@@ -99,6 +99,7 @@ public function init($extra) {
 			});
 		}
 
+		register_shutdown_function(array($this, 'finishConcurrentRequest'));
 		register_shutdown_function(array($this, 'checkDBTransactionState'));
 		register_shutdown_function(array($this, 'logTotalRequestTime'));
 		register_shutdown_function(array($this, 'checkForFatalError'));
@@ -1164,10 +1165,6 @@ protected function redirect($url, $httpCode=302) {
 
 
 	protected function end() {
-		if (Z_RequestLimiter::isConcurrentRequestActive()) {
-			Z_RequestLimiter::finishConcurrentRequest();
-		}
-
 		if ($this->profile && $this->currentRequestTime() > $this->timeLogThreshold) {
 			Zotero_DB::profileEnd($this->objectLibraryID, true, $this->uri);
 		}
@@ -1388,6 +1385,11 @@ public function handleError($no, $str, $file, $line) {
 		$this->handleException($e);
 	}
 
+	public function finishConcurrentRequest() {
+		if (Z_RequestLimiter::isConcurrentRequestActive()) {
+			Z_RequestLimiter::finishConcurrentRequest();
+		}
+	}
 
 	public function checkDBTransactionState($noLog = false) {
 		if (Zotero_DB::transactionInProgress()) {