koa2实现登录注册功能的示例代码

程序员文章站 2022-06-22 15:46:47

本文介绍了koa2实现登录注册功能的示例代码，分享给大家，具体如下: 这个主要结合前几天的内容，做个实际案例的效果版本：项目结构：前几天，我们把注册和...

本文介绍了koa2实现登录注册功能的示例代码，分享给大家，具体如下:

这个主要结合前几天的内容，做个实际案例的效果

版本：

koa2实现登录注册功能的示例代码

项目结构：

koa2实现登录注册功能的示例代码

前几天，我们把注册和登录的页面demo实现了，今天我们主要实现这么几个内容

注册新用户
判断该邮箱是否注册过
登录判断是否注册过
登录时的密码的正确

本文代码地址：

明天，我们将利用session实现登录状态判断

今天的这篇是在之前的代码基础上添加的，我们就晒主要的代码

1.编辑user.js

// db/user.js
const mongoose = require('./db')
const schema = mongoose.schema;

const ceshischema = new schema({
 email: string,
 name: string,
 password: string
});

const mymodel = mongoose.model('user', ceshischema);


class userdb {
 constructor () {

 }
// 查询
 query (obj = {}) {
  return new promise((resolve, reject) => {
  mymodel.find(obj, (err, res) => {
   if(err) {
   reject(err)
   }
   resolve(res)
  })
  })
 }
queryemail (em) {
 return new promise((resolve, reject) => {
  mymodel.find({email: em}, (err, res) => {
  if(err) {
   reject(err)
  }
  const len = res.length
  if(len >= 1){
   // 存在
   resolve(res)
  }else {
   // 不存在
   resolve(null)
  }
  })
 })
}
// 保存
 save (obj) {
  const m = new mymodel(obj)
  return new promise((resolve, reject)=> {
  m.save((err, res) => {
   if (err) {
   reject(err)
   }
   resolve(res)
   console.log(res)
  })
  })
  
 }
}
module.exports = new userdb()

上面主要查询用户，和保存用户

2.编辑登录注册路由

// router/index.js

const router = require('koa-router')


// 
const user = require('../db/user')
const home = new router()

home.get('/', async (ctx) => {
 let title = '首页'
 await ctx.render('index', {
  title
 })
})


// 子路由2
const page = new router()

page.get('/404', async (ctx) => {
 let title = "404"
 await ctx.render('err', {
  title
 })
})


const login = new router()

login.get('/', async (ctx) => {
 let title = "登录"
 await ctx.render('login', {
  title
 })
}).post('/', async (ctx) => {
 const data = ctx.request.body
 let queryres = await user.queryemail(data.email)
 console.log(queryres)
 if (queryres) {
  if(queryres[0].password === data.password) {
   ctx.body = {
    'code': 1,
    'data': queryres[0],
    'mesg': '登录成功'
   }
  }else {
   ctx.body = {
    'code': 0,
    'data': {},
    'mesg': '密码错误'
   }
  }
  
 }else {
  ctx.body = {
   'code': 0,
   'data': {},
   'mesg': '没有该用户，去注册吧'
  }
 }
})

const register = new router()

register.get('/', async (ctx) => {
 let title = "注册"
 await ctx.render('register', {
  title
 })
}).post('/', async (ctx) => {
 const data = ctx.request.body
 let queryres = await user.queryemail(data.email)
 if (queryres) {
  ctx.body = {
   'code': 0,
   'data': {},
   'mesg': '该邮箱已经存在哦'
  }
 }else {
  await user.save(data)
  ctx.body = {
   'code': 1,
   'data': {},
   'mesg': '保存成功'
  }
 }
 
})

// 装载所有子路由
let router = new router()
router.use('/', home.routes(), home.allowedmethods())
router.use('/page', page.routes(), page.allowedmethods())
router.use('/login', login.routes(), login.allowedmethods())
router.use('/register', register.routes(), register.allowedmethods())


module.exports = router

3.编辑index.js

const koa = require('koa')
const views = require('koa-views')
const statics = require('koa-static')
const bodyparser = require('koa-bodyparser')
const path = require('path')
const router = require('./router')
const app = new koa()

const staticpath = './static'

app.use(statics(
 path.join(__dirname, staticpath)
))
app.use(bodyparser())

// 加载模板引擎
app.use(views(path.join(__dirname, './views'), {
 extension: 'ejs'
 }))

// 加载路由中间件
app.use(router.routes()).use(router.allowedmethods())

app.listen(3000, () => {
 console.log('localhost:3000')
})

这个和之前的内容增加了koa-bodyparser对post数据的解析，利用ctx.request.body获取

4.启动服务