Komplettumstieg auf drizzle

crawler/package.json

@@ -16,6 +16,7 @@
   "dependencies": {
     "@types/node": "^20.12.7",
     "currency.js": "^2.0.4",
+    "drizzle-orm": "^0.30.8",
     "fs-extra": "^11.2.0",
     "inquirer": "^9.2.17",
     "ioredis": "^5.3.2",

crawler/postgres_test.ts

@@ -0,0 +1,80 @@
+import { pgTable, timestamp, integer, jsonb, uuid } from 'drizzle-orm/pg-core';
+import { InferModel } from 'drizzle-orm';
+import { sql } from 'drizzle-orm';
+import { drizzle } from 'drizzle-orm/node-postgres';
+import { Pool } from 'pg';
+
+// Definiere den benutzerdefinierten Typ für das JSON-Objekt
+type BusinessData = {
+  id?: string;
+  firstname: string;
+  lastname: string;
+  email: string;
+  phoneNumber: string;
+  companyLocation: string;
+  hasCompanyLogo: boolean;
+  hasProfile: boolean;
+};
+
+// Definiere die Tabelle "businesses"
+const businesses = pgTable('businesses', {
+  id: uuid('id').primaryKey(),
+  created: timestamp('created'),
+  updated: timestamp('updated'),
+  visits: integer('visits'),
+  lastVisit: timestamp('last_visit'),
+  data: jsonb('data'),
+});
+
+// Definiere den Typ für das Modell
+type Business = InferModel<typeof businesses, 'select'>;
+
+// Erstelle eine Verbindung zur Datenbank
+const pool = new Pool({
+  // Konfiguriere die Verbindungsoptionen
+});
+
+const db = drizzle(pool);
+
+// Beispiel für das Einfügen eines neuen Datensatzes
+const insertBusiness = async () => {
+  const businessData: BusinessData = {
+    firstname: 'Robert',
+    lastname: 'Jackson',
+    email: 'robert.jackson@texasbizbrokers.com',
+    phoneNumber: '(214) 555-7890',
+    companyLocation: 'Dallas - TX',
+    hasCompanyLogo: true,
+    hasProfile: true,
+  };
+
+  const [insertedBusiness] = await db
+    .with({
+      new_business: sql<{ generated_id: string; created: Date; updated: Date; visits: number; last_visit: Date; data: BusinessData }>`(${(qb) => {
+        return qb
+          .select({
+            generated_id: sql`uuid_generate_v4()`,
+            created: sql`NOW()`,
+            updated: sql`NOW()`,
+            visits: sql`0`,
+            last_visit: sql`NOW()`,
+            data: sql`jsonb_set(${JSON.stringify(businessData)}::jsonb, '{id}', to_jsonb(uuid_generate_v4()))`,
+          });
+      }})`
+    })
+    .insert(businesses)
+    .values((eb) => ({
+      id: eb.generated_id,
+      created: eb.created,
+      updated: eb.updated,
+      visits: eb.visits,
+      lastVisit: eb.last_visit,
+      data: sql`jsonb_set(${eb.data}::jsonb, '{id}', to_jsonb(${eb.generated_id}))`,
+    }))
+    .returning({ generatedId: businesses.id, jsonData: businesses.data });
+
+  console.log('Generated ID:', insertedBusiness.generatedId);
+  console.log('JSON Data:', insertedBusiness.jsonData);
+};
+
+insertBusiness();