Java Code Examples for reactor.core.publisher.Mono

@Test
public void sendsStreamingRequests() throws ExecutionException, InterruptedException {
    LiveHttpResponse response = new StyxHttpClient.Builder()
            .build()
            .streaming()
            .send(httpRequest.stream())
            .get();

    assertThat(response.status(), is(OK));

    Mono.from(response.aggregate(10000)).block();

    server.verify(
            getRequestedFor(urlEqualTo("/"))
                    .withHeader("Host", equalTo(hostString(server.port())))
    );
}
 

@Override
public Flux<MultiValueResponse<HGetCommand, ByteBuffer>> hMGet(Publisher<HGetCommand> commands) {
    return execute(commands, command -> {
        
        Assert.notNull(command.getKey(), "Key must not be null!");
        Assert.notNull(command.getFields(), "Fields must not be null!");
        
        byte[] keyBuf = toByteArray(command.getKey());
        List<Object> args = new ArrayList<Object>(command.getFields().size() + 1);
        args.add(keyBuf);
        args.addAll(command.getFields().stream().map(buf -> toByteArray(buf)).collect(Collectors.toList()));
        Mono<List<byte[]>> m = read(keyBuf, ByteArrayCodec.INSTANCE, HMGET, args.toArray());
        return m.map(v -> {
            List<ByteBuffer> values = v.stream().map(array -> {
                if (array != null) {
                    return ByteBuffer.wrap(array);
                }
                return null;
            }).collect(Collectors.toList());
            return new MultiValueResponse<>(command, values);
        });
    });
}
 

@HystrixCommand(commandKey = "request-token",groupKey = "auth-operations",commandProperties = {
    @HystrixProperty(name="circuitBreaker.requestVolumeThreshold",value="10"),
    @HystrixProperty(name = "circuitBreaker.errorThresholdPercentage", value = "10"),
    @HystrixProperty(name="circuitBreaker.sleepWindowInMilliseconds",value="10000"),
    @HystrixProperty(name = "execution.isolation.thread.timeoutInMilliseconds", value = "1000"),
    @HystrixProperty(name = "metrics.rollingStats.timeInMilliseconds", value = "10000")
})
public Mono<AccessToken> token(Credentials credentials) {
  val authorizationHeader = Base64Utils.encodeToString(( credentials.getClientId() + ":" + credentials.getClientSecret()).getBytes());
  return discoveryService.serviceAddressFor(this.authService).next().flatMap(address ->
      this.webClient.mutate().baseUrl(address + "/" + this.authServiceApiPath).build()
      .post()
      .contentType(MediaType.APPLICATION_FORM_URLENCODED)
      .header("Authorization","Basic " + authorizationHeader)
      .body(BodyInserters.fromFormData("grant_type", "client_credentials"))
      .retrieve()
      .onStatus(HttpStatus::is4xxClientError, clientResponse ->
          Mono.error(new RuntimeException("Invalid call"))
      ).onStatus(HttpStatus::is5xxServerError, clientResponse ->
      Mono.error(new RuntimeException("Error on server"))
  ).bodyToMono(AccessToken.class));
}
 

/**
 * Sample usage: curl $HOST:$PORT/product-composite/1
 *
 * @param productId
 * @return the composite product info, if found, else null
 */
@ApiOperation(
    value = "${api.product-composite.get-composite-product.description}",
    notes = "${api.product-composite.get-composite-product.notes}")
@ApiResponses(value = {
    @ApiResponse(code = 400, message = "Bad Request, invalid format of the request. See response message for more information."),
    @ApiResponse(code = 404, message = "Not found, the specified id does not exist."),
    @ApiResponse(code = 422, message = "Unprocessable entity, input parameters caused the processing to fail. See response message for more information.")
})
@GetMapping(
    value    = "/product-composite/{productId}",
    produces = "application/json")
Mono<ProductAggregate> getCompositeProduct(
    @PathVariable int productId,
    @RequestParam(value = "delay", required = false, defaultValue = "0") int delay,
    @RequestParam(value = "faultPercent", required = false, defaultValue = "0") int faultPercent
);
 

@Test
public void testSingle() {
	// tag::example-mono[]
	Employee e = new Employee();
	e.setFirstName("Bilbo");
	Example<Employee> example = Example.of(e);
	// end::example-mono[]

	// tag::query-mono[]
	Mono<Employee> singleEmployee = repository.findOne(example);
	// end::query-mono[]

	StepVerifier.create(singleEmployee)
		.expectNextMatches(employee -> {
			assertThat(employee).hasNoNullFieldsOrProperties();
			assertThat(employee.getFirstName()).isEqualTo("Bilbo");
			assertThat(employee.getLastName()).isEqualTo("Baggins");
			assertThat(employee.getRole()).isEqualTo("burglar");
			return true;
		})
		.expectComplete()
		.verify();
}
 

