对于这样一个简单的案例来说有点复杂，但我需要实现 5 个类，从 ClientRequest 中提取一个主体。
似乎太多了，我仍然很好奇这个问题是否有一个简短的解决方案。欢迎您对这个问题提出另一个答案，所以我可以接受。
不幸的是，必须说ClientRequest、BodyInserters 以及org.springframework.web.reactive.*** 的大部分其他东西的设计都有很大的改进空间。目前它只是一堆接口，每个接口都有大量方法，根据这些类，测试代码通常需要付出很多努力。
实现此方法的主要目标：
static <T> T extractBody(ClientRequest request, Class<T> clazz) {
InsertionReceiver<T> receiver = InsertionReceiver.forClass(clazz);
return receiver.receiveValue(request.body());
}
这里是InsertionReceiver的实现：
-
import org.springframework.http.ReactiveHttpOutputMessage;
import org.springframework.web.reactive.function.BodyInserter;
public interface InsertionReceiver<T> {
T receiveValue(BodyInserter<?, ? extends ReactiveHttpOutputMessage> bodyInserter);
static <T> InsertionReceiver<T> forClass(Class<T> clazz) {
return new SimpleValueReceiver<>(clazz);
}
}
-
import java.util.concurrent.atomic.AtomicReference;
import org.springframework.http.ReactiveHttpOutputMessage;
import org.springframework.web.reactive.function.BodyInserter;
class SimpleValueReceiver<T> implements InsertionReceiver<T> {
private static final Object DUMMY = new Object();
private final Class<T> clazz;
private final AtomicReference<Object> reference;
SimpleValueReceiver(Class<T> clazz) {
this.clazz = clazz;
this.reference = new AtomicReference<>(DUMMY);
}
@Override
public T receiveValue(BodyInserter<?, ? extends ReactiveHttpOutputMessage> bodyInserter) {
demandValueFrom(bodyInserter);
return receivedValue();
}
private void demandValueFrom(BodyInserter<?, ? extends ReactiveHttpOutputMessage> bodyInserter) {
var inserter = (BodyInserter<?, ReactiveHttpOutputMessage>) bodyInserter;
inserter.insert(
MinimalHttpOutputMessage.INSTANCE,
new SingleWriterContext(new WriteToConsumer<>(reference::set))
);
}
private T receivedValue() {
Object value = reference.get();
reference.set(DUMMY);
T validatedValue;
if (value == DUMMY) {
throw new RuntimeException("Value was not received, Check your inserter worked properly");
} else if (!clazz.isAssignableFrom(value.getClass())) {
throw new RuntimeException(
"Value has unexpected type ("
+ value.getClass().getTypeName()
+ ") instead of (" + clazz.getTypeName() + ")");
} else {
validatedValue = clazz.cast(value);
}
return validatedValue;
}
}
-
class WriteToConsumer<T> implements HttpMessageWriter<T> {
private final Consumer<T> consumer;
private final List<MediaType> mediaTypes;
WriteToConsumer(Consumer<T> consumer) {
this.consumer = consumer;
this.mediaTypes = Collections.singletonList(MediaType.ALL);
}
@Override
public List<MediaType> getWritableMediaTypes() {
return mediaTypes;
}
@Override
public boolean canWrite(ResolvableType elementType, MediaType mediaType) {
return true;
}
@Override
public Mono<Void> write(
Publisher<? extends T> inputStream,
ResolvableType elementType,
MediaType mediaType,
ReactiveHttpOutputMessage message,
Map<String, Object> hints
) {
inputStream.subscribe(new OneValueConsumption<>(consumer));
return Mono.empty();
}
}
-
class MinimalHttpOutputMessage implements ReactiveHttpOutputMessage {
public static MinimalHttpOutputMessage INSTANCE = new MinimalHttpOutputMessage();
private MinimalHttpOutputMessage() {
}
@Override
public HttpHeaders getHeaders() {
return HttpHeaders.EMPTY;
}
// other overridden methods are omitted as they do nothing,
// i.e. return null, false, or have empty bodies
}
-
class OneValueConsumption<T> implements Subscriber<T> {
private final Consumer<T> consumer;
private int remainedAccepts;
public OneValueConsumption(Consumer<T> consumer) {
this.consumer = Objects.requireNonNull(consumer);
this.remainedAccepts = 1;
}
@Override
public void onSubscribe(Subscription s) {
s.request(1);
}
@Override
public void onNext(T o) {
if (remainedAccepts > 0) {
consumer.accept(o);
remainedAccepts -= 1;
} else {
throw new RuntimeException("No more values can be consumed");
}
}
@Override
public void onError(Throwable t) {
throw new RuntimeException("Single value was not consumed", t);
}
@Override
public void onComplete() {
// nothing
}
}
-
class SingleWriterContext implements BodyInserter.Context {
private final List<HttpMessageWriter<?>> singleWriterList;
SingleWriterContext(HttpMessageWriter<?> writer) {
this.singleWriterList = List.of(writer);
}
@Override
public List<HttpMessageWriter<?>> messageWriters() {
return singleWriterList;
}
@Override
public Optional<ServerHttpRequest> serverRequest() {
return Optional.empty();
}
@Override
public Map<String, Object> hints() {
return null;
}
}