Add interface

fezfez · Jan 8, 2023 · 9f4a8f8 · 9f4a8f8
1 parent 4855ee8
commit 9f4a8f8
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 0 deletions.
diff --git a/src/Databases/Database.php b/src/Databases/Database.php
@@ -11,4 +11,10 @@ public function getDumpStructCommandLine(string $path): string;
     public function getDumpDataCommandLine(string $path): string;
 
     public function getRestoreCommandLine(string $path): string;
+
+    /** @param array<int, string> $onlyTables */
+    public function withOnlyTable(array $onlyTables): self;
+
+    /** @param array<int, string> $ignoreTables */
+    public function withIgnoreTable(array $ignoreTables): self;
 }
diff --git a/src/Databases/PostgresqlDatabase.php b/src/Databases/PostgresqlDatabase.php
@@ -9,15 +9,49 @@
 
 final class PostgresqlDatabase implements Database
 {
+    /**
+     * @param array<int, string> $ignoreTables
+     * @param array<int, string> $onlyTables
+     */
     public function __construct(
         private readonly string $host,
         private readonly string $port,
         private readonly string $user,
         private readonly string $password,
         private readonly string $database,
+        private readonly array $ignoreTables = [],
+        private readonly array $onlyTables = [],
     ) {
     }
 
+    /** @param array<int, string> $onlyTables */
+    public function withOnlyTable(array $onlyTables): self
+    {
+        return new self(
+            $this->host,
+            $this->port,
+            $this->user,
+            $this->password,
+            $this->database,
+            $this->ignoreTables,
+            $onlyTables,
+        );
+    }
+
+    /** @param array<int, string> $ignoreTables */
+    public function withIgnoreTable(array $ignoreTables): self
+    {
+        return new self(
+            $this->host,
+            $this->port,
+            $this->user,
+            $this->password,
+            $this->database,
+            $ignoreTables,
+            $this->onlyTables,
+        );
+    }
+
     public function getDumpDataCommandLine(string $path): string
     {
         return sprintf(