技术码头

node MongoDB做管理后台接口(注册登录)

2018-09-11  本文已影响84人  石菖蒲_xl

知识点

1、node
2、express
3、Mongoose
4、ES6

连接数据库

mongodb的地址我们一般需要配置在config文件中,做环境隔离,毕竟我们开发、测试、准生产、生产环境连接的数据库地址是不一样的,配置在config文件中,我们直接通过启动命令或者打包命令就可以切换到我们需要的mongodb地址,是开发必不可少的。
你可以使用 mongoose.connect()方法连接 MongoDB 。

//数据库地址
api:{
 db:mongodb://账户:密码@地址:端口/数据库名字
}
//封装连接数据库代码
import mongoose from 'mongoose';
import errorEnum from './errorEnum';
const connect = () => {
    return new Promise((res, rej) => {
        try {
            mongoose.connect(__API__.db, (err) => {
              if (err) {
                rej(errorEnum.DB_CONNECT())
              } else {
                res();
              }
            })
        } catch (err) {
            rej(errorEnum.DB_CONNECT())
        }
    })
}
export {connect}

其中errorEnum是我们集中处理异常码的文件,看一下代码,我们把所有需要抛给前端的错误码在这处理,清晰便于维护。

import resh from '../helpers/responseHelper';
const DB_CONNECT = () => {
    return resh.failureResponse('连接数据库失败', '1001')
}
...
export default {DB_CONNECT}

responseHelper中我们做的是把返回给前端接口的json格式进行封装,按照上文的异常情况返回给前端的应该是{desc:'连接数据库失败',code:'1001',body:{}},前端就可以根据错误码做数据库连接失败的异常处理机制。

const failureResponse = (desc, code) => {
    return {
        desc: desc,
        code: code,
        body: {}
    }
}
...
export default {
    failureResponse,
    ...
}

用户注册

1、用户输入账号(邮箱)密码注册账户
2、拿账号在数据库中查询账号是否已经存在,存在抛异常
3、账号在数据库中没有存在,则把账号信息存入数据库,存成功返回注册成功

定义一个Schema 、创建一个Model

Mongoose 的一切始于 Schema。每个 schema 都会映射到一个 MongoDB collection ,并定义这个collection里的文档的构成。

//我是 UserRegister.js
import mongoose from 'mongoose';
const registerSchema = new mongoose.Schema({
    id: String,
    mail: String,
    password: String,
    role:Array
}, {collection: 'user_register'});
module.exports = mongoose.model('user_register', registerSchema);

document 里每个属性的类型都会被转换为 在 blogSchema 里定义对应的 SchemaType
如上mail属性会被转化为SchemaType String,role属性会被转化为SchemaType Array,其实简单的说就是存在数据库中的类型,我们存的时候需要传的类型。

我们要把 schema 转换为一个 Model 使用 mongoose.model(modelName, schema) 函数,如上module.exports = mongoose.model('user_register', registerSchema);就是把registerSchema转化为Model,以后的增删改查都要通过这个Model实现。

其中user_register是在数据库中的表名,如下图,user_register表里边的数据结构跟schema中我们定义的对应一下,其中_id是mongodb自己生成的,是个对象,支持排序。

图1.数据库显示

数据库 增、查、改、删

find方法可按条件查询数据库
save提供往数据库新增一条数据的方法,继续以用户注册为例
下边接口的入参:{mail:'user@qq.com',password:'123456'}
增加完毕以后就如 图1 中所示,在数据库中插入一条数据,但是role[]

//定义Schema 创建Model的文件
import Register from '../model/UserRegister';
const register = (req,res)=>{
  const _params = req.body.data;//取得接口传过来的数据
  //此处省略对参数校验的过程,假设传过来的都是合法参数,但实际项目我们还是需要校验的
  const _findParams = {mail:_params.mail};
  //用账号去数据库中查找账号是否已经注册
  Register.find(_findParams, (err, data) => {
    if ( err ) {
      //抛出异常
    } else {
      //data是我们查数据库得到的数据,没有查到为[]
      if(data == 0){
        const register = new Register({
          ..._params,
        });
        register.save((err)=>{
          if(err){
            //数据库异常
          }else{
            //保存成功
            mongoose.disconnect();
          }
        });
      }else{
        //提示用户该账户已经注册
      }
    }     
  });
}
export default {register}

