init

backend/app/Support/TenantContext.php

@@ -0,0 +1,71 @@
+<?php
+
+declare(strict_types=1);
+
+namespace App\Support;
+
+final class TenantContext
+{
+    private ?string $tenantCode = null;
+
+    public function set(string $tenantCode): void
+    {
+        if ($tenantCode === '') {
+            throw new \InvalidArgumentException('tenant_code 不能为空。');
+        }
+
+        $this->tenantCode = $tenantCode;
+    }
+
+    public function forget(): void
+    {
+        $this->tenantCode = null;
+    }
+
+    public function has(): bool
+    {
+        return $this->tenantCode !== null;
+    }
+
+    public function get(): ?string
+    {
+        return $this->tenantCode;
+    }
+
+    public function code(): string
+    {
+        if ($this->tenantCode === null) {
+            throw new MissingTenantContextException;
+        }
+
+        return $this->tenantCode;
+    }
+
+    public function cacheKey(string $key): string
+    {
+        return 'tenant:'.$this->code().':'.$key;
+    }
+
+    public function lockKey(string $key): string
+    {
+        return 'lock:tenant:'.$this->code().':'.$key;
+    }
+
+    /**
+     * @template T
+     *
+     * @param  callable(): T  $callback
+     * @return T
+     */
+    public function run(string $tenantCode, callable $callback): mixed
+    {
+        $previous = $this->tenantCode;
+        $this->set($tenantCode);
+
+        try {
+            return $callback();
+        } finally {
+            $this->tenantCode = $previous;
+        }
+    }
+}