在线直播源码搭建直播平台的后端

 在线直播源码搭建直播平台的后端

后端项目初始化

1.全局安装express脚手架

额，这个应该是属于准备工作的。给忘记了，那就凑合放在这里吧，别打我，我知道错了，但我就是不改【狗头保命】

cnpm install -g express-generator 
复制代码

2.新建工程

打开cmd，cd 到大家喜欢的路径下，然后输入下面这条命令

express express_mylive
复制代码

 

 

 

接着我们使用编辑器打开我们刚刚创建好的项目。像小羽这么帅的男人，当然就是眼疾手快的打开了vs code啦。这就是我们刚刚创建的工程啦。下图就是我们的项目目录。

 

 

 

在vscode的cmd中，通过cnpm install 安装相关的模块。接着通过npm run start 启动后台。express框架默认是启动3000端口，所以老铁们，不用我多说了吧！浏览器输入http://localhost:3000。

 

 

 

3.初始化工程

但是这个工程还配不上像咱们老铁那高贵的气质。所以，痛定思痛，老铁们你们还不一起来改造这个气质低下的工程嘛？

3.1 修改package.json

修改package.json，如下。然后在vscode的cmd中输入以下两条命令，全局安装相关的插件模块和pm2。

cnpm install 
cnpm install pm2 -g
复制代码

{
  "name": "express-mylive",
  "version": "0.0.0",
  "private": true,
  "scripts": {
    "dev": "node-dev ./bin/www",
    "start": "pm2 start ./bin/www --name=express_living",
    "restart": "pm2 restart ./bin/www --name=express_living",
    "stop": "pm2 stop express_living"
  },
  "dependencies": {
    "base64-img": "^1.0.4",
    "body-parser": "^1.19.0",
    "compression": "^1.7.4",
    "cookie-parser": "~1.4.4",
    "debug": "~2.6.9",
    "express": "~4.16.1",
    "express-jwt": "^6.0.0",
    "http-errors": "~1.6.3",
    "jade": "~1.11.0",
    "morgan": "~1.9.1",
    "mysql": "^2.18.1",
    "node-media-server": "^2.1.4",
    "socket.io": "^2.3.0"
  },
  "devDependencies": {
    "node-dev": "^5.0.0"
  }
}

复制代码

3.2 修改启动端口

修改bin目录下的www文件，修改端口号为8512。老铁们喜欢改成啥都可以哈，尽量找一个少用的端口号就行了。保存之后，直接打开http://localhost:8512就可以看到我 们的后端重新运行到了8512端口，而原来的3000端口无法打开了，就好像小三一样被无情的抛弃了。呸，一群渣男！

 

 

 

3.3 配置公共文件

修改public目录下的文件如下

 

 

 

common.js

const base64Img = require('base64-img');
class Common {
    //格式化输出
    outPut(code,data,msg=""){
        return {code:code,data:data,msg:msg}
    }
    //生成len位随机字符串
    getCode(len){
        var chars = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z'];
        var nums = "";
        for (var i = 0; i < len; i++) {
        var id = parseInt(Math.random() * 62);
        nums += chars[id];
        }
        return nums;
    }
    //图片转base64
    async base64Image(url){
        return new Promise((resolve,reject)=>{
            base64Img.requestBase64(url,(err,res,body)=>{
                resolve(body)
            })
        }) 
    }
}
const common = new Common();
module.exports = common;
复制代码

mysqlModal.js

这里的话。我们使用了db2的配置，所以老铁们改db2的相关配置就阔以啦~

//mysqlModal.js
//初始化数据库配置
//test666
const mysqlConf = {
    db1:{
        host: '127.0.0.1',
        user: 'root',
        password: 'test',
        database: 'test',
        port: 3306
    },
    db2:{
        host: 'localhost',
        user: 'root',
        password: '123456',
        database: 'living',
        port: 3306
    }
};

