when user upload file, also update the database

blueBlue0102 · May 12, 2021 · 2e1ee9f · 2e1ee9f
1 parent 38ad7e1
commit 2e1ee9f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/routers/home.js b/routers/home.js
@@ -5,12 +5,14 @@ module.exports = async function (req, res) {
   else {
     const user = await sequelize.models.User.findByPk(req.user.id);
     if (user === null) {
+      // create account
       sequelize.models.User.create({
         google_id: req.user.id,
         display_name: req.user.displayName,
         last_login_date: Date.now(),
       });
     } else {
+      // update account
       sequelize.models.User.update(
         { last_login_date: Date.now() },
         {

diff --git a/routers/upload.js b/routers/upload.js
@@ -1,6 +1,7 @@
 const formidable = require("formidable");
 const path = require("path");
 const winston = require("winston");
+const sequelize = require("../models/index");
 
 const FORMIDABLE_CONFIG = {
   uploadDir: path.join(
@@ -63,6 +64,12 @@ module.exports = async (req, res, next) => {
       ip: req.ip,
     });
 
+    sequelize.models.File.create({
+      google_id: req.user.id,
+      file_name: files.file.newFilename,
+      expiration_date: null,
+    });
+
     res.status(200).json({
       imageName: files.file.newFilename,
       imagePath: path.join(