spring-boot框架下的websocket服務(wù)

        這幾天在做web端實(shí)時(shí)展示服務(wù)端日志文件新增內(nèi)容的功能。要滿足實(shí)時(shí)的需求，我選擇的方案是在web端跟服務(wù)端建立一個(gè)websocket鏈接，由服務(wù)端通過tail -f 命令將文件新增內(nèi)容發(fā)送給web端。

        關(guān)于websocket的介紹，可以參考這篇博文：http://www.cnblogs.com/lizhenghn/p/5155933.html（鏈接僅用于學(xué)習(xí)交流，如有版權(quán)問題請及時(shí)告知）。這里我主要想介紹的是在spring-boot框架下如何發(fā)布websocket服務(wù)。

一、在服務(wù)端發(fā)布websocket服務(wù)

        服務(wù)端發(fā)布websocket服務(wù)有幾種方式，包括Servlet容器掃描初始化、Spring掃描初始化。我使用的是第二種方式，可以將websocket服務(wù)定義為一個(gè)單獨(dú)的類實(shí)例。

        Spring掃描初始化時(shí)，需要先定義一個(gè)Bean：ServerEndpointExporter，以聲明服務(wù)端。我把這個(gè)Bean獨(dú)立放到一個(gè)websocket 配置類中。

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;

@Configuration
public class WebSocketConfig {
	@Bean  
    public ServerEndpointExporter serverEndpointExporter (){  
        return new ServerEndpointExporter();  
    }  

}

        接下來是定義websocket服務(wù)接口，并使用關(guān)鍵字 @ServerEndpoint("/websocketPath") 聲明一個(gè)接口的訪問路徑。

import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.concurrent.CopyOnWriteArraySet;

import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;

import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.RestController;

@ServerEndpoint("/logWebsocket")  
@Component 
@RestController
public class LogWebSocketHandle {
	
    private Session session;//每個(gè)websocket會(huì)話連接對應(yīng)一個(gè)session
    private static CopyOnWriteArraySet<LogWebSocketHandle> webSocketSet = new CopyOnWriteArraySet<>(); 
    
    /**
     * 新的WebSocket請求開啟。
     * 新建立連接后會(huì)觸發(fā)onOpen方法
     * @throws JSchException 
     * @throws IOException 
     */
    @OnOpen
    public void onOpen(Session session) throws JSchException, IOException {
    	this.session = session;
    	webSocketSet.add(this);
    	//服務(wù)端保留session信息，并返回結(jié)果給客戶端
    	//這個(gè)語句用于服務(wù)端給客戶端發(fā)送數(shù)據(jù)
    	session.getBasicRemote().sendText("正在獲取日志<br>");    	   	
    }

    /**
     * WebSocket請求關(guān)閉。
     * websocket連接關(guān)閉后，會(huì)觸發(fā)onClose方法
     */
    @OnClose
    public void onClose() {
    	webSocketSet.remove(this);                
    }

    @OnError
    public void onError(Throwable thr) {
        thr.printStackTrace();
    }
    
    /**
    * 客戶端發(fā)送消息時(shí)，服務(wù)端通過onMessage方法接收
    *／
    @OnMessage
    public void onMessage (String message, Session session) throws IOException, JSchException, InterruptedException {  
        LOG.info("來自客戶端的message:" + message);            	
        try {        	
    		//process message
    		//TODO
        } catch (IOException e) {
            e.printStackTrace();
        }
// 給客戶端群發(fā)消息  
//        for ( Session item : webSocketSet ){  
//                item.getBasicRemote().sendText(message);  
//        }  
    }         
}

二、web端建立websocket連接

var websocket_host = window.location.host;
//與服務(wù)端建立websocket連接
var websocket = new WebSocket("ws://"+websocket_host+"/項(xiàng)目名/logWebsocket");
//連接建立后，會(huì)觸發(fā)onopen方法
websocket.onopen = function(event){
	console.log("opened!");
	$("#chart_multiple div").append(event.data);
	//向服務(wù)端發(fā)送數(shù)據(jù)
	websocket.send(message);
};
//接收服務(wù)端的數(shù)據(jù)
websocket.onmessage = function(event){
	$("#chart_multiple div").append(event.data);
	$("#chart_multiple").scrollTop(
	    $("#chart_multiple div").height()-$("#chart_multiple").height()
	);
}

三、使用nginx轉(zhuǎn)發(fā)時(shí)的額外配置

        如果項(xiàng)目使用了nginx進(jìn)行負(fù)載均衡，那么需要在nginx.conf配置文件中添加一個(gè)websocket轉(zhuǎn)發(fā)配置，具體為：

location /#{websocket所在的項(xiàng)目名}/logWebsocket {
    proxy_pass http://ip:port/#{websocket所在的項(xiàng)目名}/logWebsocket;
    proxy_set_header Upgrade $http_upgrade;
    proxy_set_header Connection "upgrade";
    proxy_read_timeout 7200s;
}