webSocket以及https
阿新 • • 發佈：2020-07-15
需求:就是當後臺操作完大量的資料之後，需要通知前端修改狀態，於是第一時間想到了使用WebSocket。
概念的話這裡就不解釋了。
直接貼上我的程式碼,在springBoot中的使用。
首先在pom.xml中加入依賴
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
將配置類加入容器管理
@Configuration public class WebSocketConfiguration { @Bean public ServerEndpointExporter serverEndpointExporter(){ return new ServerEndpointExporter(); } }
websocket服務端
@ServerEndpoint("/websocket/submitOnline")
@Component
public class WebSocketServer {
private Logger log = Logger.getLogger("webSocketServer");
//這裡是我自己的業務邏輯
Result result = null;
try {
SubmitOnlineVo submitOnlineVo = new SubmitOnlineVo();
submitOnlineVo.setId(Integer.parseInt(message));
submitOnlineVo.setStatus(PlanStatusConstant.ATONLINE);
Integer i = webSocketServer.adsMsgPlanService.updatePlan(submitOnlineVo);
if(i==-1){
result = new Result(BusinessConstant.SYSTEM_ERROR_CODE, "上線人數超過30萬！！！", i);
}else if(i==0){
result = new Result(BusinessConstant.SYSTEM_ERROR_CODE, "修改失敗", i);
}else{
Map<String,String> map = new HashMap<>();
map.put("id",message);
map.put("status","上線待投放");
result = new Result(BusinessConstant.SYSTEM_SUCESS_CODE, "修改成功", map);
}
this.sendAll(JSON.toJSONString(result));
}catch (Exception e){
e.printStackTrace();
result = new Result(BusinessConstant.SYSTEM_ERROR_CODE, e.getMessage(), null);
this.sendAll(JSON.toJSONString(result));
}
}
private void sendAll(String message){
for (Map.Entry<String, Session> sessionEntry : clients.entrySet()) {
sessionEntry.getValue().getAsyncRemote().sendText(message);
}
}
}
客戶端程式碼(前端並不是我寫的，只是自己測試的時候寫的一個簡單頁面)
<html> <head></head> <body> <div id="messageId"></div> <script> var socket; if(typeof(WebSocket) == "undefined") { console.log("您的瀏覽器不支援WebSocket"); }else{ console.log("您的瀏覽器支援WebSocket"); //實現化WebSocket物件，指定要連線的伺服器地址與埠 建立連線 socket = new WebSocket("ws://localhost:8080/websocket/submitOnline"); //開啟事件 socket.onopen = function() { console.log("Socket 已開啟"); socket.send("dfgh"); }; //獲得訊息事件 socket.onmessage = function(msg) { console.log(msg.data); //發現訊息進入 開始處理前端觸發邏輯 let msgDiv = document.getElementById("messageId"); msgDiv.innerHTML += msg.data; }; //關閉事件 socket.onclose = function() { console.log("Socket已關閉"); }; //發生了錯誤事件 socket.onerror = function() { alert("Socket發生了錯誤"); //此時可以嘗試重新整理頁面 } } </script> </body> </html>
這個時候一個websocket已經搭建好了，但是這個時候的安全性存在問題.
因為是ws,就類似於http嘛，所以就想弄成wss的，也就是https
然後我這邊通過jdk生成一個金鑰來在本地實現一個https.
在cmd中輸入(要配置了jdk的環境變數)
keytool -genkey -alias tomcat -storetype PKCS12 -keyalg RSA -keysize 2048 -keystore keystore.p12 -validity 3650
引數就不一一解釋了.
然後會在c盤/使用者/當前使用者資料夾下生成一個keystore.p12的檔案.
將它放到專案的resources目錄下,然後在application.properties檔案中加入如下配置
server.port=8443 server.ssl.enabled=true server.ssl.key-store=classpath:keystore.p12 server.ssl.key-store-password=ouyang(你在cmd中輸入的密碼) server.ssl.key-store-type=PKCS12 server.ssl.key-alias=tomcat
然後一個配置類（允許http和https請求同時存在,將http轉發到https）
@Configuration public class ConnectorConfig { @Bean public ServletWebServerFactory servletContainer() { TomcatServletWebServerFactory tomcat = new TomcatServletWebServerFactory() { @Override protected void postProcessContext(Context context) { SecurityConstraint securityConstraint = new SecurityConstraint(); securityConstraint.setUserConstraint("CONFIDENTIAL"); SecurityCollection collection = new SecurityCollection(); collection.addPattern("/*");//設定所有路徑都配置https securityConstraint.addCollection(collection); context.addConstraint(securityConstraint); } }; tomcat.addAdditionalTomcatConnectors(getHttpConnector()); return tomcat; } private Connector getHttpConnector() { Connector connector = new Connector("org.apache.coyote.http11.Http11NioProtocol"); connector.setScheme("http"); connector.setPort(8080); connector.setSecure(false); connector.setRedirectPort(8443); return connector; } }
如果要在postman中測試還需要關閉ssl certificate verification
這時候我們就可以將客戶端的websocket程式碼改成wss來保證安全性了,當然這只是本地測試，在實際的生產環境應該都是https協議,所以直接使用wss就行。