Enable caching in initialization mw

src/Middleware/IdentificationMiddleware.php

@@ -6,6 +6,7 @@ namespace Stancl\Tenancy\Middleware;
 
 use Stancl\Tenancy\Contracts\TenantCouldNotBeIdentifiedException;
 use Stancl\Tenancy\Contracts\TenantResolver;
+use Stancl\Tenancy\Resolvers\CachedTenantResolver;
 use Stancl\Tenancy\Tenancy;
 
 abstract class IdentificationMiddleware
@@ -13,6 +14,15 @@ abstract class IdentificationMiddleware
     /** @var callable */
     public static $onFail;
 
+    /** @var bool */
+    public static $shouldCache = false;
+
+    /** @var int */
+    public static $cacheTTL = 3600; // seconds
+
+    /** @var string|null */
+    public static $cacheStore = null; // default
+
     /** @var Tenancy */
     protected $tenancy;
 
@@ -22,9 +32,15 @@ abstract class IdentificationMiddleware
     public function initializeTenancy($request, $next, ...$resolverArguments)
     {
         try {
-            $this->tenancy->initialize(
-                $this->resolver->resolve(...$resolverArguments)
-            );
+            if (static::$shouldCache) {
+                app(CachedTenantResolver::class)->resolve(
+                    get_class($this->resolver), $resolverArguments, static::$cacheTTL, static::$cacheStore
+                );
+            } else {
+                $this->tenancy->initialize(
+                    $this->resolver->resolve(...$resolverArguments)
+                ); 
+            }
         } catch (TenantCouldNotBeIdentifiedException $e) {
             $onFail = static::$onFail ?? function ($e) {
                 throw $e;

src/Middleware/InitializeTenancyByDomain.php

@@ -13,6 +13,15 @@ class InitializeTenancyByDomain extends IdentificationMiddleware
     /** @var callable|null */
     public static $onFail;
 
+    /** @var bool */
+    public static $shouldCache = false;
+
+    /** @var int */
+    public static $cacheTTL = 3600; // seconds
+
+    /** @var string|null */
+    public static $cacheStore = null; // default
+
     /** @var Tenancy */
     protected $tenancy;
 

src/Middleware/InitializeTenancyByPath.php

@@ -13,6 +13,18 @@ use Stancl\Tenancy\Tenancy;
 
 class InitializeTenancyByPath extends IdentificationMiddleware
 {
+    /** @var callable|null */
+    public static $onFail;
+
+    /** @var bool */
+    public static $shouldCache = false;
+
+    /** @var int */
+    public static $cacheTTL = 3600; // seconds
+
+    /** @var string|null */
+    public static $cacheStore = null; // default
+
     /** @var Tenancy */
     protected $tenancy;
 

src/Middleware/InitializeTenancyByRequestData.php

@@ -17,6 +17,18 @@ class InitializeTenancyByRequestData extends IdentificationMiddleware
     /** @var string|null */
     public static $queryParameter = 'tenant';
 
+    /** @var callable|null */
+    public static $onFail;
+
+    /** @var bool */
+    public static $shouldCache = false;
+
+    /** @var int */
+    public static $cacheTTL = 3600; // seconds
+
+    /** @var string|null */
+    public static $cacheStore = null; // default
+
     /** @var Tenancy */
     protected $tenancy;
 

src/Middleware/InitializeTenancyBySubdomain.php

@@ -23,6 +23,18 @@ class InitializeTenancyBySubdomain extends InitializeTenancyByDomain
      */
     public static $subdomainIndex = 0;
 
+    /** @var callable|null */
+    public static $onFail;
+
+    /** @var bool */
+    public static $shouldCache = false;
+
+    /** @var int */
+    public static $cacheTTL = 3600; // seconds
+
+    /** @var string|null */
+    public static $cacheStore = null; // default
+
     /**
      * Handle an incoming request.
      *

src/Resolvers/CachedTenantResolver.php

@@ -22,6 +22,8 @@ class CachedTenantResolver implements TenantResolver
     {
         $resolverClass = $args[0];
         $data = $args[1];
+        $ttl = $args[2];
+        $cacheStore = $args[3];
 
         /** @var TenantResolver $resolver */
         $resolver = app($resolverClass);

tests/CachedTenantResolverTest.php

@@ -4,12 +4,23 @@ declare(strict_types=1);
 
 namespace Stancl\Tenancy\Tests;
 
+use Illuminate\Support\Facades\Route;
+use Stancl\Tenancy\Middleware\InitializeTenancyByDomain;
+use Stancl\Tenancy\Middleware\InitializeTenancyByRequestData;
 use Stancl\Tenancy\Resolvers\CachedTenantResolver;
 use Stancl\Tenancy\Resolvers\DomainTenantResolver;
 use Stancl\Tenancy\Tests\Etc\Tenant;
 
 class CachedTenantResolverTest extends TestCase
 {
+    public function tearDown(): void
+    {
+        InitializeTenancyByDomain::$shouldCache = false;
+        InitializeTenancyByRequestData::$shouldCache = false;
+
+        parent::tearDown();
+    }
+
     /** @test */
     public function tenants_can_be_resolved_using_the_cached_resolver()
     {
@@ -38,4 +49,43 @@ class CachedTenantResolverTest extends TestCase
 
         $this->assertTrue($tenant->is(app(CachedTenantResolver::class)->resolve(DomainTenantResolver::class, ['acme'])));
     }
+
+    /** @test */
+    public function caching_can_be_configured_selectively_on_initialization_middleware()
+    {
+        InitializeTenancyByDomain::$shouldCache = true;
+
+        $tenant = Tenant::create([
+            'id' => 'acme',
+        ]);
+        $tenant->domains()->create([
+            'domain' => 'acme.localhost',
+        ]);
+
+        Route::get('/foo', function () {
+            return 'bar';
+        })->middleware(InitializeTenancyByDomain::class);
+
+        // create cache
+        $this->get('http://acme.localhost/foo')
+            ->assertSee('bar');
+        
+        $this->mock(CachedTenantResolver::class, function ($mock) {
+            return $mock->shouldReceive('resolve')->once(); // only once
+        });
+
+        // use cache
+        $this->get('http://acme.localhost/foo')
+            ->assertSee('bar');
+
+        Route::get('/abc', function () {
+            return 'xyz';
+        })->middleware(InitializeTenancyByRequestData::class);
+
+        $this->get('/abc?tenant=acme')
+            ->assertSee('xyz');
+
+        $this->get('/abc?tenant=acme')
+            ->assertSee('xyz');
+    }
 }