//封装数据库sql操作
async function DB1(sql){
    var mysql = require('mysql');
    var pool = mysql.createPool(mysqlConf.db1);
    return new Promise((resolve,reject)=>{
        pool.getConnection(function(err,conn){
            if(err){
                console.log(err)
            }else{
                conn.query(sql,function(err,results,fields){
                    //事件驱动回调
                    if(err){
                        resolve(err)
                    }else{
                        resolve(results)
                    }
                });
                //释放连接，需要注意的是连接释放需要在此处释放，而不是在查询回调里面释放
                conn.release();
            }
        });
    })
}
async function DB2(sql){
    var mysql = require('mysql');
    var pool = mysql.createPool(mysqlConf.db2);
    return new Promise((resolve,reject)=>{
        pool.getConnection(function(err,conn){
            if(err){
                console.log(err)
            }else{
                conn.query(sql,function(err,results,fields){
                    //事件驱动回调
                    if(err){
                        resolve(err)
                    }else{
                        resolve(results)
                    }
                });
                //释放连接，需要注意的是连接释放需要在此处释放，而不是在查询回调里面释放
                conn.release();
            }
        });
    })
}



module.exports = {DB1,DB2};
复制代码

4.导入数据库

像小羽这么聪明的男人，早就知道你们这群人懒得要死，所以早早就给你们准备好了sql文件。你们说说看嘛，要是没了我这么帅气的男人，你们怎么活呀？

使用navicate连接上自己的数据库，创建一个叫living的库，然后导入上面说到的sql文件。最后可以看到帮我们新建了两个表。

 

 

 

 

 

 

5.修改api接口

 

 

 

5.1 index

routes文件夹下的index.js

这里是我们的默认请求的接口，也就是传说中的牌面，所以老铁们，不要客气有多高大上就给它整到多高大上。

var express = require('express');
var router = express.Router();

/* GET home page. */
router.get('/', function(req, res, next) {
  //res.render('index', { title: 'Express' });
  res.send("小羽直播间-后台")
});

module.exports = router;

复制代码

5.2 users

router文件夹下的users.js

这里是我们的用户接口，都是用sql语句写的，不会orm，后端大佬不要埋汰小弟，小弟夸您帅好不好？

var express = require('express');
var router = express.Router();
var commonJS = require('../public/js/common');
var sqlHandle = require('../public/config/mysqlModal')
const jwt = require('jsonwebtoken')

router.get("/test",async(req,res,next)=>{
  let img = await commonJS.base64Image("https://c-ssl.duitang.com/uploads/item/201807/08/20180708222217_wayur.jpeg")
  res.send(img)
})


/**
 * @description: 用户登录
 * @Date: 2020-09-01 10:46:49
 * @author: 小羽
 * @param {type} 
 * @return {type} 
 */
router.post("/login",async(req,res,next)=>{
  let data = req.body
  if(data.account&&data.password){
    let sql = `select * from user where email='${data.account}' and password='${data.password}' limit 1`
    let result = await sqlHandle.DB2(sql)
    if (result.length == 1) {
      const token = 'Bearer ' + jwt.sign(
        {
          id:result[0].id,
          name:result[0].name,
          age:result[0].age,
          email:result[0].email
        },
        'living_xiaoyu',
        {
          expiresIn: 3600 * 24 * 7
        }
      )
      
      res.send(commonJS.outPut(200, token, 'success'))
    }else{
      res.send(commonJS.outPut(500, "账号/密码错误", 'fail'))
    }
  }else{
    res.send(commonJS.outPut(500, "账号/密码不能为空", 'fail'))
  }
})


/**
 * @description: 获取用户信息
 * @Date: 2020-08-31 11:13:14
 * @author: 小羽
 * @param {type} 
 * @return {type} 
 */
router.get('/getUserInfo',async(req,res,next)=>{
  let data = req.query;
  let sql = `select * from user where id = '${data.id}' limit 1`
  let result = await sqlHandle.DB2(sql)
  if (result.length == 1) {
    res.send(commonJS.outPut(200, result[0], 'success'))
  }else{
    res.send(commonJS.outPut(500, result, 'fail'))
  }
})


/**
 * @description: 新增用户
 * @Date: 2020-08-31 11:13:05
 * @author: 小羽
 * @param {type} 
 * @return {type} 
 */
router.post('/addUser', async (req, res, next)=>{
  let data = req.body;
  if(!data.email||!data.password){
    res.send(commonJS.outPut(500, "邮箱/密码不能为空", 'fail'))
    return
  }
  let sql = `insert into user (id,name,age,email,password) values ('${commonJS.getCode(32)}','${data.name}','${data.age}','${data.email}','${data.password}')`
  let result = await sqlHandle.DB2(sql)
  if (result.affectedRows == 1) {
    res.send(commonJS.outPut(200, data, 'success'))
  }else{
    res.send(commonJS.outPut(500, result, 'fail'))
  }
});


