discordeno/examples/bigbot/src/analytics.ts

import { InfluxDB, Point } from '@influxdata/influxdb-client';
import type { RestManager } from 'discordeno/rest';

const INFLUX_ORG = process.env.INFLUX_ORG as string;
const INFLUX_BUCKET = process.env.INFLUX_BUCKET as string;
const INFLUX_TOKEN = process.env.INFLUX_TOKEN as string;
const INFLUX_URL = process.env.INFLUX_URL as string;

export const influxDB = INFLUX_URL && INFLUX_TOKEN ? new InfluxDB({ url: INFLUX_URL, token: INFLUX_TOKEN }) : undefined;
export const Influx = influxDB?.getWriteApi(INFLUX_ORG, INFLUX_BUCKET);

export const setupAnalyticsHooks = (rest: RestManager) => {
	// If influxdb data is provided, enable analytics in this proxy.
	if (Influx) {
		rest.fetching = function (options) {
			Influx?.writePoint(
				new Point('restEvents')
					// MARK THE TIME WHEN EVENT ARRIVED
					.timestamp(new Date())
					// SET THE GUILD ID
					.stringField('type', 'REQUEST_FETCHING')
					.tag('method', options.method)
					.tag('url', options.url)
					.tag('bucket', options.bucketId ?? 'NA'),
			);
		};

		rest.fetched = function (options, response) {
			Influx?.writePoint(
				new Point('restEvents')
					// MARK THE TIME WHEN EVENT ARRIVED
					.timestamp(new Date())
					// SET THE GUILD ID
					.stringField('type', 'REQUEST_FETCHED')
					.tag('method', options.method)
					.tag('url', options.url)
					.tag('bucket', options.bucketId ?? 'NA')
					.intField('status', response.status)
					.tag('statusText', response.statusText),
			);
		};

		setInterval(() => {
			console.log(`[Influx - REST] Saving events...`);
			Influx?.flush()
				.then(() => {
					console.log(`[Influx - REST] Saved events!`);
				})
				.catch((error) => {
					console.log(`[Influx - REST] Error saving events!`, error);
				});
			// Every 30seconds
		}, 30000);
	}
};