feat: token을 업데이트 하는 로직 추가

JeongSH1 · Dec 8, 2023 · 6b7ee67 · 6b7ee67
1 parent ebee1e4
commit 6b7ee67
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/db/save-token.js b/db/save-token.js
@@ -1,6 +1,14 @@
 import { connection } from "./connection.js";
+import { getTokenByUserId } from "./get-token.js";
 
 export async function saveToken(userId, token) {
-    const [rows, fields] = await connection.execute('insert into tokens values(?, ?)', [userId, token]);
-    return rows;
+    const token = await getTokenByUserId(userId);
+
+    if (!token){
+      const [rows, fields] = await connection.execute('insert into tokens values(?, ?)', [userId, token]);
+      return rows;
+    } else {
+      const [rows, fields] = await connection.execute('update tokens set token = ? where userId = ?', [token, userId]);
+      return rows;
+    }
   }
diff --git a/noti.js b/noti.js
@@ -45,6 +45,7 @@ app.post('/send-notification', async (req, res) => {
   const { userId, title = '무제', body = 'blank' } = req.body;
 
   const token = await getTokenByUserId(userId);
+  console.log(userId, title, body, token);
 
   const message = {
     notification: {
@@ -63,6 +64,9 @@ app.post('/send-notification', async (req, res) => {
       console.error('Error sending message:', error);
       res.status(500).send('Error sending notification');
     });
+
+
+
 });
 
 app.listen(port, () => {