17
17
package io .rsocket .broker .client .spring ;
18
18
19
19
import java .net .URI ;
20
- import java .util .Collections ;
21
- import java .util .List ;
22
20
23
21
import io .rsocket .RSocket ;
24
22
import io .rsocket .broker .common .spring .ClientTransportFactory ;
25
23
import io .rsocket .broker .common .spring .DefaultClientTransportFactory ;
26
24
import io .rsocket .broker .common .spring .MimeTypes ;
27
25
import io .rsocket .broker .frames .RouteSetup ;
28
- import io .rsocket .loadbalance .LoadbalanceTarget ;
29
- import io .rsocket .loadbalance .RoundRobinLoadbalanceStrategy ;
30
26
import io .rsocket .transport .ClientTransport ;
31
27
import reactor .core .Disposable ;
32
- import reactor .core .publisher .Flux ;
33
28
import reactor .core .publisher .Sinks ;
34
29
import reactor .core .publisher .Sinks .One ;
35
30
36
31
import org .springframework .beans .factory .ObjectProvider ;
32
+ import org .springframework .beans .factory .annotation .Autowired ;
37
33
import org .springframework .boot .autoconfigure .AutoConfigureAfter ;
38
34
import org .springframework .boot .autoconfigure .AutoConfigureBefore ;
39
35
import org .springframework .boot .autoconfigure .condition .ConditionalOnClass ;
@@ -73,7 +69,7 @@ public BrokerClientProperties brokerClientProperties() {
73
69
@ ConditionalOnMissingBean
74
70
public BrokerRSocketRequesterBuilder brokerRSocketRequesterBuilder (
75
71
RSocketConnectorConfigurer configurer , RSocketStrategies strategies ,
76
- BrokerClientProperties properties ) {
72
+ BrokerClientProperties properties , @ Autowired ( required = false ) BrokerClientMimeTypeSupplier brokerClientMimeTypeSupplier ) {
77
73
RouteSetup .Builder routeSetup = RouteSetup .from (properties .getRouteId (),
78
74
properties .getServiceName ());
79
75
properties .getTags ().forEach ((key , value ) -> {
@@ -90,6 +86,7 @@ else if (key.getKey() != null) {
90
86
91
87
RSocketRequester .Builder builder = RSocketRequester .builder ()
92
88
.setupMetadata (routeSetup .build (), MimeTypes .BROKER_FRAME_MIME_TYPE )
89
+ .dataMimeType (brokerClientMimeTypeSupplier != null ? brokerClientMimeTypeSupplier .get () : null )
93
90
.rsocketStrategies (strategies ).rsocketConnector (configurer );
94
91
95
92
//TODO: BrokerRequesterBuilderCustomizer
0 commit comments