extend can modify exists windwalker-io/core#1170

windwalker-io · Nov 6, 2023 · 667f123 · 667f123
1 parent d549af5
commit 667f123
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/packages/di/src/Container.php b/packages/di/src/Container.php
@@ -378,6 +378,11 @@ public function has(string $id): bool
         return $this->getDefinition($id) !== null;
     }
 
+    public function hasCached(string $id): bool
+    {
+        return $this->getDefinition($id)?->getCache() !== null;
+    }
+
     /**
      * Remove an item from container.
      *
@@ -607,6 +612,10 @@ public function extend(string $id, Closure $closure): static
         $this->extends[$id] ??= [];
         $this->extends[$id][] = $closure;
 
+        if ($this->hasCached($id)) {
+            $this->modify($id, $closure);
+        }
+
         return $this;
     }