42 lines
1.7 KiB
Java
42 lines
1.7 KiB
Java
package com.baeldung.functional;
|
|
|
|
import org.springframework.core.io.buffer.DataBuffer;
|
|
import org.springframework.util.MultiValueMap;
|
|
import org.springframework.web.reactive.function.server.ServerRequest;
|
|
import org.springframework.web.reactive.function.server.ServerResponse;
|
|
import reactor.core.publisher.Mono;
|
|
|
|
import java.util.List;
|
|
import java.util.concurrent.atomic.AtomicLong;
|
|
|
|
import static org.springframework.web.reactive.function.BodyExtractors.toDataBuffers;
|
|
import static org.springframework.web.reactive.function.BodyExtractors.toFormData;
|
|
import static org.springframework.web.reactive.function.BodyInserters.fromObject;
|
|
import static org.springframework.web.reactive.function.server.ServerResponse.ok;
|
|
|
|
public class FormHandler {
|
|
|
|
Mono<ServerResponse> handleLogin(ServerRequest request) {
|
|
return request.body(toFormData())
|
|
.map(MultiValueMap::toSingleValueMap)
|
|
.filter(formData -> "baeldung".equals(formData.get("user")))
|
|
.filter(formData -> "you_know_what_to_do".equals(formData.get("token")))
|
|
.flatMap(formData -> ok().body(Mono.just("welcome back!"), String.class))
|
|
.switchIfEmpty(ServerResponse.badRequest()
|
|
.build());
|
|
}
|
|
|
|
Mono<ServerResponse> handleUpload(ServerRequest request) {
|
|
return request.body(toDataBuffers())
|
|
.collectList()
|
|
.flatMap(dataBuffers -> ok().body(fromObject(extractData(dataBuffers).toString())));
|
|
}
|
|
|
|
private AtomicLong extractData(List<DataBuffer> dataBuffers) {
|
|
AtomicLong atomicLong = new AtomicLong(0);
|
|
dataBuffers.forEach(d -> atomicLong.addAndGet(d.asByteBuffer()
|
|
.array().length));
|
|
return atomicLong;
|
|
}
|
|
}
|