Adding functions to postgres

tago-io · Nov 7, 2024 · 060c718 · 060c718
1 parent e1cdf8c
commit 060c718
Show file tree

Hide file tree

Showing 6 changed files with 21 additions and 10 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/plugins/mysql/src/Helpers/DeviceDatabase.ts b/plugins/mysql/src/Helpers/DeviceDatabase.ts
@@ -21,6 +21,7 @@ export async function getDeviceConnection(id: TGenericID, type: TDeviceType) {
       table.timestamp("created_at").nullable();
       table.timestamp("chunk_timestamp_start");
       table.timestamp("chunk_timestamp_end");
+      table.string("serie", 100);
 
       if (type === "mutable") {
         table.timestamp("updated_at");
@@ -32,7 +33,6 @@ export async function getDeviceConnection(id: TGenericID, type: TDeviceType) {
       table.index(["time"]);
       table.index(["group"]);
       table.index(["created_at"]);
-      table.string("serie", 100);
     });
   }
 

diff --git a/plugins/postgres/package.json b/plugins/postgres/package.json
@@ -21,7 +21,8 @@
   },
   "dependencies": {
     "knex": "3.1.0",
-    "pg": "8.13.0"
+    "pg": "8.13.1",
+    "pg-copy-streams": "6.0.6"
   },
   "devDependencies": {
     "@types/pg": "8.11.10"

diff --git a/plugins/postgres/src/index.ts b/plugins/postgres/src/index.ts
@@ -63,6 +63,8 @@ import getHourlyStatistics from "./Providers/Statistic/getHourlyStatistics.ts";
 import getSummary from "./Providers/Summary/getSummary.ts";
 import getTagKeys from "./Providers/Tag/getTagKeys.ts";
 import configs from "./settings.ts";
+import exportDeviceData from "./Providers/DeviceData/exportDeviceData.ts";
+import importDeviceData from "./Providers/DeviceData/importDeviceData.ts";
 
 export const postgreSQL = new DatabaseModule({
   id: "PostgreSQL",
@@ -135,3 +137,5 @@ postgreSQL.getSummary = getSummary;
 postgreSQL.getTagKeys = getTagKeys;
 postgreSQL.setDeviceParams = setDeviceParams;
 postgreSQL.setPluginStorageItem = setPluginStorageItem;
+postgreSQL.exportDeviceData = exportDeviceData;
+postgreSQL.importDeviceData = importDeviceData;
diff --git a/plugins/sqlite/src/Providers/DeviceData/exportDeviceData.ts b/plugins/sqlite/src/Providers/DeviceData/exportDeviceData.ts
@@ -1,5 +1,4 @@
 import fs from "node:fs";
-import { Stream } from "node:stream";
 import type { TDeviceType, TGenericID } from "@tago-io/tcore-sdk/types";
 import { stringify } from "csv";
 import { getDeviceConnection } from "../../Helpers/DeviceDatabase.ts";

diff --git a/plugins/sqlite/src/Providers/DeviceData/importDeviceData.ts b/plugins/sqlite/src/Providers/DeviceData/importDeviceData.ts
@@ -15,7 +15,6 @@ async function _insertData(client: Knex, data: any[]) {
       return;
     })
     .catch((error) => {
-      console.log(error);
       Promise.reject(error);
     });
 }
@@ -48,7 +47,6 @@ async function importDeviceData(
         resolve("Data imported successfully");
       })
       .on("error", (error) => {
-        console.log(error);
         reject(error);
       });
   });