使用http维持socket长连接

项目中有遇到问题如下:

1、旧版的cs服务,因为每个用户和唯一的长连接是在登录后绑定的,并且所有的消息报文均是基于该长连接去发送接收的,所以要求node服务要维持一个长连接,然后根据该用户获取长连接,拿到连接再去发送对应请求,tcp协议顶层是可以使用http传输的,nodejs中http模块内置的agent对象,便可以设置keepalive的方式维持这种长连接,具体方式如下:

module.exports.httpPost = ({ options, ctx }) => {
  return new Promise((resolve, reject) => {
    if (!ctx.users[ctx.query.uid + '']) ctx.users['' + ctx.query.uid] = {}
    let agent = ctx.users['' + ctx.query.uid].agent
    if (!agent) {
      const newAgent = new http.Agent({ keepAlive: true, maxSockets: 1 })
      newAgent.uid = ctx.query.uid      // 人为给agent对象关联uid
      newAgent.createConnection({
        port: config.BC_PORT,
        host: config.BC_SERVER
      }, () => {
        if (!_.isEmpty(newAgent.sockets)) {
          _.forIn(newAgent.sockets, (item, key) => {
            if (item.length) {
              item[0].on('close', () => {
                // 清除当前用户的agent
                ctx.users['' + newAgent.uid].agent = null   // 清除当前用户的agent
                newAgent.destroy()      // 回收agent
              }).on('data', data => {
              })
            }
          })
        }
      })
      agent = newAgent
    }
    const buf = new Buffer(ctx.request.body)
    const httpOptions = {
      headers: {
        'Content-Type': 'text/xml;charset=utf-8',
        'Content-length': buf.length
      },
      host: config.BC_SERVER, port: config.BC_PORT, method: 'POST', agent: agent
    }
    _.merge(httpOptions, options)
    const req = http.request(httpOptions, res => {
      let buffers = ''
      res.on('data', data => {
        buffers += data
      })
      res.on('end', () => {
        // 更新全局用户关联的长连接代理类agent,保证用户和链接一一对应
        ctx.users['' + ctx.query.uid].agent = agent
        resolve(buffers.toString())
      })
    })
    req.write(buf)
    req.end()
    req.on('error', err => {
      reject(err)
    })
  })
}

2、早期的服务是基于xml rpc的,但是浏览器的bs应用直接使用json格式最好不过,所以在请求前后可以进行xml转json的转换,在代码层面只需要加一个中间价即可。

3、早期的cs项目的服务有提供报警模块,采用UDP协议进行点对点发送,浏览器端不可能建立UDP监听吧,为了接收报警数据又不改动原有cs服务的原则,可以在node服务层开启websocket服务,浏览器端负责链接该服务,同时node服务作为udp的client去连接报警的udp服务,获取消息再进行ws转发即可解决。

总结:

1、遇到问题多思考,能通过转发解决的问题就不要通过改写接口去解决(确保旧版服务不变动的原则)

2、socket通讯模块,代码写的时候尽量多考虑一些极端情况,比如链接丢失、用户下线、服务挂了的情况,然后再做相应处理

3、遇到问题还是要多思考

原文地址:https://www.cnblogs.com/vipzhou/p/7412656.html