[英]How do I get the remote IP address for an Rsocket connection in SpringBoot
我正在尝试获取连接到 RSocket+SpringBoot 网络服务器的浏览器的远程 IP。 连接是 WebSocket 上的 RSocket。
网络服务器是 Java-8,SpringBoot-2,在 WebSocket 上使用 RSocket 并将 RequestStreams 发送到浏览器。 我将 SpringBoot 自动配置用于 RSocket 设置,因此服务器端的代码非常少 - 见下文。
下面代码中的@Headers and MessageHeader
只是为了看看他们是否有任何可能导致远程 IP 的东西,没有其他原因他们在那里。
我已经搜索了 web 寻找答案 - 很多 http,一些用于 websockets,零用于 RSocket。 这个 - https://github.com/rsocket/rsocket-java/issues/735 - 看起来很有希望,但无法处理 DuplexConnection,所以那里没有雪茄。
有任何想法吗? 谢谢 !
应用程序.yml:
spring.rsocket.server:
mapping-path: /rsocket-test
transport: websocket
server.port: 8080
测试控制器.java
/**
* TODO: get the remote IP address and log.
* Options:
* 1. @see <a href="https://github.com/rsocket/rsocket-java/issues/735">Ability to intercept requests and access channel information such as remote address</a>.
* 2. Inject IP in header by nginx. See if it shows up in the @Headers param here.
* 3. Browser gets its public IP and adds it to the request object. Doable, but lame
* 4. (Unlikely) Find a way to access thru this chain of private members: headers.req.rsocket().source.connection.source.connection.connection.channel.remoteAddress
*/
@MessageMapping("subscribe-topic")
public Flux<StreamingEvent> subscribeToEventStream(
@Headers Map<String,Object> hdrs,
MessageHeaders mh,
testRequest request) {
return testService.subscribeTopic(request.getRequestId(), request.getTopic());
}
客户端的 ip 地址在DuplexConnection
class 中可用。 您可以像这样向RSocketServer
添加拦截器:
@Bean
public RSocketServerCustomizer ipCustomizer() {
return rSocketServer -> rSocketServer.interceptors(registry -> registry.forConnection(new ConnectionInterceptor()));
}
其中ConnectionInterceptor
是:
static class ConnectionInterceptor implements DuplexConnectionInterceptor {
@Override
public DuplexConnection apply(Type type, DuplexConnection duplexConnection) {
SocketAddress socketAddress = duplexConnection.remoteAddress();
if (socketAddress instanceof InetSocketAddress) {
InetSocketAddress iso = (InetSocketAddress) socketAddress;
// Here is the ip: iso.getHostString()
}
return duplexConnection;
}
}
在我的场景中,我想将远程主机和原始 HTTP 请求标头传递给我的@MessageMapping
处理程序。 我想出的最好方法是创建自己的NettyRouteProvider
并使用所需数据丰富订阅上下文:
@Component
public class MyRouteProvider implements NettyRouteProvider {
private final String mappingPath;
private final SocketAcceptor socketAcceptor;
private final List<RSocketServerCustomizer> customizers;
@Autowired
MyRouteProvider(RSocketProperties properties, RSocketMessageHandler messageHandler,
ObjectProvider<RSocketServerCustomizer> customizers) {
this.mappingPath = properties.getServer().getMappingPath();
this.socketAcceptor = messageHandler.responder();
this.customizers = customizers.orderedStream().collect(Collectors.toList());
}
@Override
public HttpServerRoutes apply(HttpServerRoutes httpServerRoutes) {
RSocketServer server = RSocketServer.create(this.socketAcceptor);
this.customizers.forEach((customizer) -> customizer.customize(server));
ServerTransport.ConnectionAcceptor connectionAcceptor = server.asConnectionAcceptor();
return httpServerRoutes.ws(this.mappingPath, (in, out) ->
connectionAcceptor.apply(new WebsocketDuplexConnection((Connection) in)).then(out.neverComplete())
.contextWrite(context -> {
String ip = in.headers().get("x-forwarded-for");
SocketAddress remoteAddress = ((Connection) in).channel().remoteAddress();
if(in.headers().contains("x-forwarded-for")){
return context
.put("remoteHost",remoteAddress)
.put(HttpHeaders.class, in.headers())
.put("client-ip", ip);
}
return context
.put("remoteHost",remoteAddress)
.put(HttpHeaders.class, in.headers());
}));
}
}
在此之后,您将能够在任何 @MessageMapping 方法中从上下文中获取数据,例如:
@MessageMapping("remote-host")
fun remoteHost(): Mono<SocketAddress> {
return Mono.deferContextual { context ->
Mono.just(context.get<SocketAddress>("remoteHost"))
}
}
我与这个问题斗争了很长一段时间。 我不喜欢接受的答案,因为无法将信息传递给处理程序。 诺金娜的回答奏效了,但仅限于最初的请求。 在后续请求中,上下文将不包含地址。 花了几个小时后,我刚刚回到了良好的反思中,并想出了这个:
import io.rsocket.DuplexConnection;
import io.rsocket.RSocket;
import org.springframework.messaging.rsocket.RSocketRequester;
import org.springframework.util.ReflectionUtils;
import java.lang.reflect.Field;
import java.net.SocketAddress;
public class RSocketUtil {
public static DuplexConnection getConnectionFromRequester(RSocketRequester requester) {
RSocket rsocket = requester.rsocket();
Field connectionField = ReflectionUtils.findField(rsocket.getClass(), "connection");
connectionField.setAccessible(true);
return (DuplexConnection) ReflectionUtils.getField(connectionField, rsocket);
}
public static SocketAddress getRemoteAddressFromRequester(RSocketRequester requester) {
DuplexConnection connection = getConnectionFromRequester(requester);
return connection.remoteAddress();
}
}
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.