update提供了更新数据库的方法,也就是我们说的 “改”
用法Model.update(condition,doc,[options],[callback]);
condition:更新的条件,要求是一个对象。
doc:要更新的内容,要求是一个对象。
[options]:可选参数,要求也是一个对象。
[callback]:可选参数,要求是一个回调函数。

如下接口的入参:{id:'10011',role:['user']}
接口提供的是给一个用户分配权限的服务,初始注册权限为[]

//定义Schema 创建Model的文件
import Register from '../model/UserRegister';
const setAuthority = (req, res) => {
    const _params = req.body.data;
    const _id = _params.id;
    //根据id查到对应的数据,将role参数更新
    Register.update({_id: _id}, {$set: {role: _params.role}}, (err) => {
        if (err) {
            //数据库异常
        } else {
            //修改成功
        }
        mongoose.disconnect();
    })
}
export default {setAuthority}

$set用来指定一个键并更新键值,若键不存在并创建

登录使用update

登录的特殊性要求我们必须去更新数据库,毕竟我们不希望同样的账户可以到处登录,我们必须保证一个账号在登录表里只有一条记录,这样后登录的就覆盖先登录的,先登录的再次请求接口就会报token异常了。

存登录信息我们需要另一张表,就是说我们需要一个新的Schema,Model,上文已经介绍了怎么新建,这里就不重复说明了。
如下接口的入参:{mail:'user@qq.com',password:'123456'}

//定义Schema 创建Model的文件
import Login from '../model/UserLogin';
import Register from '../model/UserRegister';
const login=(req,res)=>{
  const _params = req.body.data;//取得接口传过来的数据
  //此处省略参数校验过程
  //先去注册表查询当前的登录的账号是否注册
  Register.find(_findParams, (err, data) => {
    if ( err ) {
      //抛出异常
    } else {
      //data是我们查数据库得到的数据,没有查到为[]
      if(data == 0){
        //提示用户没有注册
      }else{
        const _user = data[0];//取到注册数据
        //比较用户输入的密码和注册表里的密码是否相等
        if(_params['password']===_user['password']){
          //生成token
          const _token = uuid.v1();
          const _params = {
            ..._value,
            role: _user.role,//把注册里边的角色也保存到登录表里边
            token: _token
          };
          //账号和密码都匹配,这个时候使用注册表里的id去登录表里边查询数据,没有查到就创建,查到就更新。
          Login.upadate((_id:_user['_id']),{$set:{..._params}},{usert:true},(err)=>{
            if(err){
              //数据库错误
            }else{
              //保存成功 把token和role返回
            }
            mongoose.disconnect();//断开数据库连接
          }
        }else{
          //密码错误
        }
      }
    }     
  });
}

upsert boolean,如果不匹配,是否创建文档默认false,上边代码中用 upsert是为了如果没有在登录表中查到登录信息就新增一条登录数据,如果查到了就更新数据。

扩展知识点uuid

提供生成唯一标识的两个方法
npm install uuid --save
uuid.v1() 是基于时间戳生成的uid
uuid.v4() 是随机生成的uid
两者相比较当然是v1更好用了,v4还是有重复的可能


remove提供了删除数据的方法
以登录为例子,我们拿到token去登录表里查用户的登录信息,查到了就删除。
如下接口的入参:{token:'123'}

//定义Schema 创建Model的文件
import Login from '../model/UserLogin';
//登出
const logout = (req, res) => {
    const _params = req.body.data;//取得接口传过来的数据
    //使用token去查找
    Login.remove({token: _params.token}, (err) => {
        if (err) {
            //数据库异常
        } else {
            //删除成功
        }
    })
}
export default {
  logout
}

补充:前端在调用登出接口的时候一定要异步调用,前端自己要做清除缓存,跳转登录页面的操作,只是通知后台我退出了,即使接口调用失败前端也应该正常退出,下次登录的话就更新数据库呗。
如果前端等到登出接口返回成功再做操作,有几率用户一直退出不了。

上一篇下一篇

猜你喜欢

热点阅读