@Test
public void ofPublisher() {
	Flux<String> body = Flux.just("foo");
	BodyInserter<Flux<String>, ReactiveHttpOutputMessage> inserter = BodyInserters.fromPublisher(body, String.class);

	MockServerHttpResponse response = new MockServerHttpResponse();
	Mono<Void> result = inserter.insert(response, this.context);
	StepVerifier.create(result).expectComplete().verify();
	StepVerifier.create(response.getBody())
			.consumeNextWith(buf -> {
				String actual = DataBufferTestUtils.dumpString(buf, UTF_8);
				Assert.assertEquals("foo", actual);
			})
			.expectComplete()
			.verify();
}
 

@Test
default void returnGeneratedValues() {

    getJdbcOperations().execute("DROP TABLE test");
    getJdbcOperations().execute(getCreateTableWithAutogeneratedKey());

    Mono.from(getConnectionFactory().create())
        .flatMapMany(connection -> {
            Statement statement = connection.createStatement(getInsertIntoWithAutogeneratedKey());

            statement.returnGeneratedValues();

            return Flux.from(statement
                .execute())
                .concatWith(close(connection)).flatMap(it -> it.map((row, rowMetadata) -> row.get(0)));
        })
        .as(StepVerifier::create)
        .expectNextCount(1)
        .verifyComplete();
}
 

@Test
public void testUnSubscribe() {
    RedissonConnectionFactory factory = new RedissonConnectionFactory(redisson);
    ReactiveRedisConnection connection = factory.getReactiveConnection();
    Mono<ReactiveSubscription> s = connection.pubSubCommands().createSubscription();
    AtomicReference<byte[]> msg = new AtomicReference<byte[]>();
    ReactiveSubscription ss = s.block();

    ss.subscribe(ByteBuffer.wrap("test".getBytes())).block();
    ss.receive().doOnEach(message -> {
        msg.set(message.get().getMessage().array());
    }).subscribe();
    
    connection.pubSubCommands().publish(ByteBuffer.wrap("test".getBytes()), ByteBuffer.wrap("msg".getBytes())).block();
    Awaitility.await().atMost(Duration.ONE_SECOND)
                .until(() -> Arrays.equals("msg".getBytes(), msg.get()));
    
    ss.unsubscribe();
    
    connection.pubSubCommands().publish(ByteBuffer.wrap("test".getBytes()), ByteBuffer.wrap("msg".getBytes())).block();
    
    
}
 

@Test()
public void invokeActorMethodWithoutDataWithVoidReturnType() {
  final DaprClient daprClient = mock(DaprClient.class);
  when(daprClient.invokeActorMethod(anyString(), anyString(), anyString(), Mockito.isNull()))
      .thenReturn(Mono.empty());

  final ActorProxy actorProxy = new ActorProxyImpl(
      "myActorType",
      new ActorId("100"),
      new DefaultObjectSerializer(),
      daprClient);

  Mono<Void> result = actorProxy.invokeActorMethod("getData");
  Void emptyResponse = result.block();
  Assert.assertNull(emptyResponse);
}
 

@Test
void hasInboxShouldBeFalseWhenINBOXIsNotCreated() throws Exception {
    session = mailboxManager.createSystemSession(USER_1);
    mailboxManager.startProcessingRequest(session);

    assertThat(Mono.from(mailboxManager.hasInbox(session)).block()).isFalse();
}
 

private Mono<Inconsistency> detectInconsistency(CassandraIdAndPath pathRegistration) {
    return mailboxDAO.retrieveMailbox(pathRegistration.getCassandraId())
        .map(mailbox -> {
            if (mailbox.generateAssociatedPath().equals(pathRegistration.getMailboxPath())) {
                return NO_INCONSISTENCY;
            }
            // Mailbox references another path
            return new ConflictingEntryInconsistency(mailbox, pathRegistration);
        })
        .defaultIfEmpty(new OrphanMailboxPathDAOEntry(pathRegistration));
}
 

@Test
default void readShouldReturnEmptyCachedByteArray() {
    BlobId blobId = blobIdFactory().randomId();
    byte[] emptyByteArray = new byte[] {};

    Mono.from(testee().cache(blobId, emptyByteArray)).block();

    assertThat(new ByteArrayInputStream(Mono.from(testee().read(blobId)).block()))
        .hasSameContentAs(new ByteArrayInputStream(emptyByteArray));
}
 

