Express + MongoDB 实现用户注册，登录，获取用户信息，更新用户信息，获取用户列表功能

清疚 2022-10-30 15:26 224阅读 0赞

目录结构：

-| server
    	- model.js
    	- server.js
    	- user.js

### model.js ###

此文件用于存放模型

const mongoose = require('mongoose')
    // 连接 mongo 并且使用 goodwork 这个数据库
    const DB_URL = 'mongodb://localhost:27017/goodwork'
    mongoose.connect(DB_URL)
    
    const models = { 
    	user:{ 
    		'user':{ type:String, 'require':true},
    		'pwd':{ type:String, 'require':true},
    		'type':{ 'type':String, 'require':true},
    		'avatar':{ 'type':String},
    		'desc':{ 'type':String},
    		'title':{ 'type':String},
    		'company':{ 'type':String},
    		'money':{ 'type':String}
    	}
    }
    
    for(let m in models){ 
    	mongoose.model(m, new mongoose.Schema(models[m]))
    }
    
    module.exports = { 
    	getModel:function(name){ 
    		return mongoose.model(name)
    	}
    }

### server.js ###

主文件

const express = require('express')
    const bodyParser = require('body-parser')
    const cookieParser = require('cookie-parser')
    const model = require('./model')
    const app = express()
    
    const server = require('http').Server(app)
    
    const userRouter = require('./user')
    
    app.use(cookieParser())
    app.use(bodyParser.json())
    app.use('/user', userRouter)
    server.listen(9093, function() { 
      console.log('server port is 9093')
    })

### user.js ###

const {  json } = require('body-parser')
    const express = require('express')
    const utils = require('utility')
    const Router = express.Router()
    const model = require('./model')
    const User = model.getModel('user')
    
    const _filter = {  'pwd': 0, '__v': 0 }
    
    /** * 获取用户列表 */
    Router.get('/list', function (req, res) { 
      const {  type } = req.query
      User.find({  type }, function (err, doc) { 
        return res.json({  code: 0, data: doc })
      })
    })
    /** * 注册 */
    Router.post('/register', function (req, res) { 
      const {  user, pwd, type } = req.body
      User.findOne({  user }, function (err, doc) { 
        if (doc) { 
          return res.json({  code: 1, msg: '用户名重复' })
        }
    
        const userModel = new User({  user, type, pwd: md5Pwd(pwd) })
        userModel.save(function (e, d) { 
          if (e) { 
            return res.json({  code: 1, msg: '后端出错了' })
          }
          const {  user, type, _id } = d
          res.cookie('userid', _id)
          return res.json({  code: 0, data: {  user, type, _id } })
        })
      })
    })
    /** * 登录 */
    Router.post('/login', function (req, res) { 
      const {  user, pwd } = req.body
      User.findOne({  user, pwd: md5Pwd(pwd) }, _filter, function (err, doc) { 
        if (!doc) { 
          return res.json({ 
            code: 1,
            msg: '用户名或者密码错误'
          })
        }
        res.cookie('userid', doc._id)
        return res.json({ 
          code: 0,
          data: doc
        })
      })
    })
    /** * 用户信息 */
    Router.get('/info', function (req, res) { 
      const {  userid } = req.cookies
      if (!userid) { 
        return res.json({  code: 1 })
      }
      User.findOne({  _id: userid }, _filter, function (err, doc) { 
        if (err) { 
          return res.json({ 
            code: 1,
            msg: '服务器炸了'
          })
        }
        if (doc) { 
          return res.json({ 
            code: 0,
            data: doc
          })
        }
      })
    })
    /** * 完善信息 */
    Router.post('/update', function (req, res) { 
      const userid = req.cookies.userid
      if (!userid) { 
        return res.json({  code: 1 })
      }
      const body = req.body
      User.findByIdAndUpdate(userid, body, function (err, doc) { 
        const data = Object.assign({ }, { 
          user: doc.user,
          type: doc.type
        }, body)
        return res.json({  code: 0, data })
      })
    })
    
    function md5Pwd(pwd) { 
      const salt = 'is_good_3957x8yza6!@#IUHJh~~'
      return utils.md5(utils.md5(pwd + salt))
    }
    
    module.exports = Router

![在这里插入图片描述][20210220101347590.gif_pic_center]

[20210220101347590.gif_pic_center]: /images/20221024/2dfdddaf954b4a69aba9cb5b4fced954.png