From eaa208652de637c6ec8aed17e66eb36d9f17e7a4 Mon Sep 17 00:00:00 2001
From: Francis Cao <franciscao@gmail.com>
Date: Fri, 18 Nov 2022 12:08:13 -0800
Subject: [PATCH] add clickhouse route to deleteWebsite

---
 queries/admin/website.ts | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/queries/admin/website.ts b/queries/admin/website.ts
index 240798da..c88fb2db 100644
--- a/queries/admin/website.ts
+++ b/queries/admin/website.ts
@@ -1,6 +1,7 @@
 import { Prisma, Website } from '@prisma/client';
 import cache from 'lib/cache';
 import prisma from 'lib/prisma';
+import { runQuery, CLICKHOUSE, PRISMA } from 'lib/db';
 
 export async function createWebsiteByUser(
   userId: string,
@@ -150,7 +151,14 @@ export async function getAllWebsites(): Promise<(Website & { user: string })[]>
 
 export async function deleteWebsite(
   websiteId: string,
-): Promise<[Prisma.BatchPayload, Prisma.BatchPayload, Website]> {
+) {
+  return runQuery({
+    [PRISMA]: () => deleteWebsiteRelationalQuery(websiteId),
+    [CLICKHOUSE]: () => deleteWebsiteClickhouseQuery(websiteId),
+  });
+}
+
+async function deleteWebsiteRelationalQuery(websiteId): Promise<[Prisma.BatchPayload, Prisma.BatchPayload, Website]> {
   const { client, transaction } = prisma;
 
   return transaction([
@@ -174,3 +182,12 @@ export async function deleteWebsite(
     return data;
   });
 }
+
+async function deleteWebsiteClickhouseQuery(websiteId): Promise<Website> {
+  return prisma.client.website.update({
+    data: {
+      isDeleted: true,
+    },
+    where: { id: websiteId },
+  });
+}