socket.js 2.2 KB
import watcher from "element-ui/packages/table/src/store/watcher";

var websock = null;
var global_callback = null;
var serverPort = '9001';   //webSocket连接端口
var sid = sessionStorage.getItem('token');
var socketMessage;
var socketStatus;
var socketDataList;
var msgJson;

function getWebIP() {
    var curIP = window.location.hostname;
    return curIP;
}

//初始化weosocket
function initWebSocket() {
    var sid = sessionStorage.getItem('token');
    //ws地址
    var wsuri = "ws://" + getWebIP() + ":" + serverPort + "/websocket/" + sid;
    websock = new WebSocket(wsuri);
    websock.onmessage = function (e) {
        websocketonmessage(e);
    }
    websock.onclose = function (e) {
        websocketclose(e);
    }
    websock.onopen = function (e) {
        websocketOpen(e);
    }

    //连接发生错误的回调方法
    websock.onerror = function (e) {
        console.log("WebSocket连接发生错误");
    }
}

// 实际调用的方法
function sendSock(agentData, callback) {
    global_callback = callback;
    if (websock.readyState === websock.OPEN) {
        //若是ws开启状态
        websocketsend(agentData)
    } else if (websock.readyState === websock.CONNECTING) {
        // 若是 正在开启状态,则等待1s后重新调用
        setTimeout(function () {
            sendSock(agentData, callback);
        }, 1000);
    } else {
        // 若未开启 ,则等待1s后重新调用
        setTimeout(function () {
            sendSock(agentData, callback);
        }, 1000);
    }
}

//数据接收
function websocketonmessage(e) {

    // console.log("------>"+e.data);
    msgJson = JSON.parse(e.data);
    socketMessage = msgJson.message;
    console.log(socketMessage)
    socketStatus = msgJson.status;
    socketDataList = msgJson.data;
    console.log(msgJson);
    // global_callback(JSON.parse(e.data));

}

//数据发送
function websocketsend(agentData) {
    websock.send(JSON.stringify(agentData));
}

//关闭
function websocketclose(e) {
    console.log("connection closed");
}

function websocketOpen(e) {
    console.log("连接成功" + sid);
}

export { initWebSocket,websocketclose,socketMessage, socketStatus, socketDataList,msgJson}