Re-add realtime data

queries/analytics/event/getEvents.js

@@ -0,0 +1,45 @@
+import prisma from 'lib/prisma';
+import clickhouse from 'lib/clickhouse';
+import { runQuery, CLICKHOUSE, PRISMA } from 'lib/db';
+
+export function getEvents(...args) {
+  return runQuery({
+    [PRISMA]: () => relationalQuery(...args),
+    [CLICKHOUSE]: () => clickhouseQuery(...args),
+  });
+}
+
+function relationalQuery(websites, start_at) {
+  return prisma.client.event.findMany({
+    where: {
+      websiteId: {
+        in: websites,
+      },
+      createdAt: {
+        gte: start_at,
+      },
+    },
+  });
+}
+
+function clickhouseQuery(websites, start_at) {
+  const { rawQuery, getDateFormat, getCommaSeparatedStringFormat } = clickhouse;
+
+  return rawQuery(
+    `select
+      event_id,
+      website_id, 
+      session_id,
+      created_at,
+      url,
+      event_name
+    from event
+    where event_name != ''
+      and ${
+        websites && websites.length > 0
+          ? `website_id in (${getCommaSeparatedStringFormat(websites)})`
+          : '0 = 0'
+      }
+      and created_at >= ${getDateFormat(start_at)}`,
+  );
+}

queries/analytics/pageview/getPageviews.js

@@ -0,0 +1,43 @@
+import prisma from 'lib/prisma';
+import clickhouse from 'lib/clickhouse';
+import { runQuery, CLICKHOUSE, PRISMA } from 'lib/db';
+
+export async function getPageviews(...args) {
+  return runQuery({
+    [PRISMA]: () => relationalQuery(...args),
+    [CLICKHOUSE]: () => clickhouseQuery(...args),
+  });
+}
+
+async function relationalQuery(websites, start_at) {
+  return prisma.client.pageview.findMany({
+    where: {
+      websiteId: {
+        in: websites,
+      },
+      createdAt: {
+        gte: start_at,
+      },
+    },
+  });
+}
+
+async function clickhouseQuery(websites, start_at) {
+  const { rawQuery, getCommaSeparatedStringFormat } = clickhouse;
+
+  return rawQuery(
+    `select
+        website_id,
+        session_id,
+        created_at,
+        url
+      from event
+      where event_name = ''
+        and ${
+          websites && websites.length > 0
+            ? `website_id in (${getCommaSeparatedStringFormat(websites)})`
+            : '0 = 0'
+        }
+      and created_at >= ${clickhouse.getDateFormat(start_at)}`,
+  );
+}

queries/analytics/session/getSessions.js

@@ -0,0 +1,52 @@
+import prisma from 'lib/prisma';
+import clickhouse from 'lib/clickhouse';
+import { runQuery, PRISMA, CLICKHOUSE } from 'lib/db';
+
+export async function getSessions(...args) {
+  return runQuery({
+    [PRISMA]: () => relationalQuery(...args),
+    [CLICKHOUSE]: () => clickhouseQuery(...args),
+  });
+}
+
+async function relationalQuery(websites, start_at) {
+  return prisma.client.session.findMany({
+    where: {
+      ...(websites && websites.length > 0
+        ? {
+            websiteId: {
+              in: websites,
+            },
+          }
+        : {}),
+      createdAt: {
+        gte: start_at,
+      },
+    },
+  });
+}
+
+async function clickhouseQuery(websites, start_at) {
+  const { rawQuery, getDateFormat, getCommaSeparatedStringFormat } = clickhouse;
+
+  return rawQuery(
+    `select distinct
+      session_id,
+      website_id,
+      created_at,
+      hostname,
+      browser,
+      os,
+      device,
+      screen,
+      language,
+      country
+    from event
+    where ${
+      websites && websites.length > 0
+        ? `website_id in (${getCommaSeparatedStringFormat(websites)})`
+        : '0 = 0'
+    }
+      and created_at >= ${getDateFormat(start_at)}`,
+  );
+}

queries/analytics/stats/getRealtimeData.ts

@@ -1,3 +1,30 @@
-export async function getRealtimeData() {
-  throw new Error('Not Implemented');
+import { getPageviews } from '../pageview/getPageviews';
+import { getSessions } from '../session/getSessions';
+import { getEvents } from '../event/getEvents';
+
+export async function getRealtimeData(websites, time) {
+  const [pageviews, sessions, events] = await Promise.all([
+    getPageviews(websites, time),
+    getSessions(websites, time),
+    getEvents(websites, time),
+  ]);
+
+  return {
+    pageviews: pageviews.map(({ id, ...props }) => ({
+      __id: `p${id}`,
+      pageviewId: id,
+      ...props,
+    })),
+    sessions: sessions.map(({ id, ...props }) => ({
+      __id: `s${id}`,
+      sessionId: id,
+      ...props,
+    })),
+    events: events.map(({ id, ...props }) => ({
+      __id: `e${id}`,
+      eventId: id,
+      ...props,
+    })),
+    timestamp: Date.now(),
+  };
 }