Tomcat實現Web Socket

我是賣報滴小行家?發表於2018-10-17
TomcatWeb

原文連結：https://www.dubby.cn/detail.html?id=9103

1、依賴

本文使用的是Tomcat9

專案結構也是最基本的servlet的專案結構：

image

程式碼地址：github.com/dubby1994/t…

其實啥依賴都不需要，但是需要幾個api，這些在Tomcat裡都已經提供了，但是程式碼裡還是需要提供一下，不然編譯報錯：

<dependency>
    <groupId>javax.websocket</groupId>
    <artifactId>javax.websocket-api</artifactId>
    <version>1.0</version>
    <scope>provided</scope>
</dependency>
複製程式碼

2、配置

ExamplesConfig.java

package cn.dubby.tomcat.study.config;
import cn.dubby.tomcat.study.ws.EchoEndpoint;
import javax.websocket.Endpoint;
import javax.websocket.server.ServerApplicationConfig;
import javax.websocket.server.ServerEndpointConfig;
import java.util.HashSet;
import java.util.Set;

public class ExamplesConfig implements ServerApplicationConfig {
    @Override
    public Set<ServerEndpointConfig> getEndpointConfigs(Set<Class<? extends Endpoint>> scanned) {
        Set<ServerEndpointConfig> result = new HashSet<>();
        if (scanned.contains(EchoEndpoint.class)) {
            result.add(ServerEndpointConfig.Builder.create(EchoEndpoint.class, "/websocket").build());
        }
        return result;
    }
    @Override
    public Set<Class<?>> getAnnotatedEndpointClasses(Set<Class<?>> scanned) {
        Set<Class<?>> results = new HashSet<>();
        for (Class<?> clazz : scanned) {
            if (clazz.getPackage().getName().contains("ws")) {
                results.add(clazz);
            }
        }
        return results;
    }
}
複製程式碼

其中getEndpointConfigs是配置所有繼承Endpoint的類，而getAnnotatedEndpointClasses是配置所有被@ServerEndpoint修飾的類。

3、繼承Endpoint

public class EchoEndpoint extends Endpoint {
    private static final AtomicLong count = new AtomicLong();
    @Override
    public void onOpen(Session session, EndpointConfig endpointConfig) {
        System.out.println("線上人數:" + count.incrementAndGet());
        session.addMessageHandler(new EchoMessageHandlerText(session));
    }
    @Override
    public void onClose(Session session, CloseReason closeReason) {
        System.out.println("線上人數:" + count.decrementAndGet());
    }
    private static class EchoMessageHandlerText implements MessageHandler.Partial<String> {
        private final Session session;
        private EchoMessageHandlerText(Session session) {
            this.session = session;
        }
        @Override
        public void onMessage(String message, boolean last) {
            if (session == null)
                return;
            System.out.println(session.getId() + "\t" + message);
            try {
                session.getBasicRemote().sendText(message, last);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}
複製程式碼

4、使用註解

@ServerEndpoint("/websocket2")
public class EchoEndpoint2 {
    private static final AtomicLong count = new AtomicLong();
    @OnOpen
    public void open(Session session) {
        System.out.println("線上人數:" + count.incrementAndGet());
        try {
            session.getBasicRemote().sendText("welcome");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    @OnClose
    public void close(Session session) {
        System.out.println("線上人數:" + count.decrementAndGet());
    }
    @OnMessage
    public void echoTextMessage(Session session, String message) {
        System.out.println(session.getId() + "\t" + message);
        try {
            if (session.isOpen()) {
                session.getBasicRemote().sendText(message);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    @OnMessage
    public void echoBinaryMessage(Session session, ByteBuffer bb) {
        System.out.println(session.getId() + "\t" + bb.toString());
        try {
            if (session.isOpen()) {
                session.getBasicRemote().sendBinary(bb);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    @OnMessage
    public void echoPongMessage(PongMessage pm) {
        //pass
    }
}
複製程式碼

例子就不給了，和昨天的幾乎是一樣的，你可以直接用昨天的那個html來測試今天的。

相關文章