微信小程序使用stomp.js实现STOMP传输协议的实时聊天

简介：

uniapp开发的小程序中使用
本来使用websocket，后端同事使用了stomp协议，导致前端也需要对应修改。

如何使用

在static/js中新建stomp.js和WEBSocket.js，然后在需要使用的页面引入监听代码+发送代码即可

代码如下：

位置：项目/pages/static/js/stomp.js
1.stomp.js

// Generated by CoffeeScript 1.7.1   (function() {    var Byte, Client, Frame, Stomp,      __hasProp = {}.hasOwnProperty,      __slice = [].slice;      Byte = {      LF: '\x0A',      NULL: '\x00'    };      Frame = (function() {      var unmarshallSingle;        function Frame(command, headers, body) {        this.command = command;        this.headers = headers != null ? headers : {};        this.body = body != null ? body : '';      }        Frame.prototype.toString = function() {        var lines, name, skipContentLength, value, _ref;        lines = [this.command];        skipContentLength = this.headers['content-length'] === false ? true : false;        if (skipContentLength) {          delete this.headers['content-length'];        }        _ref = this.headers;        for (name in _ref) {          if (!__hasProp.call(_ref, name)) continue;          value = _ref[name];          lines.push("" + name + ":" + value);        }        if (this.body && !skipContentLength) {          lines.push("content-length:" + (Frame.sizeOfUTF8(this.body)));        }        lines.push(Byte.LF + this.body);        return lines.join(Byte.LF);      };        Frame.sizeOfUTF8 = function(s) {        if (s) {          return encodeURI(s).match(/%..|./g).length;        } else {          return 0;        }      };        unmarshallSingle = function(data) {        var body, chr, command, divider, headerLines, headers, i, idx, len, line, start, trim, _i, _j, _len, _ref, _ref1;        divider = data.search(RegExp("" + Byte.LF + Byte.LF));        headerLines = data.substring(0, divider).split(Byte.LF);        command = headerLines.shift();        headers = {};        trim = function(str) {          return str.replace(/^\s+|\s+$/g, '');        };        _ref = headerLines.reverse();        for (_i = 0, _len = _ref.length; _i < _len; _i++) {          line = _ref[_i];          idx = line.indexOf(':');          headers[trim(line.substring(0, idx))] = trim(line.substring(idx + 1));        }        body = '';        start = divider + 2;        if (headers['content-length']) {          len = parseInt(headers['content-length']);          body = ('' + data).substring(start, start + len);        } else {          chr = null;          for (i = _j = start, _ref1 = data.length; start <= _ref1 ? _j < _ref1 : _j > _ref1; i = start <= _ref1 ? ++_j : --_j) {            chr = data.charAt(i);            if (chr === Byte.NULL) {              break;            }            body += chr;          }        }        return new Frame(command, headers, body);      };        Frame.unmarshall = function(datas) {        var frame, frames, last_frame, r;        frames = datas.split(RegExp("" + Byte.NULL + Byte.LF + "*"));        r = {          frames: [],          partial: ''        };        r.frames = (function() {          var _i, _len, _ref, _results;          _ref = frames.slice(0, -1);          _results = [];          for (_i = 0, _len = _ref.length; _i < _len; _i++) {            frame = _ref[_i];            _results.push(unmarshallSingle(frame));          }          return _results;        })();        last_frame = frames.slice(-1)[0];        if (last_frame === Byte.LF || (last_frame.search(RegExp("" + Byte.NULL + Byte.LF + "*$"))) !== -1) {          r.frames.push(unmarshallSingle(last_frame));        } else {          r.partial = last_frame;        }        return r;      };        Frame.marshall = function(command, headers, body) {        var frame;        frame = new Frame(command, headers, body);        return frame.toString() + Byte.NULL;      };        return Frame;      })();      Client = (function() {      var now;        function Client(ws) {        this.ws = ws;        this.ws.binaryType = "arraybuffer";        this.counter = 0;        this.connected = false;        this.heartbeat = {          outGoing: 10000,          incoming: 10000        };        this.maxWebSocketFrameSize = 16 * 1024;        this.subscriptions = {};        this.partialData = '';      }        Client.prototype.debug = function(message) {        var _ref;        return typeof window !== "undefined" && window !== null ? (_ref = window.console) != null ? _ref.log(message) : void 0 : void 0;      };        now = function() {        if (Date.now) {          return Date.now();        } else {          return new Date().valueOf;        }      };        Client.prototype._transmit = function(command, headers, body) {        var out;        out = Frame.marshall(command, headers, body);        if (typeof this.debug === "function") {          this.debug(">>> " + out);        }        while (true) {          if (out.length > this.maxWebSocketFrameSize) {            this.ws.send(out.substring(0, this.maxWebSocketFrameSize));            out = out.substring(this.maxWebSocketFrameSize);            if (typeof this.debug === "function") {              this.debug("remaining = " + out.length);            }          } else {            return this.ws.send(out);          }        }      };        Client.prototype._setupHeartbeat = function(headers) {        var serverIncoming, serverOutgoing, ttl, v, _ref, _ref1;        if ((_ref = headers.version) !== Stomp.VERSIONS.V1_1 && _ref !== Stomp.VERSIONS.V1_2) {          return;        }        _ref1 = (function() {          var _i, _len, _ref1, _results;          _ref1 = headers['heart-beat'].split(",");          _results = [];          for (_i = 0, _len = _ref1.length; _i < _len; _i++) {            v = _ref1[_i];            _results.push(parseInt(v));          }          return _results;        })(), serverOutgoing = _ref1[0], serverIncoming = _ref1[1];        if (!(this.heartbeat.outgoing === 0 || serverIncoming === 0)) {          ttl = Math.max(this.heartbeat.outgoing, serverIncoming);          if (typeof this.debug === "function") {            this.debug("send PING every " + ttl + "ms");          }          this.pinger = Stomp.setInterval(ttl, (function(_this) {            return function() {              _this.ws.send(Byte.LF);              return typeof _this.debug === "function" ? _this.debug(">>> PING") : void 0;            };          })(this));        }        if (!(this.heartbeat.incoming === 0 || serverOutgoing === 0)) {          ttl = Math.max(this.heartbeat.incoming, serverOutgoing);          if (typeof this.debug === "function") {            this.debug("check PONG every " + ttl + "ms");          }          return this.ponger = Stomp.setInterval(ttl, (function(_this) {            return function() {              var delta;              delta = now() - _this.serverActivity;              if (delta > ttl * 2) {                if (typeof _this.debug === "function") {                  _this.debug("did not receive server activity for the last " + delta + "ms");                }                return _this.ws.close();              }            };          })(this));        }      };        Client.prototype._parseConnect = function() {        var args, connectCallback, errorCallback, headers;        args = 1 <= arguments.length ? __slice.call(arguments, 0) : [];        headers = {};        switch (args.length) {          case 2:            headers = args[0], connectCallback = args[1];            break;          case 3:            if (args[1] instanceof Function) {              headers = args[0], connectCallback = args[1], errorCallback = args[2];            } else {              headers.login = args[0], headers.passcode = args[1], connectCallback = args[2];            }            break;          case 4:            headers.login = args[0], headers.passcode = args[1], connectCallback = args[2], errorCallback = args[3];            break;          default:            headers.login = args[0], headers.passcode = args[1], connectCallback = args[2], errorCallback = args[3], headers.host = args[4];        }        return [headers, connectCallback, errorCallback];      };        Client.prototype.connect = function() {        var args, errorCallback, headers, out;        args = 1 <= arguments.length ? __slice.call(arguments, 0) : [];        out = this._parseConnect.apply(this, args);        headers = out[0], this.connectCallback = out[1], errorCallback = out[2];        if (typeof this.debug === "function") {          this.debug("Opening Web Socket...");        }        this.ws.onmessage = (function(_this) {          return function(evt) {            var arr, c, client, data, frame, messageID, onreceive, subscription, unmarshalledData, _i, _len, _ref, _results;            data = typeof ArrayBuffer !== 'undefined' && evt.data instanceof ArrayBuffer ? (arr = new Uint8Array(evt.data), typeof _this.debug === "function" ? _this.debug("--- got data length: " + arr.length) : void 0, ((function() {              var _i, _len, _results;              _results = [];              for (_i = 0, _len = arr.length; _i < _len; _i++) {                c = arr[_i];                _results.push(String.fromCharCode(c));              }              return _results;            })()).join('')) : evt.data;            _this.serverActivity = now();            if (data === Byte.LF) {              if (typeof _this.debug === "function") {                _this.debug("<<< PONG");              }              return;            }            if (typeof _this.debug === "function") {              _this.debug("<<< " + data);            }            unmarshalledData = Frame.unmarshall(_this.partialData + data);            _this.partialData = unmarshalledData.partial;            _ref = unmarshalledData.frames;            _results = [];            for (_i = 0, _len = _ref.length; _i < _len; _i++) {              frame = _ref[_i];              switch (frame.command) {                case "CONNECTED":                  if (typeof _this.debug === "function") {                    _this.debug("connected to server " + frame.headers.server);                  }                  _this.connected = true;                  _this._setupHeartbeat(frame.headers);                  _results.push(typeof _this.connectCallback === "function" ? _this.connectCallback(frame) : void 0);                  break;                case "MESSAGE":                  subscription = frame.headers.subscription;                  onreceive = _this.subscriptions[subscription] || _this.onreceive;                  if (onreceive) {                    client = _this;                    messageID = frame.headers["message-id"];                    frame.ack = function(headers) {                      if (headers == null) {                        headers = {};                      }                      return client.ack(messageID, subscription, headers);                    };                    frame.nack = function(headers) {                      if (headers == null) {                        headers = {};                      }                      return client.nack(messageID, subscription, headers);                    };                    _results.push(onreceive(frame));                  } else {                    _results.push(typeof _this.debug === "function" ? _this.debug("Unhandled received MESSAGE: " + frame) : void 0);                  }                  break;                case "RECEIPT":                  _results.push(typeof _this.onreceipt === "function" ? _this.onreceipt(frame) : void 0);                  break;                case "ERROR":                  _results.push(typeof errorCallback === "function" ? errorCallback(frame) : void 0);                  break;                default:                  _results.push(typeof _this.debug === "function" ? _this.debug("Unhandled frame: " + frame) : void 0);              }            }            return _results;          };        })(this);        this.ws.onclose = (function(_this) {          return function() {            var msg;            msg = "Whoops! Lost connection to " + _this.ws.url;            if (typeof _this.debug === "function") {              _this.debug(msg);            }            _this._cleanUp();            return typeof errorCallback === "function" ? errorCallback(msg) : void 0;          };        })(this);        return this.ws.onopen = (function(_this) {          return function() {            if (typeof _this.debug === "function") {              _this.debug('Web Socket Opened...');            }            headers["accept-version"] = Stomp.VERSIONS.supportedVersions();            headers["heart-beat"] = [_this.heartbeat.outgoing, _this.heartbeat.incoming].join(',');            return _this._transmit("CONNECT", headers);          };        })(this);      };        Client.prototype.disconnect = function(disconnectCallback, headers) {        if (headers == null) {          headers = {};        }        this._transmit("DISCONNECT", headers);        this.ws.onclose = null;        this.ws.close();        this._cleanUp();        return typeof disconnectCallback === "function" ? disconnectCallback() : void 0;      };        Client.prototype._cleanUp = function() {        this.connected = false;        if (this.pinger) {          Stomp.clearInterval(this.pinger);        }        if (this.ponger) {          return Stomp.clearInterval(this.ponger);        }      };        Client.prototype.send = function(destination, headers, body) {        if (headers == null) {          headers = {};        }        if (body == null) {          body = '';        }        headers.destination = destination;        return this._transmit("SEND", headers, body);      };        Client.prototype.subscribe = function(destination, callback, headers) {        var client;        if (headers == null) {          headers = {};        }        if (!headers.id) {          headers.id = "sub-" + this.counter++;        }        headers.destination = destination;        this.subscriptions[headers.id] = callback;        this._transmit("SUBSCRIBE", headers);        client = this;        return {          id: headers.id,          unsubscribe: function() {            return client.unsubscribe(headers.id);          }        };      };        Client.prototype.unsubscribe = function(id) {        delete this.subscriptions[id];        return this._transmit("UNSUBSCRIBE", {          id: id        });      };        Client.prototype.begin = function(transaction) {        var client, txid;        txid = transaction || "tx-" + this.counter++;        this._transmit("BEGIN", {          transaction: txid        });        client = this;        return {          id: txid,          commit: function() {            return client.commit(txid);          },          abort: function() {            return client.abort(txid);          }        };      };        Client.prototype.commit = function(transaction) {        return this._transmit("COMMIT", {          transaction: transaction        });      };        Client.prototype.abort = function(transaction) {        return this._transmit("ABORT", {          transaction: transaction        });      };        Client.prototype.ack = function(messageID, subscription, headers) {        if (headers == null) {          headers = {};        }        headers["message-id"] = messageID;        headers.subscription = subscription;        return this._transmit("ACK", headers);      };        Client.prototype.nack = function(messageID, subscription, headers) {        if (headers == null) {          headers = {};        }        headers["message-id"] = messageID;        headers.subscription = subscription;        return this._transmit("NACK", headers);      };        return Client;      })();      Stomp = {      VERSIONS: {        V1_0: '1.0',        V1_1: '1.1',        V1_2: '1.2',        supportedVersions: function() {          return '1.1,1.0';        }      },      client: function(url, protocols) {        var klass, ws;        if (protocols == null) {          protocols = ['v10.stomp', 'v11.stomp'];        }        klass = Stomp.WebSocketClass || WebSocket;        ws = new klass(url, protocols);        return new Client(ws);      },      over: function(ws) {        return new Client(ws);      },      Frame: Frame    };      if (typeof exports !== "undefined" && exports !== null) {      exports.Stomp = Stomp;    }      if (typeof window !== "undefined" && window !== null) {      Stomp.setInterval = function(interval, f) {        return window.setInterval(f, interval);      };      Stomp.clearInterval = function(id) {        return window.clearInterval(id);      };      window.Stomp = Stomp;    } else if (!exports) {      self.Stomp = Stomp;    }    }).call(this);

