Add collect limits.

queries/analytics/session/getSession.ts

@@ -1,7 +1,7 @@
 import clickhouse from 'lib/clickhouse';
 import { CLICKHOUSE, PRISMA, runQuery } from 'lib/db';
 import prisma from 'lib/prisma';
-import { Prisma } from '@prisma/client';
+import { Prisma, Session } from '@prisma/client';
 
 export async function getSession(args: { id: string }) {
   return runQuery({
@@ -10,7 +10,7 @@ export async function getSession(args: { id: string }) {
   });
 }
 
-async function relationalQuery(where: Prisma.SessionWhereUniqueInput) {
+async function relationalQuery(where: Prisma.SessionWhereUniqueInput): Promise<Session> {
   return prisma.client.session.findUnique({
     where,
   });

queries/analytics/stats/getViewTotals.ts

@@ -0,0 +1,38 @@
+import clickhouse from 'lib/clickhouse';
+import { CLICKHOUSE, PRISMA, runQuery } from 'lib/db';
+import prisma from 'lib/prisma';
+
+export async function getViewTotals(...args: [websites: string[], startDate: Date]) {
+  return runQuery({
+    [PRISMA]: () => relationalQuery(...args),
+    [CLICKHOUSE]: () => clickhouseQuery(...args),
+  });
+}
+
+async function relationalQuery(websites: string[], startDate: Date) {
+  return prisma.client.websiteEvent.count({
+    where: {
+      websiteId: {
+        in: websites,
+      },
+      createdAt: {
+        gte: startDate,
+      },
+    },
+  });
+}
+
+async function clickhouseQuery(websiteIds: string[], startDate: Date) {
+  const { rawQuery, getDateFormat, getCommaSeparatedStringFormat, findFirst } = clickhouse;
+
+  return rawQuery(
+    `
+    select        
+      count(*) as views
+    from event
+    where
+      website_id in (${getCommaSeparatedStringFormat(websiteIds)})
+      and created_at >= ${getDateFormat(startDate)}
+     `,
+  ).then(data => findFirst(data));
+}

queries/analytics/stats/getWebsiteStats.ts

@@ -18,7 +18,7 @@ async function relationalQuery(
 ) {
   const { startDate, endDate, filters = {} } = data;
   const { getDateQuery, getTimestampInterval, parseFilters, rawQuery } = prisma;
-  const params = [startDate, endDate];
+  const params: any = [startDate, endDate];
   const { filterQuery, joinSession } = parseFilters(filters, params);
 
   return rawQuery(