fix log i think

VedalAI · Jun 10, 2024 · c5dbe2a · c5dbe2a
1 parent ddec0d8
commit c5dbe2a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/ebs/src/util/db.ts b/ebs/src/util/db.ts
@@ -39,7 +39,8 @@ export async function setupDb() {
     await db.query(`
         CREATE TABLE IF NOT EXISTS prepurchases (
             token VARCHAR(255) PRIMARY KEY,
-            cart JSON NOT NULL
+            cart JSON NOT NULL,
+            userId VARCHAR(255) NOT NULL
         )
     `);
 
@@ -75,7 +76,7 @@ export async function addFulfilledTransaction(receipt: string, token: string, us
 export async function registerPrepurchase(cart: IdentifiableCart): Promise<string> {
     try {
         const token = uuid();
-        await db.query("INSERT INTO prepurchases (token, cart) VALUES (?, ?)", [token, JSON.stringify(cart)]);
+        await db.query("INSERT INTO prepurchases (token, cart, userId) VALUES (?, ?, ?)", [token, JSON.stringify(cart), cart.userId]);
         return token;
     } catch (e: any) {
         console.error("Database query failed (registerPrepurchase)");

diff --git a/logger/src/util/db.ts b/logger/src/util/db.ts
@@ -21,8 +21,13 @@ export async function canLog(token: string | null): Promise<boolean> {
 
 export async function getUserIdFromTransactionToken(token: string): Promise<string | null> {
     try {
-        const [rows] = (await db.query("SELECT userId FROM transactions WHERE token = ?", [token])) as [RowDataPacket[], any];
-        return rows[0].userId;
+        try {
+            const [rows] = (await db.query("SELECT userId FROM prepurchases WHERE token = ?", [token])) as [RowDataPacket[], any];
+            return rows[0].userId;
+        } catch (e: any) {
+            const [rows] = (await db.query("SELECT userId FROM transactions WHERE token = ?", [token])) as [RowDataPacket[], any];
+            return rows[0].userId;
+        }
     } catch (e: any) {
         console.error("Database query failed (getUserIdFromTransactionToken)");
         console.error(e);