用CountDownLatch把java的websocket请求从异步修改为同步

最新推荐文章于 2024-10-22 15:41:05 发布

touzizhuo03

最新推荐文章于 2024-10-22 15:41:05 发布

阅读量3.1k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： j2se

本文链接：https://blog.youkuaiyun.com/touzizhuo03/article/details/80020635

j2se 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种将 Java WebSocket 的异步特性封装为同步请求的方法。通过使用 CountDownLatch 实现，使得习惯于同步请求的 Java 开发者也能便捷地使用 WebSocket 进行通信。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在项目中，需要用websocket和其他服务器进行通信，而websocket是长连接，建立连接后，监听消息，然后进行消息的处理。也就是消息的异步处理。

使用java websocket需要实现的接口如下

	public abstract void onOpen( ServerHandshake handshakedata );
	public abstract void onMessage( String message );
	public abstract void onClose( int code, String reason, boolean remote );
	public abstract void onError( Exception ex );
	public void onMessage( ByteBuffer bytes ) {
	}
	public void onFragment( Framedata frame ) {
	}

对于习惯使用同步请求的java开发者来说，用起来还是不方便的。

参考网上资料，CountDownLatch把java的websocket请求从异步修改为同步，如下：

package cc.openwiki.model;

import lombok.extern.slf4j.Slf4j;
import org.java_websocket.client.WebSocketClient;
import org.java_websocket.handshake.ServerHandshake;

import java.net.URI;
import java.nio.ByteBuffer;
import java.util.concurrent.CountDownLatch;

/**
 * Created by admin on 2017/6/6.
 */
@Slf4j
public class WebSocketClientExt extends WebSocketClient {
    public WebSocketClientExt(URI serverURI){
        super(serverURI);
    }
    private Object lock = new Object();
    private Object result = null;

    private CountDownLatch downLatch = null;


    public Object sendOb(Object object){
        synchronized (lock){
            downLatch = new CountDownLatch(1);
            sendImpl(object);
            try {
                downLatch.await();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }finally {
                return result;
            }
        }
    }

    public void sendImpl( Object object ){
        if(object instanceof String){
            super.send((String)object);
        }else if(object instanceof byte[]){
            super.send((byte[])object);
        }

    }


    @Override
    public void onOpen(ServerHandshake handshakedata) {
        log.debug("open handshakedata。。。");
    }

    @Override
    public void onMessage(String message) {
        log.debug("receive message。。。");
        result = message;
        if(downLatch!=null){
            downLatch.countDown();
        }
    }

    public void onMessage( ByteBuffer bytes ) {
        log.debug("receive ByteBuffer message。。。");
        result = bytes;
        if(downLatch!=null){
            downLatch.countDown();
        }
    }

    @Override
    public void onClose(int code, String reason, boolean remote) {
        log.debug("close websocket 。。");
    }

    @Override
    public void onError(Exception ex) {
        log.debug("onError 。。");
    }
}

调用方式如下：

WebSocketClientExt webSocketClientExt = new WebSocketClientExt(uri);
try {
    webSocketClientExt.connectBlocking();
} catch (InterruptedException e) {
    log.info("connect excepiont {}", e);
}
Object result = webSocketClientExt.sendOb(data.toByteArray());

用起来是不是熟悉了 :)

by sim