kefu/io/io.js

/*
*介绍：socket.io 功能封装
*作者：TaiGuangYin
*时间：2017-09-09
* */
var redis = require('../utils/redis');
var msgType = require('./messageTpye');
var ioSvc = require('./ioHelper').ioSvc;
var AppConfig = require('../config');
var Common = require('../utils/common');
var msgModel = require('../model/message');
const sessoionModel = require('../model/session');

//服务端连接
function ioServer(io) {

    var _self = this;
    ioSvc.setInstance(io);

    var __uuids = [];

    //初始化连接人数
    redis.set('online_count', 0, null, function (err, ret) {
        if (err) {
            console.error(err);
        }
    });

    Array.prototype.remove = function (val) {
        var index = this.indexOf(val);
        if (index > -1) {
            this.splice(index, 1);
        }
    };

    io.on('connection', async function (socket) {
        console.log('有新的连接进来了:', socket.id);

        //用户与Socket进行绑定
        socket.on('login', async function (msg) {
            var uid = msg.uid;
            socket._user = msg;
            let type = msg.type; // 获取用户类型
            console.log(type, uid, '登录成功');

            //如果不是客服登录
            if (type != 'kefu') {
                let gongHao = msg.gongHao;
                console.log('new customer login process gongHao=>', gongHao)
                // 获取管理员的socket
                try {
                    //获取管理员 socket
                    let kefuData = await sessoionModel.find(gongHao);
                    if (kefuData) { // 找到客服数据
                        let location = Common.getIpLocation(msg.ip);
                        // let socket = socket.id;
                        // let type = 'customer';
                        // let kefu_id = gongHao;

                        let clientInfo = {
                            "uid": uid,
                            "name": location + ' 客户',
                            "type": 'online'
                        };
                        // 添加客户到 对应的客服
                        await sessoionModel.createOrUpdate({uid}, {
                            uid,
                            socket: socket.id,
                            type: 'customer',
                            kefu_id: gongHao,
                            status: 1,
                            nickname: clientInfo.name,
                        })
                        // 给管理员发送通知
                        io.to(kefuData.socket).emit('update-users', clientInfo);
                    }
                } catch (e) {
                    //TODO 失败重发机制
                    console.log('给管理员发送通知失败', e);
                }

            }
            else {
                try {
                    // 保存客服socket
                    sessoionModel.createOrUpdate({uid}, {
                        uid,
                        socket: socket.id,
                        type: 'kefu',
                        nickname: '客服 ' + uid,
                        status: 1
                    })
                } catch (e) {
                    console.log('客服' + uid + '登录失败了', e);
                }
            }
        });

        //断开事件
        socket.on('disconnect', async function () {
            console.log("与服务其断开");

            // _self.updateOnlineCount(false);
            let user = socket['_user'];
            if (user) {
                // 更新用户状态
                await sessoionModel.update(user.uid, {status: 0});
                if (user.type == 'customer') {
                    // 查找对应kf通知下线操作
                    let kf_data = await sessoionModel.find(user.kefu_id);
                    if (kf_data && kf_data.status == 1) { // 客服在线才通知哟

                        var info = {
                            "uid": user.uid,
                            "name": '客户下线',
                            "type": 'offline'
                        };
                        io.to(kf_data.socket).emit('update-users', info);
                    }
                }
            }


        });

        //重连事件
        socket.on('reconnect', function () {
            console.log("重新连接到服务器");
        });

        //监听客户端发送的信息,实现消息转发到各个其他客户端
        socket.on('message', async function (msg) {
            let userData = socket['_user'];
            if (!userData) {
                return;
            }
            //保存消息到数据库
            msgModel.add(msg.from_uid, msg.uid, msg.content, msg.chat_type, msg.image, function (err) {
                if (err) {
                    console.error(err);
                }
            });
            const sendMsg = {
                "uid": msg.from_uid,
                "content": msg.content,
                "chat_type": msg.chat_type ? msg.chat_type : 'text',
                "image": msg.image
            };
            if (msg.type == msgType.messageType.public) { // 广播
                socket.broadcast.emit("message", sendMsg);
            } else if (msg.type == msgType.messageType.private) {
                let uid = msg.uid; // 对方uid
                console.log('find kefu data', msg)
                // 暂时不 await
                sessoionModel.find(uid).then(toUser => {
                    console.log('find kefu data', toUser)
                    if (toUser && toUser.status == 1) { // 对方在线才发送消息哟
                        io.to(toUser.socket).emit('message', sendMsg);
                    }
                }).catch(e => {
                    console.log('find kefu data exception', e)
                });

            }

        });
    });

    this.updateOnlineCount = function (isConnect) {
        //记录在线客户连接数
        redis.get('online_count', function (err, val) {
            if (err) {
                console.error(err);
            }
            if (!val) {
                val = 0;
            }
            if (typeof val == 'string') {
                val = parseInt(val);
            }
            if (isConnect) {
                val += 1;
            } else {
                val -= 1;
                if (val <= 0) {
                    val = 0;
                }
            }

            console.log('当前在线人数：' + val);
            io.sockets.emit('update_online_count', {online_count: val});

            redis.set('online_count', val, null, function (err, ret) {
                if (err) {
                    console.error(err);
                }
            });
        });
    };

}


//模块导出
exports.ioServer = ioServer;