From 58544989fe1f5aad8696b35ce12e309550602a84 Mon Sep 17 00:00:00 2001
From: Nico Wagner <info@ni-wa.de>
Date: Tue, 1 Oct 2024 13:38:04 +0200
Subject: [PATCH] feat: add influxdb

---
 package-lock.json           |  7 +++++++
 package.json                |  1 +
 src/config/index.ts         | 32 +++++++++++++++++++++++++++++++
 src/dataHandler/influxdb.ts | 38 +++++++++++++++++++++++++++++++++++++
 src/index.ts                |  3 +++
 5 files changed, 81 insertions(+)
 create mode 100644 src/dataHandler/influxdb.ts

diff --git a/package-lock.json b/package-lock.json
index 79c9f62..9137244 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -9,6 +9,7 @@
       "version": "2.0.11",
       "license": "ISC",
       "dependencies": {
+        "@influxdata/influxdb-client": "^1.35.0",
         "convict": "^6.2.4",
         "mqtt": "^5.10.1",
         "mysql": "^2.18.1",
@@ -641,6 +642,12 @@
         "url": "https://github.com/sponsors/nzakas"
       }
     },
+    "node_modules/@influxdata/influxdb-client": {
+      "version": "1.35.0",
+      "resolved": "https://registry.npmjs.org/@influxdata/influxdb-client/-/influxdb-client-1.35.0.tgz",
+      "integrity": "sha512-woWMi8PDpPQpvTsRaUw4Ig+nOGS/CWwAwS66Fa1Vr/EkW+NEwxI8YfPBsdBMn33jK2Y86/qMiiuX/ROHIkJLTw==",
+      "license": "MIT"
+    },
     "node_modules/@istanbuljs/load-nyc-config": {
       "version": "1.1.0",
       "resolved": "https://registry.npmjs.org/@istanbuljs/load-nyc-config/-/load-nyc-config-1.1.0.tgz",
diff --git a/package.json b/package.json
index af846e8..cbaf444 100644
--- a/package.json
+++ b/package.json
@@ -23,6 +23,7 @@
   "author": "Trickfilm400",
   "license": "ISC",
   "dependencies": {
+    "@influxdata/influxdb-client": "^1.35.0",
     "convict": "^6.2.4",
     "mqtt": "^5.10.1",
     "mysql": "^2.18.1",
diff --git a/src/config/index.ts b/src/config/index.ts
index b172b15..3f6fc36 100644
--- a/src/config/index.ts
+++ b/src/config/index.ts
@@ -24,6 +24,38 @@ const config = convict({
       env: 'VANTAGE_PORT',
     },
   },
+  influxdb: {
+    enabled: {
+      doc: 'if influxdb should be used',
+      format: Boolean,
+      default: false,
+      env: 'INFLUXDB_ENABLED',
+    },
+    url: {
+      doc: 'The URL Endpoint for InfluxDB',
+      format: String,
+      default: '',
+      env: 'INFLUXDB_URL',
+    },
+    api_token: {
+      doc: 'The API Token for InfluxDB',
+      format: String,
+      default: '',
+      env: 'INFLUXDB_API_TOKEN',
+    },
+    organisation: {
+      doc: 'The organisation for InfluxDB',
+      format: String,
+      default: '',
+      env: 'INFLUXDB_ORGANISATION',
+    },
+    bucket: {
+      doc: 'The bucket for InfluxDB',
+      format: String,
+      default: '',
+      env: 'INFLUXDB_BUCKET',
+    },
+  },
   mysql: {
     enabled: {
       doc: 'The IP address Mysql should connect to',
diff --git a/src/dataHandler/influxdb.ts b/src/dataHandler/influxdb.ts
new file mode 100644
index 0000000..5fc80a4
--- /dev/null
+++ b/src/dataHandler/influxdb.ts
@@ -0,0 +1,38 @@
+import { DataReceiver } from '../lib/dataReceiver';
+import { DataPackage } from '../interfaces/IPackage';
+import { InfluxDB, Point, WriteApi } from '@influxdata/influxdb-client';
+import config from '../config';
+import { logger } from '../core/logger';
+
+export class Influxdb extends DataReceiver<DataPackage> {
+  private writerClient: WriteApi | undefined;
+  constructor() {
+    super();
+    if (config.get('influxdb.enabled')) {
+      const influxDB = new InfluxDB({
+        url: config.get('influxdb.url'),
+        token: config.get('influxdb.api_token'),
+      });
+      this.writerClient = influxDB.getWriteApi(
+        config.get('influxdb.organisation'),
+        config.get('influxdb.bucket')
+      );
+    }
+  }
+  onData(data: DataPackage): void {
+    logger.debug('Influxdb data', data);
+    if (!this.writerClient) return;
+
+    let key: keyof DataPackage;
+    const points: Point[] = [];
+    for (key in data) {
+      const point = new Point(key);
+      point.floatField('value', data[key]);
+      points.push(point);
+    }
+    this.writerClient.writePoints(points);
+  }
+  cleanup() {
+    this.writerClient?.close();
+  }
+}
diff --git a/src/index.ts b/src/index.ts
index 6cb28c2..29777ac 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -16,6 +16,8 @@ import Mqtt from './dataHandler/mqtt';
 const mqtt = new Mqtt();
 import { Prometheus } from './dataHandler/prometheus';
 const prometheus = new Prometheus();
+import { Influxdb } from './dataHandler/influxdb';
+const influxdb = new Influxdb();
 
 process.on('SIGINT', () => {
   console.log('\n');
@@ -24,6 +26,7 @@ process.on('SIGINT', () => {
     .then(() => socket.cleanup())
     .then(() => mqtt.cleanup())
     .then(() => prometheus.cleanup())
+    .then(() => influxdb.cleanup())
     .then(() => httpServer.cleanup())
     .then(() => {
       process.exit(0);