@Test
public void contentLength() {
	String body = "foo";
	Mono<EntityResponse<String>> result = EntityResponse.fromObject(body).contentLength(42).build();
	StepVerifier.create(result)
			.expectNextMatches(response -> Long.valueOf(42).equals(response.headers().getContentLength()))
			.expectComplete()
			.verify();
}
 

@Override
public Mono<Void> updateAgentInstanceAttributes(String instanceId, Map<String, String> attributes) {
    return stub.updateAgentInstanceAttributes(AgentInstanceAttributesUpdate.newBuilder()
            .setAgentInstanceId(instanceId)
            .putAllAttributes(attributes)
            .build()
    );
}
 

@Override
public Flux<ReactiveRedisConnection.NumericResponse<ReactiveRedisConnection.KeyCommand, Long>> xTrim(Publisher<TrimCommand> publisher) {
    return execute(publisher, command -> {

        Assert.notNull(command.getKey(), "Key must not be null!");
        Assert.notNull(command.getCount(), "Count must not be null!");

        byte[] k = toByteArray(command.getKey());

        Mono<Long> m = write(k, StringCodec.INSTANCE, RedisCommands.XTRIM, k, "MAXLEN", command.getCount());
        return m.map(v -> new ReactiveRedisConnection.NumericResponse<>(command, v));
    });
}
 

@Override
public Flux<NumericResponse<SDiffStoreCommand, Long>> sDiffStore(Publisher<SDiffStoreCommand> commands) {
    return execute(commands, command -> {

        Assert.notNull(command.getKeys(), "Keys must not be null!");
        Assert.notNull(command.getKey(), "Destination key must not be null!");

        List<Object> args = new ArrayList<Object>(command.getKeys().size() + 1);
        args.add(toByteArray(command.getKey()));
        args.addAll(command.getKeys().stream().map(v -> toByteArray(v)).collect(Collectors.toList()));

        Mono<Long> m = write((byte[])args.get(0), StringCodec.INSTANCE, RedisCommands.SDIFFSTORE, args.toArray());
        return m.map(v -> new NumericResponse<>(command, v));
    });
}
 

@Override
public void subscribe(CoreSubscriber<? super T> actual) {
    long waitDuration = rateLimiter.reservePermission();
    if (waitDuration >= 0) {
        if (waitDuration > 0) {
            Mono.delay(Duration.ofNanos(waitDuration))
                .subscribe(delay -> source.subscribe(new RateLimiterSubscriber<>(actual)));
        } else {
            source.subscribe(new RateLimiterSubscriber<>(actual));
        }
    } else {
        Operators.error(actual, createRequestNotPermitted(rateLimiter));
    }
}
 

@Override
public Mono<HttpResponse> forward(HttpRequest request, HttpRequestExecution execution) {
    logStart(execution.getMappingName());
    Retry retry = getRegistry().retry(execution.getMappingName());
    setupMetrics(registry -> createMetrics(registry, execution.getMappingName()));
    return execution.execute(request)
            .transform(of(retry))
            .doOnSuccess(response -> logEnd(execution.getMappingName()));
}
 

@Override
public Mono<RSocket> accept(ConnectionSetupPayload setupPayload, RSocket sendingRSocket) {
	MessagingRSocket rsocket = createRSocket(sendingRSocket);

	// Allow handling of the ConnectionSetupPayload via @MessageMapping methods.
	// However, if the handling is to make requests to the client, it's expected
	// it will do so decoupled from the handling, e.g. via .subscribe().
	return rsocket.handleConnectionSetupPayload(setupPayload).then(Mono.just(rsocket));
}
 

/**
 * Apply {@link Flux#reduce(Object, BiFunction) reduce} on the body, count
 * the number of bytes produced, release data buffers without writing, and
 * set the {@literal Content-Length} header.
 */
@Override
public final Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {
	return Flux.from(body)
			.reduce(0, (current, buffer) -> {
				int next = current + buffer.readableByteCount();
				DataBufferUtils.release(buffer);
				return next;
			})
			.doOnNext(count -> getHeaders().setContentLength(count))
			.then();
}
 

@Override
public Mono<HttpResponse> forward(HttpRequest request, HttpRequestExecution execution) {
    logStart(execution.getMappingName());
    rewritePath(request.url(), request::setUrl);
    return execution.execute(request)
            .doOnSuccess(response -> logEnd(execution.getMappingName()));
}
 

