Java通過(guò)WebSocket實(shí)現(xiàn)異步導(dǎo)出解決思路
更新時(shí)間：2024年01月12日 11:38:01 作者：夜の雨
這篇文章主要介紹了通過(guò)WebSocket實(shí)現(xiàn)異步導(dǎo)出,本篇文章記錄大批量數(shù)據(jù)導(dǎo)出時(shí)間過(guò)長(zhǎng),導(dǎo)致接口請(qǐng)求超時(shí)問(wèn)題,本文給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下
前言
本篇文章記錄大批量數(shù)據(jù)導(dǎo)出時(shí)間過(guò)長(zhǎng),導(dǎo)致接口請(qǐng)求超時(shí)問(wèn)題。
解決思路與流程
- 前端點(diǎn)擊導(dǎo)出按鈕時(shí)開啟websocket連接
- 邏輯處理異步執(zhí)行
- 文件處理好后，得到文件的絕對(duì)路徑
- 后臺(tái)socket通知前端絕對(duì)路徑的地址
- 前端下載文件到瀏覽器
本地環(huán)境
Vue版本：2.6.12
Java版本：1.8
SpringBoot版本：2.2.13
項(xiàng)目使用若依前后分離版
一、WebSocket配置
1.pom文件配置
<!-- SpringBoot Websocket -->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
<version>2.2.13.RELEASE</version>
</dependency>2.信號(hào)量相關(guān)處理
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.concurrent.Semaphore;
/**
* 信號(hào)量相關(guān)處理
*
* @author
*/
public class SemaphoreUtils {
/**
* SemaphoreUtils 日志控制器
*/
private static final Logger LOGGER = LoggerFactory.getLogger(SemaphoreUtils.class);
/**
* 獲取信號(hào)量
*
* @param semaphore
* @return
*/
public static boolean tryAcquire(Semaphore semaphore) {
boolean flag = false;
try {
flag = semaphore.tryAcquire();
} catch (Exception e) {
LOGGER.error("獲取信號(hào)量異常", e);
}
return flag;
}
/**
* 釋放信號(hào)量
*
* @param semaphore
*/
public static void release(Semaphore semaphore) {
try {
semaphore.release();
} catch (Exception e) {
LOGGER.error("釋放信號(hào)量異常", e);
}
}
}3.websocket 配置
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;
/**
* websocket 配置
*
* @author
*/
@Configuration
public class WebSocketConfig {
@Bean
public ServerEndpointExporter serverEndpointExporter() {
return new ServerEndpointExporter();
}
}4.websocket 消息處理
import com.hnxr.scada.utils.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
import javax.websocket.*;
import javax.websocket.server.ServerEndpoint;
import java.util.concurrent.Semaphore;
/**
* websocket 消息處理
*
* @author
*/
@Component
@ServerEndpoint("/websocket/message")
public class WebSocketServer {
/**
* WebSocketServer 日志控制器
*/
private static final Logger LOGGER = LoggerFactory.getLogger(WebSocketServer.class);
/**
* 默認(rèn)最多允許同時(shí)在線人數(shù)100
*/
public static int socketMaxOnlineCount = 100;
public static Semaphore socketSemaphore = new Semaphore(socketMaxOnlineCount);
/**
* 連接建立成功調(diào)用的方法
*/
@OnOpen
public void onOpen(Session session) throws Exception {
boolean semaphoreFlag = false;
// 嘗試獲取信號(hào)量
semaphoreFlag = SemaphoreUtils.tryAcquire(socketSemaphore);
if (!semaphoreFlag) {
// 未獲取到信號(hào)量
LOGGER.error("\n 當(dāng)前在線人數(shù)超過(guò)限制數(shù)- {}", socketMaxOnlineCount);
WebSocketUsers.sendMessageToUserByText(session, "當(dāng)前在線人數(shù)超過(guò)限制數(shù)：" + socketMaxOnlineCount);
session.close();
} else {
//截取前端傳入的id
String sessionId = StringUtils.substringAfterLast(session.getQueryString(), "socketId=");
System.out.println("---------------------解析socketId" + sessionId);
// 添加用戶
WebSocketUsers.put(sessionId, session);
System.out.println("-----------------當(dāng)前用戶" + WebSocketUsers.getUsers());
LOGGER.info("\n 建立連接 - {}", session);
LOGGER.info("\n 當(dāng)前人數(shù) - {}", WebSocketUsers.getUsers().size());
WebSocketUsers.sendMessageToUserByText(session, "連接成功");
}
}
/**
* 連接關(guān)閉時(shí)處理
*/
@OnClose
public void onClose(Session session) {
LOGGER.info("\n 關(guān)閉連接 - {}", session);
String sessionId = StringUtils.substringAfterLast(session.getQueryString(), "socketId=");
// 移除用戶
WebSocketUsers.remove(sessionId);
// 獲取到信號(hào)量則需釋放
SemaphoreUtils.release(socketSemaphore);
}
/**
* 拋出異常時(shí)處理
*/
@OnError
public void onError(Session session, Throwable exception) throws Exception {
if (session.isOpen()) {
// 關(guān)閉連接
session.close();
}
String sessionId = StringUtils.substringAfterLast(session.getQueryString(), "socketId=");
LOGGER.info("\n 連接異常 - {}", sessionId);
LOGGER.info("\n 異常信息 - {}", exception);
// 移出用戶
WebSocketUsers.remove(sessionId);
// 獲取到信號(hào)量則需釋放
SemaphoreUtils.release(socketSemaphore);
}
/**
* 服務(wù)器接收到客戶端消息時(shí)調(diào)用的方法
*/
@OnMessage
public void onMessage(String message, Session session) {
WebSocketUsers.sendMessageToUserByText(session, message);
}
}白名單放開socket配置接口
.antMatchers("/websocket/**").permitAll()
5.websocket 用戶集
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.websocket.Session;
import java.io.IOException;
import java.util.Collection;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
/**
* websocket 客戶端用戶集
*
* @author
*/
public class WebSocketUsers {
/**
* WebSocketUsers 日志控制器
*/
private static final Logger LOGGER = LoggerFactory.getLogger(WebSocketUsers.class);
/**
* 用戶集
*/
private static Map<String, Session> USERS = new ConcurrentHashMap<String, Session>();
/**
* 存儲(chǔ)用戶
*
* @param key 唯一鍵
* @param session 用戶信息
*/
public static void put(String key, Session session) {
USERS.put(key, session);
}
/**
* 移除用戶
*
* @param session 用戶信息
* @return 移除結(jié)果
*/
public static boolean remove(Session session) {
String key = null;
boolean flag = USERS.containsValue(session);
if (flag) {
Set<Map.Entry<String, Session>> entries = USERS.entrySet();
for (Map.Entry<String, Session> entry : entries) {
Session value = entry.getValue();
if (value.equals(session)) {
key = entry.getKey();
break;
}
}
} else {
return true;
}
return remove(key);
}
/**
* 移出用戶
*
* @param key 鍵
*/
public static boolean remove(String key) {
LOGGER.info("\n 正在移出用戶 - {}", key);
Session remove = USERS.remove(key);
if (remove != null) {
boolean containsValue = USERS.containsValue(remove);
LOGGER.info("\n 移出結(jié)果 - {}", containsValue ? "失敗" : "成功");
return containsValue;
} else {
return true;
}
}
/**
* 獲取在線用戶列表
*
* @return 返回用戶集合
*/
public static Map<String, Session> getUsers() {
return USERS;
}
/**
* 群發(fā)消息文本消息
*
* @param message 消息內(nèi)容
*/
public static void sendMessageToUsersByText(String message) {
Collection<Session> values = USERS.values();
for (Session value : values) {
sendMessageToUserByText(value, message);
}
}
/**
* 發(fā)送文本消息
*
* @param session 緩存
* @param message 消息內(nèi)容
*/
public static void sendMessageToUserByText(Session session, String message) {
if (session != null) {
try {
session.getBasicRemote().sendText(message);
} catch (IOException e) {
LOGGER.error("\n[發(fā)送消息異常]", e);
}
} else {
LOGGER.info("\n[你已離線]");
}
}
}二、建立WebSocket連接
1.導(dǎo)出按鈕
<el-button type="warning" plain icon="el-icon-download" size="mini" :loading="exportLoading"
@click="handleExport">導(dǎo)出
</el-button>2.導(dǎo)出方法
/** 導(dǎo)出按鈕操作 */
handleExport() {
this.$confirm('是否確認(rèn)導(dǎo)出所有?', '警告', {
confirmButtonText: '確定',
cancelButtonText: '取消',
type: 'warning'
})
.then(() => {
const ip = window.location.host;
//TODO 服務(wù)器端口號(hào) nginx代理為4430
const hostIp = ip.split(":")[0];
console.log('ip：', hostIp)
const url = "wss://" + hostIp + ":4430/websocket/message";
//TODO 本地端口號(hào)為9996 項(xiàng)目啟動(dòng)端口
// let url = "wss://localhost:9996/websocket/message";
console.log('url', url)
const socketId = this.getRandomId();
//將socketId當(dāng)做參數(shù)傳入后端
this.queryParams.sessionId = socketId
const socket = new WebSocket(url + '?socketId=' + socketId);
socket.onopen = function (event) {
console.log("已開啟websocket連接!" + "sessionId:" + socketId)
};
socket.onmessage = function (event) {
const val = event.data
console.log("收到消息", val)
if (val.endsWith("asyExport")) {
const path = val.slice(0, val.indexOf("asyExport"));
console.log("文件名：" + path)
const baseURL = process.env.VUE_APP_BASE_API
window.location.href = baseURL + "/common/download?fileName=" + encodeURI(path) + "&delete=" + true;
//延時(shí)關(guān)閉連接 確保WebSocket處于打開狀態(tài)
setTimeout(function () {
if (socket.readyState === WebSocket.OPEN) {
socket.close();
}
}, 5000);
}
};
// 處理連接關(guān)閉
socket.onclose = function (event) {
console.log("已經(jīng)關(guān)閉連接!");
};
// 處理連接錯(cuò)誤
socket.onerror = function (error) {
console.error("WebSocket 出現(xiàn)錯(cuò)誤:", error);
if (socket.readyState === WebSocket.OPEN) {
socket.close();
}
};
exportChineseParsing(this.queryParams)
})
},
// 生成唯一標(biāo)識(shí)符
getRandomId() {
return Math.random().toString(36).substring(2) + Date.now().toString(36);
}本地項(xiàng)目是http請(qǐng)求，前綴設(shè)置為ws
后端application.yml 的ssl配置將server下的enabled改為false
前端代理地址保持http
https請(qǐng)求，前綴設(shè)置為wss
ssl配置文件將server下的enabled改為true
前端代理地址改為https
3.導(dǎo)出接口地址
//導(dǎo)出中文解析
export function exportChineseParsing(query) {
return request({
url: '/basic/chineseParsing/export',
method: 'get',
params: query,
responseType: 'blob'
})
}瀏覽器控制臺(tái)查看連接時(shí)是否正常
前后端socket配置完成后在瀏覽器控制臺(tái)輸入
new WebSocket("ws://localhost:9996/websocket/message")
測(cè)試是否正常 如下圖 拋出異常需要檢查上方配置是否正確
4.流程說(shuō)明
首先獲取當(dāng)前ip，本地使用localhost即可，服務(wù)器部署需要配置nginx代理
ip后跟端口號(hào)，端口后是后端websocket接口地址
socket.onopen回調(diào)方法驗(yàn)證是否成功建立連接
socket.onmessage是接收后端接口異步導(dǎo)出后的消息
后端文件處理好后，約定發(fā)送的消息以asyExport字符為后綴，方便前端識(shí)別和截取文件絕對(duì)路徑
調(diào)用下載接口，關(guān)閉socket連接
如不是若依前后分離版，替換為自己的下載文件的接口就行
三、異步導(dǎo)出接口
1.導(dǎo)出接口示例
@GetMapping("/export")
public void export(BasicChineseParsing basicChineseParsing) throws Exception {
//存入文件名稱
AtomicReference<String> path = new AtomicReference<>("");
//邏輯異步處理
CompletableFuture<Void> runAsync = CompletableFuture.runAsync(() -> {
try {
//模擬邏輯處理時(shí)長(zhǎng)
System.out.println("-----------------------睡眠5秒");
Thread.sleep(5000);
.......
//創(chuàng)建臨時(shí)文件 絕對(duì)路徑
File tempFile = PoiUtils.createTempNameFile(rows, title, "_中文解析表");
String result = tempFile.getCanonicalPath().substring(tempFile.getCanonicalPath().indexOf("vehicle"));
System.out.println("-----------------------處理后文件名：" + result);
path.set(result);
} catch (InterruptedException | IOException e) {
e.printStackTrace();
}
});
//異步處理完成后
runAsync.thenRun(() -> {
// 發(fā)送消息 拿到前端傳入的socketId
String sessionId = basicChineseParsing.getSessionId();
System.out.println("-----------------------sessionId:" + sessionId);
Session session = WebSocketUsers.getUsers().get(sessionId);
webSocketServer.onMessage(path.toString() + "asyExport", session);
System.out.println("-----------------------導(dǎo)出成功");
//前端關(guān)閉socket連接
});
}2.debug調(diào)試流程截圖
下載方法大家的和我的會(huì)不一樣
但是File tempFile只要獲取到下載好后的絕對(duì)路徑就可以
我這里path存入的是下載的文件名稱
因?yàn)榍岸藭?huì)再次調(diào)用下載接口，有文件名稱，就可以將文件下載至瀏覽器了
在webSocketServer.onMessage方法中發(fā)送的消息就是文件名稱+asyExport一個(gè)特殊約定的后綴
前端在收到消息后會(huì)截取asyExport前的文件名稱
完成流程日志打印如下
因使用若依前后分離版，大家調(diào)整后的下載方法可能和我的不一樣
比如socket需要發(fā)送的信息是文件的絕對(duì)路徑，前端調(diào)用下載方法傳入絕對(duì)路徑即可
四、服務(wù)器部署nginx配置
這個(gè)配置看大家是否需要部署至服務(wù)器
server {
listen 4430 ssl;
server_name localhost;
.....
location /websocket/message {
proxy_pass https://服務(wù)器ip:4430;
proxy_http_version 1.1;
proxy_set_header Upgrade $http_upgrade;
proxy_set_header Connection "upgrade";
}
}配置需要放在當(dāng)前代理的端口配置下
如果是ssl認(rèn)證的proxy_pass后地址需要用https，沒(méi)有認(rèn)證就用http
ip后的端口是你的程序服務(wù)使用端口，不是代理映射的端口
這里配置的地址是和前端創(chuàng)建連接時(shí)const url = "wss://" + hostIp + ":4430/websocket/message";
ip和端口是一致的
總結(jié)
第一步WebSocket配置大家是可以直接復(fù)制使用
第二步我的處理是只有導(dǎo)出時(shí)才啟用socket連接，下載文件后關(guān)閉，也可以設(shè)置為全局性的，方便處理其他通知
socket通知前端的下載方式也有很多種，需要考慮的是會(huì)有其他用戶或其他導(dǎo)出接口，其他socket通知，所以通知下載的socket消息是需要具有唯一性的，上方建立socket連接時(shí)就加了保障唯一性的socketId，并當(dāng)為參數(shù)傳入后端。
到此這篇關(guān)于通過(guò)WebSocket實(shí)現(xiàn)異步導(dǎo)出的文章就介紹到這了,更多相關(guān)WebSocket異步導(dǎo)出內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
JavaScript逆向分析instagram登入過(guò)程
這篇文章主要介紹了JavaScript逆向分析instagram登入過(guò)程,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2022-07-07
three.js中文文檔學(xué)習(xí)之通過(guò)模塊導(dǎo)入
這篇文章主要給大家介紹了關(guān)于three.js中文文檔學(xué)習(xí)之通過(guò)模塊導(dǎo)入的相關(guān)資料，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家學(xué)習(xí)或使用three.js具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧。2017-11-11
js實(shí)現(xiàn)把時(shí)間戳轉(zhuǎn)換為yyyy-MM-dd hh:mm 格式(es6語(yǔ)法)
下面小編就為大家分享一篇js實(shí)現(xiàn)把時(shí)間戳轉(zhuǎn)換為yyyy-MM-dd hh:mm 格式(es6語(yǔ)法)，具有很的參考價(jià)值，希望對(duì)大家有所幫助2017-12-12
利用純js + transition動(dòng)畫實(shí)現(xiàn)移動(dòng)端web輪播圖詳解
這篇文章主要給大家介紹了利用純js + transition動(dòng)畫實(shí)現(xiàn)移動(dòng)端web輪播圖的相關(guān)資料，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友可以參考借鑒，下面來(lái)一起看看吧。2017-09-09
今天小編就為大家分享一篇關(guān)于js獲取form表單中name屬性的值，小編覺(jué)得內(nèi)容挺不錯(cuò)的，現(xiàn)在分享給大家，具有很好的參考價(jià)值，需要的朋友一起跟隨小編來(lái)看看吧2019-02-02
JavaScript實(shí)現(xiàn)在頁(yè)面間傳值的方法
這篇文章主要介紹了JavaScript實(shí)現(xiàn)在頁(yè)面間傳值的方法,涉及javascript傳值的技巧,非常具有實(shí)用價(jià)值,需要的朋友可以參考下2015-04-04
通過(guò)js實(shí)現(xiàn)分頁(yè)的代碼，一般情況下需要指定頁(yè)數(shù)，腳本之家以前也發(fā)布過(guò)一些，大家結(jié)合下即可。2010-09-09
這篇文章主要為大家詳細(xì)介紹了小程序?qū)崿F(xiàn)手寫簽名功能，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2022-07-07
這篇文章主要介紹了JS文件中加載jquery.js的方法，JS文件添加其他JS的實(shí)例代碼，感興趣的小伙伴們可以參考一下，具體如下2018-05-05
js通過(guò)iframe加載外部網(wǎng)頁(yè)的實(shí)現(xiàn)代碼
這篇文章主要介紹了js通過(guò)iframe加載外部網(wǎng)頁(yè)的實(shí)現(xiàn)代碼,需要的朋友可以參考下2015-04-04