add reference in d1

outerbase · Oct 22, 2024 · 5a2eed2 · 5a2eed2
1 parent 778057b
commit 5a2eed2
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/src/connections/sqlite/cloudflare.ts b/src/connections/sqlite/cloudflare.ts
@@ -250,6 +250,10 @@ export class CloudflareD1Connection extends SqliteBaseConnection {
                     });
                 }
 
+                const columnConstraint = fkConstraintData.find(
+                    (fk) => fk.from === column.name
+                );
+
                 const currentColumn: TableColumn = {
                     name: column.name,
                     position: column.cid,
@@ -259,6 +263,12 @@ export class CloudflareD1Connection extends SqliteBaseConnection {
                         default: column.dflt_value,
                         primaryKey: column.pk === 1,
                         unique: column.pk === 1,
+                        references: columnConstraint
+                            ? {
+                                  column: [columnConstraint.to],
+                                  table: columnConstraint.table,
+                              }
+                            : undefined,
                     },
                 };
 

diff --git a/tests/connections/connection.test.ts b/tests/connections/connection.test.ts
@@ -14,6 +14,15 @@ afterAll(async () => {
     await db.disconnect();
 });
 
+afterEach(async () => {
+    if (
+        ['cloudflare', 'bigquery'].includes(process.env.CONNECTION_TYPE ?? '')
+    ) {
+        // 3 seconds delay on each operation
+        await new Promise((r) => setTimeout(r, 3000));
+    }
+});
+
 function cleanup(data: Record<string, unknown>[]) {
     // Remove some database specified fields
     return data.map((d) => {