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 1/2] =?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{ From 073211bd97a6686f13ee5c6a7df9b1c349ccb11a Mon Sep 17 00:00:00 2001 From: pplokijuhyg <1162963624@qq.com> Date: Tue, 19 Nov 2019 13:37:11 +0800 Subject: [PATCH 2/2] =?UTF-8?q?=E8=87=AA=E5=8A=A8=E9=80=92=E5=A2=9E?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- bin/mongodb.js | 53 +++++++++++++++++++++++++++++++++------ controllers/login.js | 4 +-- description/mongodb.js.md | 7 +++++- 3 files changed, 53 insertions(+), 11 deletions(-) diff --git a/bin/mongodb.js b/bin/mongodb.js index 9fbf4d5..abd2524 100644 --- a/bin/mongodb.js +++ b/bin/mongodb.js @@ -1,7 +1,38 @@ var MongoClient = require('mongodb').MongoClient; var dburl; -let add = (set, data, url = dburl) => { +let idadd = async (set) => { + let id = 0 + if (set == "idadd") { + return 0; + } else { + await find('idadd', { 'set': set }).then((res) => { + if (res.data.length != 0) { + id = res.data[0].num_key + } else { + id = 1 + } + }) + } + + if (id == 1) { + await add('idadd', { set, 'num_key': 2 }) + } else if (set != "idadd") { + console.log(id) + + await update('idadd', { 'set': set }, { 'num_key': id + 1 }) + } + console.log(id) + return id + +} +let add = async (set, data, url = dburl) => { + if (set != "idadd") { + let id = await idadd(set) + data.num_key = id + } + return new Promise((resolve, reject) => { + MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true }, function (err, db) { if (err) throw err; var dbo = db.db(url.dbname); @@ -18,8 +49,16 @@ let add = (set, data, url = dburl) => { }); }) } -let adds = (set, data, url = dburl) => { +let adds = async (set, data, url = dburl) => { + + if (set != "idadd") { + for (let i in data) { + let id = await idadd(set) + data[i].num_key = id + } + } return new Promise((resolve, reject) => { + MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true }, function (err, db) { if (err) throw err; var dbo = db.db(url.dbname); @@ -36,7 +75,7 @@ let adds = (set, data, url = dburl) => { }); }) } -let remove = (set, where, url = dburl) => { +let remove = async (set, where, url = dburl) => { return new Promise((resolve, reject) => { MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true }, function (err, db) { if (err) throw err; @@ -54,7 +93,7 @@ let remove = (set, where, url = dburl) => { }); }) } -let removes = (set, where, url = dburl) => { +let removes = async (set, where, url = dburl) => { return new Promise((resolve, reject) => { if (where == "" || Object.keys(where).length == 0) { reject({ code: 1 }) @@ -75,7 +114,7 @@ let removes = (set, where, url = dburl) => { }); }) } -let find = (set, where = {}, page = 0, num = 0, url = dburl) => { +let find = async (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) { @@ -111,7 +150,7 @@ let find = (set, where = {}, page = 0, num = 0, url = dburl) => { }) } -let update = (set, where = {}, data, defin = true, url = dburl) => { +let update = async (set, where = {}, data, defin = true, url = dburl) => { return new Promise((resolve, reject) => { MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true }, function (err, db) { if (err) throw err; @@ -132,7 +171,7 @@ let update = (set, where = {}, data, defin = true, url = dburl) => { }); }) } -let updates = (set, where = {}, data, url = dburl) => { +let updates = async (set, where = {}, data, url = dburl) => { return new Promise((resolve, reject) => { MongoClient.connect(url.url, { useUnifiedTopology: true, useNewUrlParser: true }, function (err, db) { if (err) throw err; diff --git a/controllers/login.js b/controllers/login.js index d1e3a19..57c8f4b 100644 --- a/controllers/login.js +++ b/controllers/login.js @@ -6,15 +6,13 @@ var loginup = async (ctx,next) => { 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 }) } - + console.timeEnd("asd") //判断用户名密码 if(ctx.request.body.name == arr[0].name && ctx.request.body.pwd == arr[0].pwd){ //颁发token diff --git a/description/mongodb.js.md b/description/mongodb.js.md index 1c8114f..fd064a8 100644 --- a/description/mongodb.js.md +++ b/description/mongodb.js.md @@ -41,7 +41,7 @@ find 查询数据 ``` update 更新单条数据 ``` -参数(操作的集合名,条件,要改为的{}[,{数据库地址,数据库名}]) +参数(操作的集合名,条件,要改为的{}[,默认$set 如果其他 为flase 要改为的加操作符,{数据库地址,数据库名}]) 返回值类型 promise 成功返回 {code:0}, 失败返回 {code:1} @@ -53,3 +53,8 @@ updates 更新多条数据 成功返回 {code:0,num:修改成功的数量}, 失败返回 {code:1} ``` + +####默认 + + +默认idadd表为id存储表，num_key为id 自动递增 1开始