websocket基于jetty9的java服务端+客户端实现

接上文,“websocket协议简析”。分享一下基于jetty9实现java版的webcoket服务端和客户端demo。
没有理论,话不多说,代码奉上。
准备工作:
新建maven工程,pom文件引入:

          <dependency>
            <groupId>org.eclipse.jetty.websocket</groupId>
            <artifactId>websocket-client</artifactId>
            <version>9.4.6.v20170531</version>
        </dependency>
        <dependency>
            <groupId>org.eclipse.jetty.websocket</groupId>
            <artifactId>websocket-server</artifactId>
            <version>9.4.6.v20170531</version>
        </dependency>
        <dependency>
            <groupId>org.eclipse.jetty.websocket</groupId>
            <artifactId>websocket-api</artifactId>
            <version>9.4.6.v20170531</version>
    </dependency>

服务端实现:
服务端启动类:

package jasonleewebsocket.websocket.server;

import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.handler.ContextHandler;
import org.eclipse.jetty.websocket.server.WebSocketHandler;
import org.eclipse.jetty.websocket.servlet.WebSocketServletFactory;

public class WebSocketServerTest{
       public static void main(String args[]) 
        { 
            Server server = new Server(7778); 
            WebSocketHandler  wsHandler = new WebSocketHandler(){ 
                @Override 
                public void configure(WebSocketServletFactory factory) 
                { 
                    //注册自定义事件监听器
                    factory.register(MyEchoSocket.class); 
                } 
            }; 
            ContextHandler context = new ContextHandler(); 
            context.setContextPath("/echo"); 
            context.setHandler(wsHandler); 
            
            server.setHandler(wsHandler); 
            try 
            { 
                server.start(); 
                server.join(); 
            } 
            catch (Exception e) 
            { 
                e.printStackTrace(); 
            } 
        } 
}

自定义监听类:

package jasonleewebsocket.websocket.server;

import java.io.IOException;
import java.nio.ByteBuffer;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;

import org.eclipse.jetty.websocket.api.Session;
import org.eclipse.jetty.websocket.api.WebSocketListener;
import org.eclipse.jetty.websocket.api.WriteCallback;

public class MyEchoSocket implements WebSocketListener{
    //维持session
    private static final Set< Session > sessions = 
            Collections.synchronizedSet( new HashSet< Session >() );
    private Session session;
    
    //连接关闭,移除session列表
    public void onWebSocketClose(int statusCode, String reason) {
        System.out.println(reason);
        sessions.remove(this.session);
        this.session = null;
    }

    //建立连接后保存session
    public void onWebSocketConnect(Session session) {
        this.session = session;
        System.out.println(session.isOpen());
        sessions.add(session);
    }
    
    //错误处理
    public void onWebSocketError(Throwable cause) {
        System.out.println(cause.getMessage());
        
    }

    //接收字符串类型消息,并通知所有客户端
    public void onWebSocketBinary(byte[] payload, int offset, int len) {
        System.out.println("WebSocketBinary:"+new String(payload));
        for(final Session session : sessions){
            session.getRemote().sendBytes(ByteBuffer.wrap(payload), new WriteCallback() {
                //回调处理发送是否成功
                public void writeSuccess() {
                    System.out.println("success");
                }
                
                public void writeFailed(Throwable x) {
                    System.out.println("senderror:"+x.getMessage());
                }
            });
        }
    }

    //接收字符串类型消息,并转发给所有客户端
    public void onWebSocketText(String message) {
        System.out.println("text message:"+message);
        try {
            for(final Session session : sessions){
                session.getRemote().sendString("server to convert text:"+message);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

客户端实现:
客户端启动类:

package jasonleewebsocket.websocket.client;

import java.net.URI;
import java.util.concurrent.TimeUnit;

import org.eclipse.jetty.websocket.client.ClientUpgradeRequest;
import org.eclipse.jetty.websocket.client.WebSocketClient;

public class SimpleEchoClient
{
    public static void main(String[] args)
    {
         String destUri = "ws://localhost:7778/echo";
        if (args.length > 0)
        {
            destUri = args[0];
        }

        WebSocketClient client = new WebSocketClient();
        SimpleEchoSocket socket = new SimpleEchoSocket();
        try
        {
            client.start();

            URI echoUri = new URI(destUri);
            ClientUpgradeRequest request = new ClientUpgradeRequest();
            client.connect(socket,echoUri,request);
            System.out.printf("Connecting to : %s%n",echoUri);
            // wait for closed socket connection.
            socket.awaitClose(500,TimeUnit.SECONDS);
        }
        catch (Throwable t)
        {
            t.printStackTrace();
        }
        finally
        {
            try
            {
                client.stop();
            }
            catch (Exception e)
            {
                e.printStackTrace();
            }
        }
    }
}

客户端监听类:

package jasonleewebsocket.websocket.client;

import java.nio.ByteBuffer;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

import org.eclipse.jetty.websocket.api.Session;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketConnect;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketMessage;
import org.eclipse.jetty.websocket.api.annotations.WebSocket;

@WebSocket
public class SimpleEchoSocket {

    private final CountDownLatch closeLatch;

    public SimpleEchoSocket() {
        this.closeLatch = new CountDownLatch(1);
    }

    public boolean awaitClose(int duration, TimeUnit unit) throws InterruptedException {
        return this.closeLatch.await(duration, unit);
    }

    @OnWebSocketMessage
    public void onMessage(String msg) {
        System.out.printf("Got msg: %s%n", msg);
    }

    @OnWebSocketMessage
    public void onMessage(byte[] buffer, int offset, int length) {
        System.out.printf("Got msg: %s%n", new String(buffer));
    }

    @OnWebSocketConnect
    public void onConnect(Session session) {
        System.out.printf("Got connect: %s%n", session);
        try {
            session.getRemote().sendBytes(ByteBuffer.wrap("test byte".getBytes()));
            session.getRemote().sendString("456");
        } catch (Throwable t) {
            t.printStackTrace();
        }
    }
}

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,332评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,508评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,812评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,607评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,728评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,919评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,071评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,802评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,256评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,576评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,712评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,389评论 4 332
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,032评论 3 316
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,798评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,026评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,473评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,606评论 2 350

推荐阅读更多精彩内容