Remove nested flatMap in RSocketFactory

Author: rdegnan

rsocket-core/src/main/java/io/rsocket/RSocketFactory.java

@@ -228,28 +228,21 @@ public Mono<RSocket> start() {
                           ackTimeout,
                           missedAcks);
 
-                  Mono<RSocket> wrappedRSocketClient =
-                      Mono.just(rSocketClient).map(plugins::applyClient);
-
-                  DuplexConnection finalConnection = connection;
-                  return wrappedRSocketClient.flatMap(
-                      wrappedClientRSocket -> {
-                        RSocket unwrappedServerSocket = acceptor.get().apply(wrappedClientRSocket);
-
-                        Mono<RSocket> wrappedRSocketServer =
-                            Mono.just(unwrappedServerSocket).map(plugins::applyServer);
-
-                        return wrappedRSocketServer
-                            .doOnNext(
-                                rSocket ->
-                                    new RSocketServer(
-                                        multiplexer.asServerConnection(),
-                                        rSocket,
-                                        frameDecoder,
-                                        errorConsumer))
-                            .then(finalConnection.sendOne(setupFrame))
-                            .then(wrappedRSocketClient);
-                      });
+                  RSocket wrappedRSocketClient = plugins.applyClient(rSocketClient);
+
+                  RSocket unwrappedServerSocket = acceptor.get().apply(wrappedRSocketClient);
+
+                  RSocket wrappedRSocketServer = plugins.applyServer(unwrappedServerSocket);
+
+                  RSocketServer rSocketServer = new RSocketServer(
+                      multiplexer.asServerConnection(),
+                      wrappedRSocketServer,
+                      frameDecoder,
+                      errorConsumer);
+
+                  return connection
+                      .sendOne(setupFrame)
+                      .thenReturn(wrappedRSocketClient);
                 });
       }
     }