GH-792 Fix Supplier streaming in s-c-function-web

Resolves #792
This commit is contained in:
Oleg Zhurakousky
2022-01-24 15:36:24 +01:00
parent 2771fe905d
commit 73c46c5877
10 changed files with 108 additions and 41 deletions

View File

@@ -1,7 +1,7 @@
<?xml version="1.0" encoding="UTF-8"?>
<project xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xmlns="http://maven.apache.org/POM/4.0.0"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
xmlns="http://maven.apache.org/POM/4.0.0"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<groupId>io.spring.sample</groupId>
@@ -26,9 +26,9 @@
<dependencies>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-actuator</artifactId>
</dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-actuator</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.cloud</groupId>
<artifactId>spring-cloud-starter-function-webflux</artifactId>

View File

@@ -0,0 +1,44 @@
/*
* Copyright 2012-2019 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.example;
import org.springframework.web.reactive.function.client.WebClient;
/**
* Sample client to test infinite stream from function.
*
* @author Oleg Zhurakousky
*
*/
public class Client {
public static void main(String[] args) throws Exception {
WebClient client = WebClient.create();
WebClient.ResponseSpec responseSpec = client.post()
.uri("http://localhost:8080/infinite")
.header("accept", "text/event-stream")
.retrieve();
responseSpec.bodyToFlux(String.class).subscribe(v -> {
System.out.println(v);
});
System.in.read();
}
}

View File

@@ -16,10 +16,12 @@
package com.example;
import java.time.Duration;
import java.util.function.Function;
import java.util.function.Supplier;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
@@ -55,9 +57,11 @@ public class SampleApplication {
}
@Bean
public Supplier<Flux<String>> words() {
return () -> Flux.fromArray(new String[] {"foo", "bar"});
public Supplier<Flux<String>> infinite() {
return () -> Flux
.interval(Duration.ofSeconds(1))
.log()
.map(counter -> String.format("Counter: %s", counter));
}
}
// @checkstyle:on

View File

@@ -51,15 +51,6 @@ public class FunctionTests {
assertThat(output).isEqualTo("hello");
}
@Test
public void testWords() {
Flux<String> output = this.functions.words().get();
List<String> results = output.collectList().block();
assertThat(results.size()).isEqualTo(2);
assertThat(results.get(0)).isEqualTo("foo");
assertThat(results.get(1)).isEqualTo("bar");
}
@Test
public void testGreeter() {
assertThat(new Greeter().apply("World")).isEqualTo("Hello World");