|
29 | 29 | import org.springframework.context.ApplicationEventPublisherAware;
|
30 | 30 | import org.springframework.core.io.buffer.DataBuffer;
|
31 | 31 | import org.springframework.lang.Nullable;
|
32 |
| -import org.springframework.messaging.MessageHeaders; |
33 |
| -import org.springframework.messaging.handler.DestinationPatternsMessageCondition; |
34 | 32 | import org.springframework.messaging.rsocket.RSocketRequester;
|
35 |
| -import org.springframework.messaging.rsocket.annotation.support.RSocketRequesterMethodArgumentResolver; |
36 | 33 | import org.springframework.util.Assert;
|
37 |
| -import org.springframework.util.RouteMatcher; |
38 | 34 |
|
39 | 35 | import io.rsocket.RSocketFactory;
|
40 | 36 | import io.rsocket.SocketAcceptor;
|
@@ -181,17 +177,10 @@ public SocketAcceptor serverAcceptor() {
|
181 | 177 | return (setupPayload, sendingRSocket) -> {
|
182 | 178 | IntegrationRSocket rsocket = createRSocket(setupPayload, sendingRSocket);
|
183 | 179 | return rsocket.handleConnectionSetupPayload(setupPayload)
|
184 |
| - .doOnNext((message) -> { |
185 |
| - MessageHeaders messageHeaders = message.getHeaders(); |
186 |
| - DataBuffer dataBuffer = message.getPayload(); |
187 |
| - String destination = |
188 |
| - messageHeaders.get(DestinationPatternsMessageCondition.LOOKUP_DESTINATION_HEADER, |
189 |
| - RouteMatcher.Route.class) |
190 |
| - .value(); |
| 180 | + .doOnNext((dataBuffer) -> { |
| 181 | + String destination = rsocket.getDestination(setupPayload); |
191 | 182 | Object rsocketRequesterKey = this.clientRSocketKeyStrategy.apply(destination, dataBuffer);
|
192 |
| - RSocketRequester rsocketRequester = |
193 |
| - messageHeaders.get(RSocketRequesterMethodArgumentResolver.RSOCKET_REQUESTER_HEADER, |
194 |
| - RSocketRequester.class); |
| 183 | + RSocketRequester rsocketRequester = rsocket.getRequester(); |
195 | 184 | this.clientRSocketRequesters.put(rsocketRequesterKey, rsocketRequester);
|
196 | 185 | RSocketConnectedEvent rSocketConnectedEvent =
|
197 | 186 | new RSocketConnectedEvent(rsocket, destination, dataBuffer, rsocketRequester);
|
|
0 commit comments