/**
 * @description: 编辑用户信息
 * @Date: 2020-08-31 13:38:44
 * @author: 小羽
 * @param {type} 
 * @return {type} 
 */
router.post("/editUser",async(req,res,next)=>{
  let data = req.data
  if(!data.id){
    let sql = `update user set name=${data.name},age=${data.age},password=${data.password} where id=${data.id}`
    let result = await sqlHandle.DB2(sql)
    if (result.affectedRows == 1) {
      res.send(commonJS.outPut(200, data, 'success'))
    }else{
      res.send(commonJS.outPut(500, result, 'fail'))
    }
  }
})

module.exports = router;

复制代码

5.3 livingRoom

router文件夹下的livingRoom.js

直播间接口，也是用sql语句写的。

/*
 * @description: 
 * @author: 小羽
 * @github: https://github.com/lyff1006
 * @lastEditors: 小羽
 * @Date: 2020-08-31 22:27:40
 * @LastEditTime: 2020-09-05 17:58:10
 * @Copyright: 1.0.0
 */
var express = require('express');
var router = express.Router();
var commonJS = require('../public/js/common');
var sqlHandle = require('../public/config/mysqlModal')

/**
 * @description: 获取直播间信息
 * @Date: 2020-08-31 11:32:31
 * @author: 小羽
 * @param {type} 
 * @return {type} 
 */
router.get("/roomList",async (req,res,next)=>{
    let data = req.query
    let sql
    if(!data.keyword){
        sql = `select living_room.id,living_room.user_id,living_room.title,user.name,living_room.image,user.avatar,living_room.type from living_room left join user on living_room.user_id = user.id  where living_room.status != 0`
    }else{
        sql = `select living_room.id,living_room.user_id,living_room.title,user.name,living_room.image,user.avatar,living_room.type from living_room left join user on living_room.user_id = user.id where title like '%${data.keyword}%' or user.name like '%${data.keyword}%' and living_room.status !=0 limit 20`
    }
    let result = await sqlHandle.DB2(sql)
    if (result.length >= 0) {
        res.send(commonJS.outPut(200, result, 'success'))
    }else{
        res.send(commonJS.outPut(500, result, 'fail'))
    }
})

/**
 * @description: 根据类型获取直播间信息
 * @Date: 2020-08-31 11:32:31
 * @author: 小羽
 * @param {type} 
 * @return {type} 
 */
router.get("/roomListByType",async (req,res,next)=>{
    let data = req.query
    let sql
    if(!data.type){
        sql = `select living_room.id,living_room.user_id,living_room.title,user.name,living_room.image,user.avatar,living_room.type from living_room left join user on living_room.user_id = user.id  where living_room.status != 0`
    }else{
        sql = `select living_room.id,living_room.user_id,living_room.title,user.name,living_room.image,user.avatar,living_room.type from living_room left join user on living_room.user_id = user.id where type = '${data.type}' and living_room.status !=0 limit 20`
    }
    let result = await sqlHandle.DB2(sql)
    if (result.length >= 0) {
        res.send(commonJS.outPut(200, result, 'success'))
    }else{
        res.send(commonJS.outPut(500, result, 'fail'))
    }
})


/**
 * @description: 新建房间
 * @Date: 2020-08-31 11:38:49
 * @author: 小羽
 * @param {type} 
 * @return {type} 
 */
router.post("/addRoom",async (req,res,next)=>{
    let data = req.body
    let sql = `insert into living_room (id,title,user_id,type) value ('${commonJS.getCode(32)}','${data.title}','${data.user_id}','${data.type}')`
    let result = await sqlHandle.DB2(sql)
    if (result.affectedRows == 1) {
        res.send(commonJS.outPut(200, data, 'success'))
    }else{
        res.send(commonJS.outPut(500, result, 'fail'))
    }
})


/**
 * @description: 编辑房间
 * @Date: 2020-08-31 11:40:27
 * @author: 小羽
 * @param {type} 
 * @return {type} 
 */
router.post("/editRoom",async (req,res,next)=>{
    let data = req.body
    let sql = `update living_room set title='${data.title}',status='${data.status}',user_id='${data.user_id}' where id ='${data.id}' `
    let result = await sqlHandle.DB2(sql)
    if (result.affectedRows == 1) {
        res.send(commonJS.outPut(200, data, 'success'))
    }else{
        res.send(commonJS.outPut(500, result, 'fail'))
    }
})

