在dropwizard中使用websocket

最新推荐文章于 2023-03-07 09:56:41 发布

年糕coder

最新推荐文章于 2023-03-07 09:56:41 发布

阅读量268

点赞数

分类专栏： Java 文章标签： websocket servlet java

本文链接：https://blog.csdn.net/jsq916/article/details/126531860

版权

Java 专栏收录该内容

30 篇文章 2 订阅

订阅专栏

依赖

        <dependency>
            <groupId>org.eclipse.jetty.websocket</groupId>
            <artifactId>javax-websocket-server-impl</artifactId>
            <version>9.4.48.v20220622</version>
        </dependency>

服务端的websocket的resource类

作用：定义一个websocket接口需要做什么，使用注解

package com.jsq.hibernate.websocket;

import lombok.extern.slf4j.Slf4j;
import org.eclipse.jetty.websocket.api.Session;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketClose;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketConnect;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketMessage;
import org.eclipse.jetty.websocket.api.annotations.WebSocket;

import java.io.IOException;

@Slf4j
@WebSocket()
public class HelloWebSocketServer {

    @OnWebSocketConnect
    public void onConnect(Session session) {
        log.info("on connect");
    }

    @OnWebSocketMessage
    public void onMessage(Session session, String msg) {
        log.info("msg of onMessage is {}", msg);
        try{
            session.getRemote().sendString("ok");
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    @OnWebSocketClose
    public void onWebSocketClose(Session session, int statusCode, String errorMsg) {
        log.info("connection close. host of session:{}, statusCode:{}, errorMsg:{}", session.getUpgradeRequest().getHost(), statusCode, errorMsg);
        session.close();
    }

    public void onWebSocketError(Session session, Throwable throwable) {
        log.error("have a error on web socket",throwable);
        session.close();
    }
}

servlet类

继承WebSocketServlet

package com.jsq.hibernate.websocket;

import org.eclipse.jetty.websocket.api.WebSocketPolicy;
import org.eclipse.jetty.websocket.servlet.*;

public class HelloWebSocketServlet extends WebSocketServlet {
    @Override
    public void configure(WebSocketServletFactory webSocketServletFactory) {
        WebSocketPolicy policy = webSocketServletFactory.getPolicy();
        //用户连接的超时时间（长时间没有操作自动断开，一般为10分钟）
        policy.setIdleTimeout(6000 * 1000L);
        policy.setInputBufferSize(8 * 1024);
        webSocketServletFactory.register(HelloWebSocketServer.class);
    }
}

启动类

环境的servlets中添加自己的WebSocketServlet，并指定url

package com.jsq.hibernate;

import com.jsq.hibernate.config.HelloWorldConfiguration;
import com.jsq.hibernate.websocket.HelloWebSocketServlet;
import io.dropwizard.Application;
import io.dropwizard.db.PooledDataSourceFactory;
import io.dropwizard.setup.Bootstrap;
import io.dropwizard.setup.Environment;
import java.net.URL;

public class HelloWorldApplication extends Application<HelloWorldConfiguration> {

    public static void main(String[] args) throws Exception {
        URL yml = HelloWorldApplication.class.getClassLoader().getResource("server.yml");
        String configFile = yml.getFile();
        System.out.println(configFile);
        args = new String[]{"server", configFile};
        new HelloWorldApplication().run(args);
    }

    @Override
    public void run(HelloWorldConfiguration helloWorldConfiguration, Environment environment) {
        environment.servlets().addServlet("ws", new HelloWebSocketServlet()).addMapping("/ws/hello");
    }

    @Override
    public String getName() {
        return "hello-world";
    }

    @Override
    public void initialize(Bootstrap<HelloWorldConfiguration> bootstrap) {
    }
}