Node.js + Express 4.x + MongoDB 构建登录注册-简易用户管理（五）

　　Node.js连接MongoDB、封装DBHelp、增删查改 都集齐了。

　　下面开始Node.js连接MySQL。

　　假设，正在看博客的读者有一定的T-SQL基础。

一、安装MySQL

　　MySQL 5.7.17版： http://pan.baidu.com/s/1hsMcQFi

　　提取码：8leg

　　安装MySQL的时候，步骤中会要求输入root的密码，请妥善保管，后面有用。

　　MySQL也是命令行形式的操作不方便，需要操作管理平台，这里推荐Navicat。

　　Navicat Premium 11.0.17 64位：http://pan.baidu.com/s/1bSezqY

　　提取码：q2hl

二、安装mysql模块

　　执行命令：npm install mysql

三、连接MySQL

　　mysql数据库，新建库userinfo，新建表users，表里有3个字段（id、username、password），其中id自动增加。

　　废话不多说，直接贴上封装成DBHelp以后的代码，如下：

const mysql=require('mysql');

let connection=mysql.createConnection({
    host:'localhost',
    user:'root',
    password:'****',  //填自己mysql数据库的密码
    database:'userinfo'
});


function DBHelp(){}

/**************************
*
* 功能：查询
* 参数：sqlStr(SQL语句)、callback(回调函数)
*
**************************/
DBHelp.prototype.Find=function(sqlStr,callback)
{
    connection=mysql.createConnection(connection.config);
    connection.connect();
    connection.query(sqlStr,function(err,result)
    {
        if(err)
        {
            console.log(err);
            return;
        }
        callback(result);
        /*
            result是数组对象,类似：[RowDataPacket{username:'fuzq'},RowDataPacket{username:'admin'}]
        */
    });
    connection.end();
};

/**************************
*
* 功能：添加
* 参数：sqlStr(SQL语句)、data(添加的数据)、callback(回调函数)
*
**************************/
DBHelp.prototype.Add=function(sqlStr,data,callback)
{
    connection=mysql.createConnection(connection.config);
    connection.connect();
    connection.query(sqlStr,data,function(err)
    {
        if(err)
        {
            console.log(err);
            return;
        }
        callback();
    });
    connection.end();
};

/**************************
*
* 功能：删除
* 参数：sqlStr(SQL语句)、callback(回调函数)
*
**************************/
DBHelp.prototype.Delete=function(sqlStr,callback)
{
    connection=mysql.createConnection(connection.config);
    connection.connect();
    connection.query(sqlStr,function(err)
    {
        if(err)
        {
            console.log(err);
            return;
        }
        callback();
    });
    connection.end();
};

/**************************
*
* 功能：修改
* 参数：sqlStr(SQL语句)、data(修改的数据)、callback(回调函数)
*
**************************/
DBHelp.prototype.Update=function(sqlStr,data,callback)
{
    connection=mysql.createConnection(connection.config);
    connection.connect();
    connection.query(sqlStr,data,function(err)
    {
        if(err)
        {
            console.log(err);
            return;
        }
        callback();
    });
    connection.end();
};

module.exports=DBHelp;

/*
     Error: Cannot enqueue Query after invoking quit.
     原因：多次调用connection.end()导致的
     原因在于node连接上mysql后如果因网络原因丢失连接或者用户手工关闭连接后，原有的连接挂掉，需要重新连接。
     那么每次访问都断开连接，就要每次访问都重新建立连接。
     另一种解决办法是不使用“mysql”模块，改用“easymysql”模块
     “easymysql”模块链接：https://github.com/aleafs/easymysql
*/

　　对应的index.js也要修改，代码如下：

var express = require('express');
var router = express.Router();
const DBHelp=require('./DBHelp');

/* GET home page. */
router.get('/', function(req, res, next)
{
  res.render('index', { title: 'Index' });
});


//登录
router.route('/login').all(Logined).get(function(req,res)
{
    res.render('login',{title:'Login'});
}).post(function(req,res)
{
    let user={username:req.body.username,password:req.body.password};
    let sqlStr='select count(*) as count from users where username="'+user.username+'" and password="'+user.password+'"';
    let dbhelp=new DBHelp();
    dbhelp.Find(sqlStr,function(result)
    {
        if(result[0].count==1)
        {
            //出于安全，只把包含用户名存入session
            req.session.user={username:user.username};
            return res.redirect('/home');
        }
        else
        {
            req.session.error='用户名不存在或者密码错误！';
            return res.redirect('/login');
        }
    });
});


//注册
router.route('/register').all(Logined).get(function(req,res)
{
    res.render('register',{title:'Register'});
}).post(function(req,res)
{

    let newUser={username:req.body.username,password:req.body.password,passwordSec:req.body.passwordSec};
    let sqlStr2='select count(*) as count from users where username="'+newUser.username+'"';
    let dbhelp=new DBHelp();
    dbhelp.Find(sqlStr2,function(result)
    {
        if(result[0].count!=1)
        {
            if(newUser.password===newUser.passwordSec)
            {
                let sqlStr='insert into users(username,password) values(?,?)';
                let data=[newUser.username,newUser.password]
                dbhelp.Add(sqlStr,data,function()
                {
                    req.session.error='注册成功，请登录！';
                    return res.redirect('/login');
                });
            }
            else
            {
                req.session.error='两次密码不一致！';
                return res.redirect('/register');
            }
        }
        else
        {
            req.session.error='用户名已存在！';
            return res.redirect('/register');
        }
    });
});


//Home用户管理
router.route('/home').all(LoginFirst).get(function(req,res)
{
    let sqlStr='select username from users';
    let dbhelp=new DBHelp();
    dbhelp.Find(sqlStr,function(result)
    {
        if(result)
        {
            res.render('home',{title:'Home',Allusers:result});
        }
        else
        {
            res.render('home',{title:'Home'});
        }
    });
});

//删除
router.route('/delete/:URLusername').get(function(req,res)
{
    let reqUsername=req.params.URLusername;
    if(reqUsername!==req.session.user.username)
    {

        let sqlStr='delete from users where username="'+reqUsername+'"';
        let dbhelp=new DBHelp();
        dbhelp.Delete(sqlStr,function()
        {
            req.session.error='移除用户 '+reqUsername+' 成功！';
            return res.redirect('/home');
        });
    }
    else
    {
        req.session.error="不能操作当前登录用户！";
        return res.redirect('/home');
    }
});

//重置密码
router.get('/resetPwd/:URLusername',function(req,res)
{
    let reqUsername=req.params.URLusername;
    if(reqUsername!==req.session.user.username)
    {
        let sqlStr='update users set password=? where username=?';
        let data=['123456',reqUsername];
        let dbhelp=new DBHelp();
        dbhelp.Update(sqlStr,data,function()
        {
            req.session.error=reqUsername+' 的密码已重置为 123456！';
            return res.redirect('/home');
        });
    }
    else
    {
        req.session.error="不能操作当前登录用户！";
        return res.redirect('/home');
    }
});

//注销
router.get('/logout',function(req,res)
{
    req.session.user=null;
    return res.redirect('/');
});

function Logined(req,res,next)
{
    if(req.session.user)
    {
        req.session.error='您已登录！';
        return res.redirect('/home');
    }
    next();
}

function LoginFirst(req,res,next)
{
    if(!req.session.user)
    {
        req.session.error='请先登录!';
        return res.redirect('/login');
    }
    next();
}

module.exports = router;