/**
 * @description: 获取直播间详情
 * @Date: 2020-09-10 22:04:29
 * @author: 小羽
 * @param {type} 
 * @return {type} 
 */
router.get("/roomDetail",async (req,res,next)=>{
    let data = req.query
    let sql = `select living_room.title,living_room.type,user.name,user.id,user.avatar from living_room left join user on living_room.user_id = user.id  where living_room.id = '${data.id}' and living_room.status != 0`
    let result = await sqlHandle.DB2(sql)
    if(result.length==1){
        let resultData = {
            ...result[0],
            room_id: data.id
        }
        res.send(commonJS.outPut(200, resultData, 'success'))
    }else{
        res.send(commonJS.outPut(500, result, 'fail'))
    }
})

module.exports = router;

5.4 修改app.js

5.4.1 添加路由

//路由相关
app.use('/', indexRouter);
app.use('/users', usersRouter);
app.use("/livingRoom", roomRouter);
复制代码

5.4.2 开启gzip压缩

//开启gzip
var compression = require('compression');
app.use(compression());
复制代码

5.4.3 设置跨域头

//设置跨域头，这里设置全部允许跨域了
app.all('*', function (req, res, next) {
  res.header('Access-Control-Allow-Origin', req.headers.origin || '*');
  res.header('Access-Control-Allow-Headers', 'Content-Type,Content-Length, Authorization,Origin,Accept,X-Requested-With');
  res.header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS, PUT, PATCH, DELETE');
  res.header('Access-Control-Allow-Credentials', true);
  res.header('X-Powered-By', ' 3.2.1');
  res.header('Content-Type', 'application/json;charset=utf-8');
  next();
});
复制代码

5.4.4 设置jwt

app.use(expressJwt({
  secret:"living_xiaoyu",
  algorithms:['HS256'],
  credentialsRequired:true, //是否校验
}).unless({
  path:['/users/login','/livingRoom/roomList','/livingRoom/roomListByType']
}))
复制代码

5.4.5 设置完的app.js

var createError = require('http-errors');
var express = require('express');
var path = require('path');
var cookieParser = require('cookie-parser');
var logger = require('morgan');
var expressJwt = require("express-jwt")
var compression = require('compression');


var indexRouter = require('./routes/index');
var usersRouter = require('./routes/users');
var roomRouter = require('./routes/livingRoom');

var app = express();

//开启gzip
app.use(compression());

//设置跨域头
app.all('*', function (req, res, next) {
  res.header('Access-Control-Allow-Origin', req.headers.origin || '*');
  res.header('Access-Control-Allow-Headers', 'Content-Type,Content-Length, Authorization,Origin,Accept,X-Requested-With');
  res.header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS, PUT, PATCH, DELETE');
  res.header('Access-Control-Allow-Credentials', true);
  res.header('X-Powered-By', ' 3.2.1');
  res.header('Content-Type', 'application/json;charset=utf-8');
  next();
});

// view engine setup
app.set('views', path.join(__dirname, 'views'));
app.set('view engine', 'jade');

//使用中间件
app.use(logger('dev'));
app.use(express.json());
app.use(express.urlencoded({ extended: false }));
app.use(cookieParser());
app.use(express.static(path.join(__dirname, 'public')));
app.use(expressJwt({
  secret:"living_xiaoyu",
  algorithms:['HS256'],
  credentialsRequired:true, //是否校验
}).unless({
  path:['/users/login','/livingRoom/roomList','/livingRoom/roomListByType']
}))

//路由相关
app.use('/', indexRouter);
app.use('/users', usersRouter);
app.use("/livingRoom", roomRouter);

// catch 404 and forward to error handler
app.use(function(req, res, next) {
  next(createError(404));
});

// error handler
app.use(function(err, req, res, next) {
  // set locals, only providing error in development
  res.locals.message = err.message;
  res.locals.error = req.app.get('env') === 'development' ? err : {};

  // render the error page
  res.status(err.status || 500);
  res.render('error');
});

module.exports = app;
复制代码

6.使用postman测试api接口

6.1 设置全局变量

 

 

 

6.2 将api接口导入postman

嗯，不用你们提醒，我也给你们准备好了，拿走吧www.getpostman.com/collections…

 

 

 

6.3 设置token

这里可以设置token哦，token可以从登录接口中获取，啥？账号密码？自己进数据看看就知道了嘛。敲黑板，敲黑板，敲黑板，token 中不需要带Bearer

 

 

 

 

