Arsse/vendor/JKingWeb/NewsSync/Db/Common.php

<?php
declare(strict_types=1);
namespace JKingWeb\NewsSync\Db;
use JKingWeb\DrUUID\UUID as UUID;

Trait Common {
	protected $transDepth = 0;
	
	public function schemaVersion(): int {
		try {
			return $this->data->db->settingGet("schema_version");
		} catch(\Throwable $e) {
			return 0;
		}
	}
	
	public function begin(): bool {
		$this->exec("SAVEPOINT newssync_".($this->transDepth));
		$this->transDepth += 1;
		return true;
	}

	public function commit(bool $all = false): bool {
		if($this->transDepth==0) return false;
		if(!$all) {
			$this->exec("RELEASE SAVEPOINT newssync_".($this->transDepth - 1));
			$this->transDepth -= 1;
		} else {
			$this->exec("COMMIT TRANSACTION");
			$this->transDepth = 0;	
		}
		return true;
	}

	public function rollback(bool $all = false): bool {
		if($this->transDepth==0) return false;
		if(!$all) {
			$this->exec("ROLLBACK TRANSACTION TO SAVEPOINT newssync_".($this->transDepth - 1));
			// rollback to savepoint does not collpase the savepoint
			$this->commit();
			$this->transDepth -= 1;
			if($this->transDepth==0) $this->exec("ROLLBACK TRANSACTION");
		} else {
			$this->exec("ROLLBACK TRANSACTION");
			$this->transDepth = 0;
		}
		return true;
	}

	public function lock(): bool {
		if($this->schemaVersion() < 1) return true;
		if($this->isLocked()) return false;
		$uuid = UUID::mintStr();
		if(!$this->data->db->settingSet("lock", $uuid)) return false;
		sleep(1);
		if($this->data->db->settingGet("lock") != $uuid) return false;
		return true;
	}

	public function unlock(): bool {
		return $this->data->db->settingRemove("lock");
	}

	public function isLocked(): bool {
		if($this->schemaVersion() < 1) return false;
		return ($this->query("SELECT count(*) from newssync_settings where key = 'lock'")->getSingle() > 0);
	}

	public function prepare(string $query, string ...$paramType): Statement {
		return $this->prepareArray($query, $paramType);
	}

}