From f3f53e251993865070aa011846edab700cea2ca5 Mon Sep 17 00:00:00 2001
From: pplokijuhyg <1162963624@qq.com>
Date: Wed, 13 Nov 2019 20:17:51 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E5=96=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app.js               |   4 +-
 bin/mongodb.js       | 116 ++++++++++++++++++++++---------------------
 controllers/login.js |   8 ++-
 3 files changed, 68 insertions(+), 60 deletions(-)

diff --git a/app.js b/app.js
index aeb897e..69a85d3 100644
--- a/app.js
+++ b/app.js
@@ -8,9 +8,11 @@ const app = new Koa();
 app.use(async (ctx,next)=>{
     // await next();
     console.log(ctx.url)
+    // console.time("asd")
     // ctx.response.body = {a:1};
     await next()
-    console.log("执行")
+    // console.log("执行")
+    // console.timeEnd("asd")
 });
 
 
diff --git a/bin/mongodb.js b/bin/mongodb.js
index 5c5571b..9fbf4d5 100644
--- a/bin/mongodb.js
+++ b/bin/mongodb.js
@@ -1,158 +1,160 @@
 var MongoClient = require('mongodb').MongoClient;
 var dburl;
-let add = (set,data,url = dburl) =>{
+let add = (set, data, url = dburl) => {
     return new Promise((resolve, reject) => {
-        MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true}, function(err, db) {
+        MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true }, function (err, db) {
             if (err) throw err;
             var dbo = db.db(url.dbname);
             try {
-                dbo.collection(set).insertOne(data, function(err, res) {
+                dbo.collection(set).insertOne(data, function (err, res) {
                     if (err) throw err;
                     db.close();
-                    resolve({code:0});
+                    resolve({ code: 0 });
                 });
             } catch (error) {
-                reject({code:1,error})
+                reject({ code: 1, error })
             }
-            
+
         });
     })
 }
-let adds = (set,data,url = dburl) =>{
+let adds = (set, data, url = dburl) => {
     return new Promise((resolve, reject) => {
-        MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true}, function(err, db) {
+        MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true }, function (err, db) {
             if (err) throw err;
             var dbo = db.db(url.dbname);
             try {
-                dbo.collection(set).insertMany(data, function(err, res) {
+                dbo.collection(set).insertMany(data, function (err, res) {
                     if (err) throw err;
                     db.close();
-                    resolve({code:0,num:res.insertedCount});
+                    resolve({ code: 0, num: res.insertedCount });
                 });
             } catch (error) {
-                reject({code:1,error})
+                reject({ code: 1, error })
             }
-            
+
         });
     })
 }
-let remove = (set,where,url = dburl) =>{
+let remove = (set, where, url = dburl) => {
     return new Promise((resolve, reject) => {
-        MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true}, function(err, db) {
+        MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true }, function (err, db) {
             if (err) throw err;
             var dbo = db.db(url.dbname);
             try {
-                dbo.collection(set).deleteOne(where, function(err, obj) {
+                dbo.collection(set).deleteOne(where, function (err, obj) {
                     if (err) throw err;
                     db.close();
-                    resolve({code:0});
+                    resolve({ code: 0 });
                 });
             } catch (error) {
-                reject({code:1,error})
+                reject({ code: 1, error })
             }
-            
+
         });
     })
 }
-let removes = (set,where,url = dburl) =>{
+let removes = (set, where, url = dburl) => {
     return new Promise((resolve, reject) => {
-        if(where == "" ||Object.keys(where).length == 0){
-            reject({code:1})
+        if (where == "" || Object.keys(where).length == 0) {
+            reject({ code: 1 })
         }
-        MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true}, function(err, db) {
+        MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true }, function (err, db) {
             if (err) throw err;
             var dbo = db.db(url.dbname);
             try {
-                dbo.collection(set).deleteMany(where, function(err, obj) {
+                dbo.collection(set).deleteMany(where, function (err, obj) {
                     if (err) throw err;
                     db.close();
-                    resolve({code:0,num:obj.result.n});
+                    resolve({ code: 0, num: obj.result.n });
                 });
             } catch (error) {
-                reject({code:1,error})
+                reject({ code: 1, error })
             }
-            
+
         });
     })
 }
-let find = (set,where = {},page = 0,num = 0,url = dburl) =>{
+let find = (set, where = {}, page = 0, num = 0, url = dburl) => {
     return new Promise((resolve, reject) => {
-        if(num == 0){
-            MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true}, function(err, db) {
+        if (num == 0) {
+            MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true }, function (err, db) {
                 if (err) throw err;
                 var dbo = db.db(url.dbname);
                 try {
-                    dbo.collection(set). find(where).toArray(function(err, result) {
+                    dbo.collection(set).find(where).toArray(function (err, result) {
                         if (err) throw err;
-                        resolve({code:0,data:result});
+                        resolve({ code: 0, data: result });
                         db.close();
                     });
                 } catch (error) {
-                    reject({code:1,error})
+                    reject({ code: 1, error })
                 }
-                
+
             });
-        }else{
-            MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true}, function(err, db) {
+        } else {
+            MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true }, function (err, db) {
                 if (err) throw err;
                 var dbo = db.db(url.dbname);
                 try {
-                    dbo.collection(set). find(where).skip(num * page).limit(num).toArray(function(err, result) { 
+                    dbo.collection(set).find(where).skip(num * page).limit(num).toArray(function (err, result) {
                         if (err) throw err;
-                        resolve({code:0,data:result});
+                        resolve({ code: 0, data: result });
                         db.close();
                     });
                 } catch (error) {
-                    reject({code:1,error})
+                    reject({ code: 1, error })
                 }
-                
+
             });
         }
-        
+
     })
 }
-let update = (set,where = {},data,url = dburl) =>{
+let update = (set, where = {}, data, defin = true, url = dburl) => {
     return new Promise((resolve, reject) => {
-        MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true}, function(err, db) {
+        MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true }, function (err, db) {
             if (err) throw err;
             var dbo = db.db(url.dbname);
-            data = {$set:data}
+            if (defin) {
+                data = { $set: data }
+            }
             try {
-                dbo.collection(set).updateOne(where, data, function(err, res) {
+                dbo.collection(set).updateOne(where, data, function (err, res) {
                     if (err) throw err;
-                    resolve({code:0});
+                    resolve({ code: 0 });
                     db.close();
                 });
             } catch (error) {
-                reject({code:1,error})
+                reject({ code: 1, error })
             }
-            
+
         });
     })
 }
-let updates = (set,where = {},data,url = dburl) =>{
+let updates = (set, where = {}, data, url = dburl) => {
     return new Promise((resolve, reject) => {
-        MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true}, function(err, db) {
+        MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true }, function (err, db) {
             if (err) throw err;
             var dbo = db.db(url.dbname);
-            data = {$set:data}
+            data = { $set: data }
             try {
-                dbo.collection(set).updateMany(where, data, function(err, res) {
+                dbo.collection(set).updateMany(where, data, function (err, res) {
                     if (err) throw err;
                     db.close();
-                    resolve({code:0,num:res.result.nModified});
+                    resolve({ code: 0, num: res.result.nModified });
                 });
             } catch (error) {
-                reject({code:1,error})
+                reject({ code: 1, error })
             }
-            
+
         });
     })
 }
 
-module.exports = function (url,dbname){
-    dburl = {url,dbname};
-    return{
+module.exports = function (url, dbname) {
+    dburl = { url, dbname };
+    return {
         add,
         adds,
         remove,
diff --git a/controllers/login.js b/controllers/login.js
index e69ff08..d1e3a19 100644
--- a/controllers/login.js
+++ b/controllers/login.js
@@ -1,22 +1,26 @@
-
 var loginup = async (ctx,next) => {
     let arr;
     //查询数据库 没有用户默认admin 123456
+    console.time("asd")
+
     await dbs.find('admin').then((res)=>{
         arr = res.data
     })
+    console.timeEnd("asd")
+
     if(arr.length == 0){
         await dbs.add('admin',{name:"admin",pwd:"123456"})
         await dbs.find('admin').then((res)=>{
             arr = res.data
         })
     }
+
     //判断用户名密码
     if(ctx.request.body.name == arr[0].name && ctx.request.body.pwd == arr[0].pwd){
         //颁发token
         const token = jwt.sign({
             name: ctx.request.body.name,
-            _id: 1
+            id: arr[0]._id
         }, 'my_token', { expiresIn: '2h' });
         ctx.body={code:0,token}
     }else{