feat: improve cache of eloquent datasource

ForestAdmin · Nov 12, 2024 · 0460b00 · 0460b00
1 parent 0b033c1
commit 0460b00
Show file tree

Hide file tree

Showing 4 changed files with 62 additions and 7 deletions.
diff --git a/packages/BaseDatasource/src/BaseDatasource.php b/packages/BaseDatasource/src/BaseDatasource.php
@@ -83,16 +83,24 @@ private function makeDoctrineConnection(array $databaseConfig): void
     /**
      * @codeCoverageIgnore
      */
-    public function __sleep(): array
+    public function __serialize(): array
     {
-        return ['collections', 'charts', 'databaseConfig'];
+        return [
+            'collections'    => $this->collections,
+            'charts'         => $this->charts,
+            'databaseConfig' => $this->databaseConfig,
+        ];
     }
 
     /**
      * @codeCoverageIgnore
      */
-    public function __wakeup(): void
+    public function __unserialize(array $data): void
     {
+        $this->collections = $data['collections'];
+        $this->charts = $data['charts'];
+        $this->databaseConfig = $data['databaseConfig'];
+
         $this->makeOrm($this->databaseConfig);
         $this->makeDoctrineConnection($this->databaseConfig);
     }

diff --git a/packages/DatasourceEloquent/src/EloquentCollection.php b/packages/DatasourceEloquent/src/EloquentCollection.php
@@ -338,4 +338,33 @@ private function nullablePolymorphicMorphManyRelationFields($record, $relationNa
             // Do nothing
         }
     }
+
+    public function __serialize(): array
+    {
+        return [
+            'fields'     => $this->fields,
+            'actions'    => $this->actions,
+            'segments'   => $this->segments,
+            'charts'     => $this->charts,
+            'schema'     => $this->schema,
+            'dataSource' => $this->dataSource,
+            'name'       => $this->name,
+            'tableName'  => $this->tableName,
+            'model'      => $this->model::class,
+        ];
+    }
+
+    public function __unserialize(array $data): void
+    {
+        $this->fields = $data['fields'];
+        $this->actions = $data['actions'];
+        $this->segments = $data['segments'];
+        $this->charts = $data['charts'];
+        $this->schema = $data['schema'];
+        $this->dataSource = $data['dataSource'];
+        $this->name = $data['name'];
+        $this->tableName = $data['tableName'];
+        $this->model = new $data['model']();
+        $this->nativeDriver = null;
+    }
 }
diff --git a/packages/DatasourceEloquent/src/EloquentDatasource.php b/packages/DatasourceEloquent/src/EloquentDatasource.php
@@ -69,4 +69,26 @@ public function getModels(): array
     {
         return $this->models;
     }
+
+    /**
+     * @codeCoverageIgnore
+     */
+    public function __serialize(): array
+    {
+        return array_merge(
+            parent::__serialize(),
+            ['supportPolymorphicRelations' => $this->supportPolymorphicRelations]
+        );
+    }
+
+    /**
+     * @codeCoverageIgnore
+     */
+    public function __unserialize(array $data): void
+    {
+        parent::__unserialize($data);
+
+        $finder = new ClassFinder(config('projectDir'));
+        $this->models = $finder->getModelsInNamespace('App');
+    }
 }
diff --git a/packages/DatasourceToolkit/src/CollectionMethods.php b/packages/DatasourceToolkit/src/CollectionMethods.php
@@ -60,10 +60,6 @@ public function setSegments(array $segments): Collection
      */
     public function addField(string $name, ColumnSchema|RelationSchema $field): void
     {
-        if ($this->fields->has($name)) {
-            throw new ForestException('Field ' . $name . ' already defined in collection');
-        }
-
         $this->fields->put($name, $field);
     }