@ParameterizedTest
@MethodSource("allPools")
void allocatorErrorInWarmupIsPropagated(Function<PoolBuilder<Object, ?>, AbstractPool<Object>> configAdjuster) {
	final PoolBuilder<Object, ?> builder = PoolBuilder
			.from(Mono.error(new IllegalStateException("boom")))
			.sizeBetween(1, 1)
			.evictionPredicate((poolable, metadata) -> true);
	AbstractPool<Object> pool = configAdjuster.apply(builder);

	StepVerifier.create(pool.warmup())
	            .verifyErrorSatisfies(e -> assertThat(e).isInstanceOf(IllegalStateException.class)
	                                                    .hasMessage("boom"));
}
 

public boolean testMailboxExists(Username username, MailboxName mailboxName) throws MailboxException, UsersRepositoryException {
    usernamePreconditions(username);
    MailboxSession mailboxSession = mailboxManager.createSystemSession(username);
    MailboxPath mailboxPath = MailboxPath.forUser(username, mailboxName.asString())
        .assertAcceptable(mailboxSession.getPathDelimiter());
    return Mono.from(mailboxManager.mailboxExists(mailboxPath, mailboxSession))
        .block();
}
 

public Flux<String> serviceAddressFor(String service) {
  return Flux.defer(() ->  Flux.just(this.dClient.getInstances(service)).flatMap(srv ->
      Mono.just(this.lbClient.choose(service))
  ).flatMap(serviceInstance ->
      Mono.just(serviceInstance.getUri().toString())
  ));
}
 

@Override
public Mono<Map<StorPool.Key, Tuple2<SpaceInfo, List<ApiCallRc>>>> fetchThinFreeSpaceInfo(Set<NodeName> nodesFilter)
{
    return scopeRunner.fluxInTransactionalScope(
        "Fetch thin capacity info",
        lockGuardFactory.buildDeferred(LockType.READ, LockObj.NODES_MAP, LockObj.STOR_POOL_DFN_MAP),
        () -> assembleRequests(nodesFilter).flatMap(this::parseFreeSpaces)
    )
        .collectMap(
        t -> t.getT1(),
        t -> t.getT2()
    );
}
 

@Test
default void storeShouldOverrideOldRecord() {
    MessageId messageId = newMessageId();
    Mono.from(testee().store(messageId, MESSAGE_FAST_VIEW_PRECOMPUTED_PROPERTIES_1))
        .block();

    Mono.from(testee().store(messageId, MESSAGE_FAST_VIEW_PRECOMPUTED_PROPERTIES_2))
        .block();

    assertThat(Mono.from(testee().retrieve(messageId)).block())
        .isEqualTo(MESSAGE_FAST_VIEW_PRECOMPUTED_PROPERTIES_2);
}
 

@Override
public Mono<InsertionId> store(Group registeredGroup, Event failDeliveredEvent) {
    Preconditions.checkArgument(registeredGroup != null, REGISTERED_GROUP_CANNOT_BE_NULL);
    Preconditions.checkArgument(failDeliveredEvent != null, FAIL_DELIVERED_EVENT_CANNOT_BE_NULL);

    InsertionId insertionId = InsertionId.random();
    return cassandraEventDeadLettersDAO.store(registeredGroup, failDeliveredEvent, insertionId)
        .then(cassandraEventDeadLettersGroupDAO.storeGroup(registeredGroup))
        .thenReturn(insertionId);
}
 

@Override
public <T> Result<List<T>> zip(List<Result<T>> results) {
	if (results.isEmpty()) {
		return just(new ArrayList<>());
	}
	List<Mono<T>> monos = new ArrayList<>();
	for (Result<T> result : results) {
		monos.add(((MonoResult) result).getMono());
	}
	Mono<List<T>> zipped = Mono.zip(monos, a -> Arrays.asList((T[]) a));
	return toResult(zipped);
}
 

private void checkReactiveTransaction(boolean expected) {
	Mono.subscriberContext().handle((context, sink) -> {
		if (context.hasKey(TransactionContext.class) != expected){
			fail("Should have thrown NoTransactionException");
		}
		sink.complete();
	}).block();
}
 

/**
 * 刷新路由
 *
 * @return
 */
public Mono<Void> refresh() {
    this.loadRoutes();
    // 触发默认路由刷新事件,刷新缓存路由
    this.publisher.publishEvent(new RefreshRoutesEvent(this));
    return Mono.empty();
}