要建立RSocket连接，客户端必须发送第一个称为SETUP帧的帧。您可以使用Spring @ConnectMapping annotation捕获它。

@ConnectMapping
    fun onConnect(rSocketRequester: RSocketRequester, clientId: String) {
        if (clientId == "wrong_client") rSocketRequester.rsocket().dispose()   //to reject connection
    }

字符串
要拒绝连接，请使用如上所示的rSocketRequester.rsocket().dispose()。注意，使用SETUP帧，您可以像往常一样从客户端传递负载和元数据。

您可以通过实现WebSocketHandler拦截第一个请求，并使用ServerHttpRequest访问请求的头和查询参数，并使用ServerHttpResponse修改响应以返回101状态代码。

import io.rsocket.RSocket;
import io.rsocket.core.RSocketServer;
import io.rsocket.transport.netty.server.CloseableChannel;
import io.rsocket.transport.netty.server.WebsocketServerTransport;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.http.HttpStatus;
import org.springframework.web.reactive.HandlerMapping;
import org.springframework.web.reactive.handler.SimpleUrlHandlerMapping;
import org.springframework.web.reactive.socket.WebSocketHandler;
import org.springframework.web.reactive.socket.server.support.WebSocketHandlerAdapter;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpResponse;
import java.util.HashMap;
import java.util.Map;
@Configuration
public class WebSocketConfig {
    @Bean
    public RSocket rSocket() {
        // Your RSocket bean configuration
        return RSocketServer.create()
                .bind(WebsocketServerTransport.create("localhost", 8080))
                .start()
                .block();
    }
    @Bean
    public HandlerMapping webSocketHandlerMapping() {
        Map<String, WebSocketHandler> map = new HashMap<>();
        map.put("/rsocket", new CustomWebSocketHandler());
        SimpleUrlHandlerMapping mapping = new SimpleUrlHandlerMapping();
        mapping.setOrder(1);
        mapping.setUrlMap(map);
        return mapping;
    }
    @Bean
    public WebSocketHandlerAdapter handlerAdapter() {
        return new WebSocketHandlerAdapter();
    }
    private static class CustomWebSocketHandler implements WebSocketHandler {
        @Override
        public CloseableChannel handle(ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) {
            // Access headers and query parameters here
            String headerValue = serverHttpRequest.getHeaders().getFirst("your-header");
            String queryParamValue = serverHttpRequest.getQueryParams().getFirst("your-query-param");
            // Modify the response to return 101 status code
            serverHttpResponse.setStatusCode(HttpStatus.SWITCHING_PROTOCOLS);
            // Implement your logic here
            return null; // Return the CloseableChannel or handle as needed
        }
    }
}

字符串