位置：项目/pages/static/js/websocket.js
2.websocket.js

const Stomp = require('./stomp.js').Stomp;let socketOpen = falselet socketMsgQueue = []export default {  client: null,  init(url, header ,connectWS) {    if (this.client) {      return Promise.resolve(this.client)    }    return new Promise((resolve, reject) => {      const ws = {        send: this.sendMessage,        onopen: null,        onmessage: null,        close: this.closeSocket,      }      wx.connectSocket({ url, header })      wx.onSocketOpen(function (res) {        console.log('WebSocket连接已打开！', res)        socketOpen = true        for (let i = 0; i < socketMsgQueue.length; i++) {          ws.send(socketMsgQueue[i])        }        socketMsgQueue = []        ws.onopen && ws.onopen()      })      wx.onSocketMessage(function (res) {        ws.onmessage && ws.onmessage(res)      })      wx.onSocketError(function (res) {        console.log('WebSocket 错误！', res)      })      wx.onSocketClose((res) => {        this.client.disconnect()        this.client = null        socketOpen = false        console.log('WebSocket 已关闭！', res)        setTimeout(()=>{          connectWS()        },3000)      })      Stomp.setInterval = function (interval, f) {        return setInterval(f, interval)      }      Stomp.clearInterval = function (id) {        return clearInterval(id)      }      const client = (this.client = Stomp.over(ws))      client.connect(header, function () {        console.log('stomp connected')        resolve(client)      })    })  },  disconnect() {    wx.closeSocket()  },  sendMessage(message) {    if (socketOpen) {      wx.sendSocketMessage({        data: message,      })    } else {      socketMsgQueue.push(message)    }  },  closeSocket() {    console.log('closeSocket')  },}

监听+发送代码

import WebSocket from "../../static/js/websocket"const app = getApp();data： {objUid: '1',client: null,content: '发送的内容'},onLoad(options) {// stomp协议请求 this.initWS()},initWS() {WebSocket.init(`${app.globalData.WSURL}/chat`,// 传参{// login: 'admin',// passcode: 'admin',},// ws断开回调() => {this.initWS()}).then((client) => {this.setData({client: client})// 订阅client.subscribe(// 路径`/response/${app.globalData.uid}/${this.data.objUid}`,// 接收到的数据(res) => {console.log(res)},// 消息不会被确认接收，不确认每次连接都会推送// { ack: 'client' } )})},// 直接调用发送即可send() {this.data.client.send(// 路径`/child/${app.globalData.uid}/${this.data.objUid}`,// 自定义参数 Http://jmesnil.net/stomp-websocket/doc/{},//priority: 9 // 发送文本JSON.stringify({ 'content': this.data.content }));},

来源地址：https://blog.csdn.net/qq_25252769/article/details/132560764