使用promise改了几个mysql函数

database/db.js

@@ -66,14 +66,51 @@ db.updateLastUseTags = function(user_id, tags) {
     });
 }
 
-db.clickBookmark = function(id){
+db.clickBookmark = function(id) {
     var sql = "UPDATE `bookmarks` SET `click_count`=`click_count`+1, `last_click`=now() WHERE (`id`='" + id + "')";
     return new Promise(function(resolve, reject) {
         client.query(sql, (err, result) => {
             if (err) {
                 reject(err);
             } else {
-                resolve(id);
+                resolve(result.affectedRows);
+            }
+        });
+    });
+};
+
+db.checkLogin = function(username, password) {
+    console.log('checkLogin');
+    var sql = "SELECT * FROM `users` WHERE `username` = '" + username + "'";
+    return new Promise(function(resolve, reject) {
+        client.query(sql, (err, result) => {
+            if (err) {
+                reject(err);
+            } else {
+                var ret = {
+                    logined: false,
+                    user: {},
+                }
+                if (password === result[0].password) {
+                    ret.logined = true;
+                    ret.user = result[0];
+                }
+
+                resolve(ret);
+            }
+        });
+    });
+};
+
+db.updateUserLastLogin = function(id) {
+    console.log('updateUserLastLogin');
+    var sql = "UPDATE `users` SET `last_login`=now() WHERE (`id`='" + id + "')";
+    return new Promise(function(resolve, reject) {
+        client.query(sql, (err, result) => {
+            if (err) {
+                reject(err);
+            } else {
+                resolve(result.affectedRows);
             }
         });
     });

routes/api.js

@@ -69,38 +69,28 @@ api.post('/logout', function(req, res) {
 });
 
 api.post('/clickBookmark', function(req, res) {
-    var params = req.body.params;
-    var id = params.id;
-    var sql = "UPDATE `bookmarks` SET `click_count`=`click_count`+1, `last_click`=now() WHERE (`id`='" + id + "')";
-    console.log(sql);
-    client.query(sql, function(error, result, fields) {
-        res.json({
-            id: id,
-        });
-    })
+    db.clickBookmark(req.body.params.id)
+        .then((affectedRows) => res.json({}))
+        .catch((err) => console.log('clickBookmark error', err));
 });
 
 api.post('/login', function(req, res) {
     var params = req.body.params;
     var username = params.username;
     var password = md5(params.password);
-
-    var sql = "SELECT * FROM `users` WHERE `username` = '" + username + "'";
-    client.query(sql, function(error, result, fields) {
-        var id = '';
-        var logined = false;
-        var pass = !error && result.length === 1 && password === result[0].password
-        console.log(password, result[0].password, pass)
-        if (pass) {
-            req.session.username = username;
-            logined = true;
-            id = result[0].id;
+    db.checkLogin(username, password)
+        .then((ret) => {
+            if (ret.logined) {
+                req.session.username = ret.user.username;
+                req.session.userId = ret.user.id;
+                db.updateUserLastLogin(ret.user.id);
             }
-        res.json({
-            logined: logined,
-            userId: id
-        });
+            res.json(ret);
         })
+        .then((affectedRows) => {
+            console.log('updateUserLastLogin affectedRows ', affectedRows)
+        })
+        .catch((err) => console.log('login error', err));
 });
 
 api.get('/autoLogin', function(req, res) {