websocket通信 實現java模擬一個client與webclient通信

發文原由：

熟悉socket通信的同學，對於socket模擬server與client，實現相互通信，

或者使用websocket與java模擬的websocket服務器通信（比如一個聊天室），對於這些都比較熟悉了。但是可能會有下面這種情況，

java項目（比如storm流式處理）會在處理數據時候產生一些信息，比如監控某車間、風場，發電廠的機器數據，一旦出現異常需要即時推送異常信息到UI端/Web端，實時顯示出來。這時候以上兩種較為熟知的通信就會不太方便。我也是查了好多資料，得到下面這種方法：

web項目啟動一個websocket的的server和一個web端的client。然后另外也用websocket（這是很少人采用的一種方式）寫一個java端的client。我們把這個client放在產生實際項目中產生數據的地方，通過client與server的連接，把異常消息推送到server，由server廣播給web端實時顯示。

環境：

tomcat 7

maven項目

IDE是IDEA2017

項目功能描述：

啟動項目，會啟動一個web端的websocket-client和一個java模擬的websocket-server。

項目中另外還有一個Main類，單獨啟動，會模擬啟動一個java端的websocket-client。

java-client可以發消息，通過server即時將消息推送到web端。

　　先看web端的js代碼，非常簡單：

<script type="text/javascript">
    var socket = new WebSocket("ws:localhost:8080/websocket/chat")
    var name = "web"
    socket.onopen = function () {
        socket.send("webinit" + "-" + name);
    }
    socket.onmessage = function (messageMeta) {
        var message = messageMeta.data
        document.getElementById("show").innerHTML +=  "<br>"+message
    }
</script>
…………………………………………
<div id="show">初始的一句話</div>

js使用websocket與服務端建立鏈接，這里指定了web的name是"web"，onopen的時候發送一條消息給服務端，用來告訴服務端當前web的name，后續會用到這個name。

onmessage用來接收服務器發來的消息。追加到web頁面。

 

　　再看服務端代碼，也很好理解：

首先看一下截圖，代碼非常少，采用注解的方式，下圖中紅色筆跡1用來標識當前websocket的名字。

當有連接接入server的時候，會執行connect()方法。把當前ChatAn對象加入到connections集合中。接着執行3處的方法。會根據接收到的消息進行相應的處理。下面看“3”處的代碼：

@OnMessage
public void receiveMessage(String message) throws IOException {
    ChatAn chatAn = new ChatAn();
    System.out.println(message);
    if (message.startsWith("webinit")) {
        //web-client建立連接的時候自動發的第一天消息 用來命名連接userName  message格式為 webinit-web
        this.userName = message.split("-")[1];
        sendMessage(this, "連接建立");//這句話用來反饋web與server建立連接的
    } else if (message.startsWith("javaclient")) {
        //接收來自java-client的消息 每個連接無需命名username connectionID
        for (int i = 0; i < connections.size(); i++) {
            if (connections.get(i).userName.equals("web")) {//
                chatAn = connections.get(i);
                break;
            }
        }
        sendMessage(chatAn, message.substring("javaclient".length(), message.length()));
    }
}

startsWith("webinit")是當web端連接的時候執行，只會執行一次。

startsWith("javaclient")是java端連接的時候執行。在這里會遍歷connections找到web的連接，然后調用sendMessage傳參數，發消息。

 

　　最后是重點了，看java怎么用websocket模擬client，有兩個類:

　　第一個：client.java

package socket1;
import javax.websocket.*;
@ClientEndpoint()
public class client {
    @OnOpen
    public void onOpen(Session session) {}
    @OnMessage
    public void onMessage(String message) {
        System.out.println("Client onMessage: " + message);
    }
    @OnClose
    public void onClose() {}
}

第二個Main.java，非常簡單，重點看方法：

package socket1;
import javax.websocket.ContainerProvider;
import javax.websocket.DeploymentException;
import javax.websocket.Session;
import javax.websocket.WebSocketContainer;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.URI;
public class Main {
    private static String uri = "ws://localhost:8080/websocket/chat";
    private static Session session;
    private void start() {
        WebSocketContainer container = null;
        try {
            container = ContainerProvider.getWebSocketContainer();
        } catch (Exception ex) {
            System.out.println("error" + ex);
        }
        try {
            URI r = URI.create(uri);
            session = container.connectToServer(client.class, r);
        } catch (DeploymentException | IOException e) {
            e.printStackTrace();
        }
    }
    public static void main(String[] args) {
        Main client = new Main();
        client.start();
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String input = "";
        try {
            do {
                input = br.readLine();
                if (!input.equals("exit"))
                    client.session.getBasicRemote().sendText("javaclient"+input);
            } while (!input.equals("exit"));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

 

效果演示

動圖中的數據是我自己運用到實際的項目產生的數據。是把java客戶端嵌在了實際項目中。運行我的源碼也可以得到類似的效果，

web端可以實時地不停地接收來自java客戶端的數據。

動圖

 

github項目地址：

https://github.com/simuhunluo/socket5

 

歡迎關注微信公眾號“IT客“ ，投稿郵箱 [e-mail]