Moduletrace/datasquirel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updates

Browse Source
This commit is contained in:
Benjamin Toby
2025-01-14 10:49:37 +01:00
parent 2118a55ab6
commit 0cd3eb418c
19 changed files with 307 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files
package-shared/shell/checkDb.ts
+16 -4
View File
@@ -1,6 +1,5 @@
require("dotenv").config({ path: "./../.env" });
import mysql from "serverless-mysql";
import grabDbSSL from "../utils/backend/grabDbSSL";
/**
* # Main DB Handler Function
@@ -14,17 +13,30 @@ import grabDbSSL from "../utils/backend/grabDbSSL";
* @returns {Promise<object|null>}
*/
(async () => {
const connection = global.DSQL_DB_CONN;
const CONNECTION =
global.DSQL_DB_CONN ||
mysql({
config: {
host: process.env.DSQL_DB_HOST,
user: process.env.DSQL_DB_USERNAME,
password: process.env.DSQL_DB_PASSWORD,
database: process.env.DSQL_DB_NAME,
port: process.env.DSQL_DB_PORT
? Number(process.env.DSQL_DB_PORT)
: undefined,
charset: "utf8mb4",
},
});
try {
const result = await connection.query(
const result = await CONNECTION.query(
"SELECT id,first_name,last_name FROM users LIMIT 3"
);
console.log("Connection Query Success =>", result);
} catch (error: any) {
console.log("Connection query ERROR =>", error.message);
} finally {
connection.end();
CONNECTION?.end();
process.exit();
}
})();
package-shared/shell/updateSSLUsers.ts
+17 -5
View File
@@ -1,5 +1,4 @@
require("dotenv").config({ path: "./../.env" });
import grabDbSSL from "../utils/backend/grabDbSSL";
import mysql from "serverless-mysql";
/**
@@ -14,10 +13,23 @@ import mysql from "serverless-mysql";
* @returns {Promise<object|null>}
*/
(async () => {
const connection = global.DSQL_DB_CONN;
const CONNECTION =
global.DSQL_DB_CONN ||
mysql({
config: {
host: process.env.DSQL_DB_HOST,
user: process.env.DSQL_DB_USERNAME,
password: process.env.DSQL_DB_PASSWORD,
database: process.env.DSQL_DB_NAME,
port: process.env.DSQL_DB_PORT
? Number(process.env.DSQL_DB_PORT)
: undefined,
charset: "utf8mb4",
},
});
try {
const result = await connection.query(
const result = await CONNECTION.query(
"SELECT user,host,ssl_type FROM mysql.user"
);
const parsedResults = JSON.parse(JSON.stringify(result));
@@ -39,7 +51,7 @@ import mysql from "serverless-mysql";
continue;
}
const addUserSSL = await connection.query(
const addUserSSL = await CONNECTION.query(
`ALTER USER '${User}'@'${Host}'`
);
@@ -48,7 +60,7 @@ import mysql from "serverless-mysql";
} catch (error: any) {
console.log("Connection query ERROR =>", error.message);
} finally {
connection.end();
CONNECTION.end();
process.exit();
}
})();
package-shared/shell/utils/dbHandler.ts
+18 -5
View File
@@ -17,15 +17,28 @@ export default async function dbHandler({
query,
values,
}: Param): Promise<any[] | object | null> {
let connection = global.DSQL_DB_CONN;
const CONNECTION =
global.DSQL_DB_CONN ||
mysql({
config: {
host: process.env.DSQL_DB_HOST,
user: process.env.DSQL_DB_USERNAME,
password: process.env.DSQL_DB_PASSWORD,
database: process.env.DSQL_DB_NAME,
port: process.env.DSQL_DB_PORT
? Number(process.env.DSQL_DB_PORT)
: undefined,
charset: "utf8mb4",
},
});
let results;
try {
if (query && values) {
results = await connection.query(query, values);
results = await CONNECTION.query(query, values);
} else {
results = await connection.query(query);
results = await CONNECTION.query(query);
}
} catch (/** @type {any} */ error: any) {
if (process.env.FIRST_RUN) {
@@ -34,7 +47,7 @@ export default async function dbHandler({
console.log("ERROR in dbHandler =>", error.message);
console.log(error);
console.log(connection.config());
console.log(CONNECTION.config());
fs.appendFileSync(
path.resolve(__dirname, "../.tmp/dbErrorLogs.txt"),
@@ -43,7 +56,7 @@ export default async function dbHandler({
);
results = null;
} finally {
await connection?.end();
await CONNECTION?.end();
}
if (results) {