uni-app封装WebSocket

utils           
├ config.js      
├ websocket.js

import {
	getToken
} from "./token"

let localServer = "127.0.0.1:8001" // 本地API
let devServer = "127.0.0.1:8001"   // 测试服API
let prodServer = "127.0.0.1:8001"  // 正式服API
let env = "local" // 环境：local|dev|prod
let suffix = "/home" // 接口请求URL前缀
let wsSuffix = "/socket?authorization=" // ws连接URL前缀

// baseUrl 获取接口请求URL
export function baseUrl() {
	if (env == "dev") {
		return `https://${devServer}${suffix}`
	} else if (env == "prod") {
		return `https://${prodServer}${suffix}`
	} else {
		return `http://${localServer}${suffix}`
	}
}

// wsUrl 获取ws连接URL
export function wsUrl() {
	if (env == "dev") {
		return `wss://${devServer}${wsSuffix}${getToken()}`
	} else if (env == "prod") {
		return `wss://${prodServer}${wsSuffix}${getToken()}`
	} else {
		return `ws://${localServer}${wsSuffix}${getToken()}`
	}
}

const sec = 1000
const min = 60 * sec
const hour = 6 * min
const day = 24 * hour
export const wsHeartBeatTime = 3 * min // 心跳间隔时间

import {
	wsHeartBeatTime,
	wsUrl
} from "./config"

export var is_open_socket = false;
var data = null;
var heartbeatInterval = null;
var ws_reconnectTimeOut = null;
var socketTask = null;

export function ws_init() {
	console.log("ws初始化")
	is_open_socket = false //避免重复连接
	data = null
	//心跳检测
	heartbeatInterval = null //检测服务器端是否还活着
	ws_reconnectTimeOut = null //重连之后多久再次重连

	try {
		return ws_connectSocketInit()
	} catch (e) {
		console.log('catch');
		is_open_socket = false
		ws_reconnect();
	}
}

// 进入这个页面的时候创建websocket连接【整个页面随时使用】
export function ws_connectSocketInit() {
	socketTask = uni.connectSocket({
		url: wsUrl(),
		success: () => {
			console.log("正准备建立websocket中...");
			// 返回实例
			return socketTask
		}
	});
	socketTask.onOpen((res) => {
		console.log("WebSocket连接正常！");
		clearTimeout(ws_reconnectTimeOut)
		clearInterval(heartbeatInterval)
		ws_start();
		// 注：只有连接正常打开中 ，才能正常收到消息
		socketTask.onMessage((res) => {
            // 收到消息后的操作
		});
	})
	// 监听连接失败，这里代码我注释掉的原因是因为如果服务器关闭后，和下面的onclose方法一起发起重连操作，这样会导致重复连接
	// uni.onSocketError((res) => {
	// 	console.log('WebSocket连接打开失败，请检查！');
	// 	this.is_open_socket = false;
	// 	this.ws_reconnect();
	// });
	// 这里仅是事件监听【如果socket关闭了会执行】
	socketTask.onClose(() => {
		console.log("已经被关闭了")
		is_open_socket = false;
		ws_reconnect();
	})
}

//发送消息
export function ws_send(value) {
	// 注：只有连接正常打开中 ，才能正常成功发送消息
	socketTask.send({
		data: value,
		async success() {
			console.log("消息发送成功");
		},
	});
}
//开启心跳检测
export function ws_start() {
	heartbeatInterval = setInterval(() => {
		data = {
			event: "ping"
		}
		console.log(data)
		ws_send(JSON.stringify(data));
	}, wsHeartBeatTime)
}
//重新连接
export function ws_reconnect() {
	ws_close()
	console.log("是否认为关闭", is_open_socket)
	//停止发送心跳
	clearInterval(heartbeatInterval)
	//如果不是人为关闭的话，进行重连
	if (!is_open_socket) {
		ws_reconnectTimeOut = setTimeout(() => {
			ws_connectSocketInit();
		}, 3000)
	}
}
//外部获取消息
export function ws_getMessage(callback) {
	socketTask.onMessage((res) => {
		return callback(res)
	})
}

export function ws_close(loguot) {
    // 判断是否为认为退出登录
	loguot ? is_open_socket = true : ""
	socketTask.close()
}