Examples with Payload io.rsocket.Payload used on opensource projects

Example 1 with Payload

use of io.rsocket.Payload in project spring-security by spring-projects.

the class PayloadInterceptorRSocketTests method metadataPushWhenSecurityContextThenDelegateContext.

@Test
public void metadataPushWhenSecurityContextThenDelegateContext() {
    TestingAuthenticationToken authentication = new TestingAuthenticationToken("user", "password");
    given(this.interceptor.intercept(any(), any())).willAnswer(withAuthenticated(authentication));
    given(this.delegate.metadataPush(any())).willReturn(this.voidResult.mono());
    RSocket assertAuthentication = new RSocketProxy(this.delegate) {

        @Override
        public Mono<Void> metadataPush(Payload payload) {
            return assertAuthentication(authentication).flatMap((a) -> super.metadataPush(payload));
        }
    };
    PayloadInterceptorRSocket interceptor = new PayloadInterceptorRSocket(assertAuthentication, Arrays.asList(this.interceptor), this.metadataMimeType, this.dataMimeType);
    StepVerifier.create(interceptor.metadataPush(this.payload)).verifyComplete();
    verify(this.interceptor).intercept(this.exchange.capture(), any());
    assertThat(this.exchange.getValue().getPayload()).isEqualTo(this.payload);
    verify(this.delegate).metadataPush(this.payload);
    this.voidResult.assertWasSubscribed();
}

Example 2 with Payload

use of io.rsocket.Payload in project spring-security by spring-projects.

the class PayloadInterceptorRSocketTests method fireAndForgetWhenSecurityContextThenDelegateContext.

@Test
public void fireAndForgetWhenSecurityContextThenDelegateContext() {
    TestingAuthenticationToken authentication = new TestingAuthenticationToken("user", "password");
    given(this.interceptor.intercept(any(), any())).willAnswer(withAuthenticated(authentication));
    given(this.delegate.fireAndForget(any())).willReturn(Mono.empty());
    RSocket assertAuthentication = new RSocketProxy(this.delegate) {

        @Override
        public Mono<Void> fireAndForget(Payload payload) {
            return assertAuthentication(authentication).flatMap((a) -> super.fireAndForget(payload));
        }
    };
    PayloadInterceptorRSocket interceptor = new PayloadInterceptorRSocket(assertAuthentication, Arrays.asList(this.interceptor), this.metadataMimeType, this.dataMimeType);
    interceptor.fireAndForget(this.payload).block();
    verify(this.interceptor).intercept(this.exchange.capture(), any());
    assertThat(this.exchange.getValue().getPayload()).isEqualTo(this.payload);
    verify(this.delegate).fireAndForget(this.payload);
}

Example 3 with Payload

use of io.rsocket.Payload in project spring-security by spring-projects.

the class PayloadInterceptorRSocketTests method requestChannelWhenInterceptorCompletesThenAllPayloadsRetained.

// gh-9345
@Test
public void requestChannelWhenInterceptorCompletesThenAllPayloadsRetained() {
    ExecutorService executors = Executors.newSingleThreadExecutor();
    Payload payload = ByteBufPayload.create("data");
    Payload payloadTwo = ByteBufPayload.create("moredata");
    Payload payloadThree = ByteBufPayload.create("stillmoredata");
    Context ctx = Context.empty();
    Flux<Payload> payloads = this.payloadResult.flux();
    given(this.interceptor.intercept(any(), any())).willReturn(Mono.empty()).willReturn(Mono.error(() -> new AccessDeniedException("Access Denied")));
    given(this.delegate.requestChannel(any())).willAnswer((invocation) -> {
        Flux<Payload> input = invocation.getArgument(0);
        return Flux.from(input).switchOnFirst((signal, innerFlux) -> innerFlux.map(Payload::getDataUtf8).transform((data) -> Flux.<String>create((emitter) -> {
            Runnable run = () -> data.subscribe(new CoreSubscriber<String>() {

                @Override
                public void onSubscribe(Subscription s) {
                    s.request(3);
                }

                @Override
                public void onNext(String s) {
                    emitter.next(s);
                }

                @Override
                public void onError(Throwable t) {
                    emitter.error(t);
                }

                @Override
                public void onComplete() {
                    emitter.complete();
                }
            });
            executors.execute(run);
        })).map(DefaultPayload::create));
    });
    PayloadInterceptorRSocket interceptor = new PayloadInterceptorRSocket(this.delegate, Arrays.asList(this.interceptor), this.metadataMimeType, this.dataMimeType, ctx);
    StepVerifier.create(interceptor.requestChannel(payloads).doOnDiscard(Payload.class, Payload::release)).then(() -> this.payloadResult.assertSubscribers()).then(() -> this.payloadResult.emit(payload, payloadTwo, payloadThree)).assertNext((next) -> assertThat(next.getDataUtf8()).isEqualTo(payload.getDataUtf8())).verifyError(AccessDeniedException.class);
    verify(this.interceptor, times(2)).intercept(this.exchange.capture(), any());
    assertThat(this.exchange.getValue().getPayload()).isEqualTo(payloadTwo);
    verify(this.delegate).requestChannel(any());
}

Example 4 with Payload

use of io.rsocket.Payload in project spring-framework by spring-projects.

the class MessagingRSocket method handleAndReply.

@SuppressWarnings("deprecation")
private Flux<Payload> handleAndReply(Payload firstPayload, FrameType frameType, Flux<Payload> payloads) {
    AtomicReference<Flux<Payload>> responseRef = new AtomicReference<>();
    MessageHeaders headers = createHeaders(firstPayload, frameType, responseRef);
    AtomicBoolean read = new AtomicBoolean();
    Flux<DataBuffer> buffers = payloads.map(this::retainDataAndReleasePayload).doOnSubscribe(s -> read.set(true));
    Message<Flux<DataBuffer>> message = MessageBuilder.createMessage(buffers, headers);
    return Mono.defer(() -> this.messageHandler.handleMessage(message)).doFinally(s -> {
        // Subscription should have happened by now due to ChannelSendOperator
        if (!read.get()) {
            firstPayload.release();
        }
    }).thenMany(Flux.defer(() -> responseRef.get() != null ? responseRef.get() : Mono.error(new IllegalStateException("Expected response"))));
}

Example 5 with Payload

use of io.rsocket.Payload in project spring-framework by spring-projects.

the class DefaultRSocketRequesterBuilder method initConnector.

@SuppressWarnings("deprecation")
private RSocketConnector initConnector(List<RSocketConnectorConfigurer> connectorConfigurers, MimeType metaMimeType, MimeType dataMimeType, RSocketStrategies rsocketStrategies) {
    RSocketConnector connector = RSocketConnector.create();
    connectorConfigurers.forEach(c -> c.configure(connector));
    if (rsocketStrategies.dataBufferFactory() instanceof NettyDataBufferFactory) {
        connector.payloadDecoder(PayloadDecoder.ZERO_COPY);
    }
    connector.metadataMimeType(metaMimeType.toString());
    connector.dataMimeType(dataMimeType.toString());
    Mono<Payload> setupPayloadMono = getSetupPayload(dataMimeType, metaMimeType, rsocketStrategies);
    if (setupPayloadMono != EMPTY_SETUP_PAYLOAD) {
        connector.setupPayload(setupPayloadMono);
    }
    return connector;
}