7.添加websocket和rtmp服务

修改app.js

var createError = require('http-errors');
var express = require('express');
var path = require('path');
var cookieParser = require('cookie-parser');
var logger = require('morgan');
var expressJwt = require("express-jwt")
var compression = require('compression');
var indexRouter = require('./routes/index');
var usersRouter = require('./routes/users');
var roomRouter = require('./routes/livingRoom');
var http = require('http').Server(app);
var io = require('socket.io')(http);
var port = process.env.PORT || 8511;

var app = express();

//开启gzip
app.use(compression());

//设置跨域头
app.all('*', function (req, res, next) {
  res.header('Access-Control-Allow-Origin', req.headers.origin || '*');
  res.header('Access-Control-Allow-Headers', 'Content-Type,Content-Length, Authorization,Origin,Accept,X-Requested-With');
  res.header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS, PUT, PATCH, DELETE');
  res.header('Access-Control-Allow-Credentials', true);
  res.header('X-Powered-By', ' 3.2.1');
  res.header('Content-Type', 'application/json;charset=utf-8');
  next();
});

// view engine setup
app.set('views', path.join(__dirname, 'views'));
app.set('view engine', 'jade');

//使用中间件
app.use(logger('dev'));
app.use(express.json());
app.use(express.urlencoded({ extended: false }));
app.use(cookieParser());
app.use(express.static(path.join(__dirname, 'public')));
app.use(expressJwt({
  secret:"living_xiaoyu",
  algorithms:['HS256'],
  credentialsRequired:true, //是否校验
}).unless({
  path:['/users/login','/livingRoom/roomList','/livingRoom/roomListByType']
}))

//路由相关
app.use('/', indexRouter);
app.use('/users', usersRouter);
app.use("/livingRoom", roomRouter);

// catch 404 and forward to error handler
app.use(function(req, res, next) {
  next(createError(404));
});

// error handler
app.use(function(err, req, res, next) {
  // set locals, only providing error in development
  res.locals.message = err.message;
  res.locals.error = req.app.get('env') === 'development' ? err : {};

  // render the error page
  res.status(err.status || 500);
  res.render('error');
});


//rtmp服务
const NodeMediaServer = require('node-media-server');
const config = {
  rtmp: {
    port: 1935,
    chunk_size: 60000,
    gop_cache: true,
    ping: 30,
    ping_timeout: 60
  },
  http: {
    port: 8000,
    allow_origin: '*'
  }
};
var nms = new NodeMediaServer(config)
nms.run();


//socket服务
require('events').EventEmitter.prototype._maxListeners = 1000;
let namespaceList = [{name:"barrage",description:"弹幕模块"}]
for(let i in namespaceList){
  if(namespaceList[i].name){
    newNamespace(namespaceList[i].name)
  }
}
//监听新的命名空间
function newNamespace(space){
  io.of(space).on('connection',(socket)=>{
    socket.on('chat message',(data)=>{
      socket.emit('chat message',data)
    })
    //加入房间
    socket.on('JOINROOM',(name)=>{
      socket.join(name)
    })
    //离开房间
    socket.on('LEAVEROOM',(name)=>{
      socket.leave(name)
    })
    //发送房间聊天
    socket.on('ROOMCHAT',(data)=>{
      io.of(space).to(data.room).emit(data.func,data.data)
    })
    //加入自己的房间
    socket.on('JOINUSER',(id)=>{
      socket.join(id)
    })
    //发送用户私聊
    socket.on('USETCHAT',(data)=>{
      io.of(space).to(data.id).emit(data.func,data.data)
    })
    //发送公聊
    socket.on('COMMONCHAT',(data)=>{
      socket.emit(data.func,data.data)
    })
  })
}
http.listen(port, function(){
  console.log('listening on *:' + port);
});

module.exports = app;

复制代码

websocket服务就暂时不测试了，到时候接入前端再测，现在测一下rtmp服务。

obs中设置推流如下，其中密钥为直播间的id可以在数据库中找到，然后添加窗口捕获后，点击开始推流，在http:localhost:8000/admin中即可看到自己的直播啦。

 

 

 

 

 在线直播源码搭建直播平台的后端本文转载自网络，感谢（小羽曜上进）的分享，转载仅为分享干货知识，如有侵权欢迎联系云豹科技进行删除处理