Examples with DataBufferUtils org.springframework.core.io.buffer.DataBufferUtils used on opensource projects

Example 1 with DataBufferUtils

use of org.springframework.core.io.buffer.DataBufferUtils in project spring-framework by spring-projects.

the class MultipartHttpMessageWriter method writeMultipart.

private Mono<Void> writeMultipart(MultiValueMap<String, ?> map, ReactiveHttpOutputMessage outputMessage, @Nullable MediaType mediaType, Map<String, Object> hints) {
    byte[] boundary = generateMultipartBoundary();
    mediaType = getMultipartMediaType(mediaType, boundary);
    outputMessage.getHeaders().setContentType(mediaType);
    LogFormatUtils.traceDebug(logger, traceOn -> Hints.getLogPrefix(hints) + "Encoding " + (isEnableLoggingRequestDetails() ? LogFormatUtils.formatValue(map, !traceOn) : "parts " + map.keySet() + " (content masked)"));
    DataBufferFactory bufferFactory = outputMessage.bufferFactory();
    Flux<DataBuffer> body = Flux.fromIterable(map.entrySet()).concatMap(entry -> encodePartValues(boundary, entry.getKey(), entry.getValue(), bufferFactory)).concatWith(generateLastLine(boundary, bufferFactory)).doOnDiscard(PooledDataBuffer.class, DataBufferUtils::release);
    if (logger.isDebugEnabled()) {
        body = body.doOnNext(buffer -> Hints.touchDataBuffer(buffer, hints, logger));
    }
    return outputMessage.writeWith(body);
}

Example 2 with DataBufferUtils

use of org.springframework.core.io.buffer.DataBufferUtils in project spring-framework by spring-projects.

the class AbstractServerHttpResponse method writeWith.

@Override
@SuppressWarnings("unchecked")
public final Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {
    // We must resolve value first however, for a chance to handle potential error.
    if (body instanceof Mono) {
        return ((Mono<? extends DataBuffer>) body).flatMap(buffer -> {
            touchDataBuffer(buffer);
            AtomicBoolean subscribed = new AtomicBoolean();
            return doCommit(() -> {
                try {
                    return writeWithInternal(Mono.fromCallable(() -> buffer).doOnSubscribe(s -> subscribed.set(true)).doOnDiscard(PooledDataBuffer.class, DataBufferUtils::release));
                } catch (Throwable ex) {
                    return Mono.error(ex);
                }
            }).doOnError(ex -> DataBufferUtils.release(buffer)).doOnCancel(() -> {
                if (!subscribed.get()) {
                    DataBufferUtils.release(buffer);
                }
            });
        }).doOnError(t -> getHeaders().clearContentHeaders()).doOnDiscard(PooledDataBuffer.class, DataBufferUtils::release);
    } else {
        return new ChannelSendOperator<>(body, inner -> doCommit(() -> writeWithInternal(inner))).doOnError(t -> getHeaders().clearContentHeaders());
    }
}

Example 3 with DataBufferUtils

use of org.springframework.core.io.buffer.DataBufferUtils in project spring-data-mongodb by spring-projects.

the class ReactiveGridFsTemplateTests method storeSavesGridFsUploadWithGivenIdCorrectly.

// DATAMONGO-625
@Test
public void storeSavesGridFsUploadWithGivenIdCorrectly() throws IOException {
    String id = "id-1";
    byte[] content = StreamUtils.copyToByteArray(resource.getInputStream());
    Flux<DataBuffer> data = DataBufferUtils.read(resource, new DefaultDataBufferFactory(), 256);
    ReactiveGridFsUpload<String> upload = // 
    ReactiveGridFsUpload.fromPublisher(data).id(// 
    id).filename(// 
    "gridFsUpload.xml").contentType(// 
    "xml").build();
    operations.store(upload).as(StepVerifier::create).expectNext(id).verifyComplete();
    operations.findOne(query(where("_id").is(id))).flatMap(operations::getResource).flatMapMany(// 
    ReactiveGridFsResource::getDownloadStream).transform(// 
    DataBufferUtils::join).as(// 
    StepVerifier::create).consumeNextWith(dataBuffer -> {
        byte[] actual = new byte[dataBuffer.readableByteCount()];
        dataBuffer.read(actual);
        assertThat(actual).isEqualTo(content);
    }).verifyComplete();
    operations.findOne(query(where("_id").is(id))).as(StepVerifier::create).consumeNextWith(it -> {
        assertThat(it.getFilename()).isEqualTo("gridFsUpload.xml");
        assertThat(it.getId()).isEqualTo(new BsonString(id));
        assertThat(it.getMetadata()).containsValue("xml");
    }).verifyComplete();
}

Example 4 with DataBufferUtils

use of org.springframework.core.io.buffer.DataBufferUtils in project spring-framework by spring-projects.

the class EncoderHttpMessageWriter method write.

@Override
public Mono<Void> write(Publisher<? extends T> inputStream, ResolvableType elementType, @Nullable MediaType mediaType, ReactiveHttpOutputMessage message, Map<String, Object> hints) {
    MediaType contentType = updateContentType(message, mediaType);
    Flux<DataBuffer> body = this.encoder.encode(inputStream, message.bufferFactory(), elementType, contentType, hints);
    if (inputStream instanceof Mono) {
        return body.singleOrEmpty().switchIfEmpty(Mono.defer(() -> {
            message.getHeaders().setContentLength(0);
            return message.setComplete().then(Mono.empty());
        })).flatMap(buffer -> {
            Hints.touchDataBuffer(buffer, hints, logger);
            message.getHeaders().setContentLength(buffer.readableByteCount());
            return message.writeWith(Mono.just(buffer).doOnDiscard(PooledDataBuffer.class, DataBufferUtils::release));
        }).doOnDiscard(PooledDataBuffer.class, DataBufferUtils::release);
    }
    if (isStreamingMediaType(contentType)) {
        return message.writeAndFlushWith(body.map(buffer -> {
            Hints.touchDataBuffer(buffer, hints, logger);
            return Mono.just(buffer).doOnDiscard(PooledDataBuffer.class, DataBufferUtils::release);
        }));
    }
    if (logger.isDebugEnabled()) {
        body = body.doOnNext(buffer -> Hints.touchDataBuffer(buffer, hints, logger));
    }
    return message.writeWith(body);
}

Example 5 with DataBufferUtils

use of org.springframework.core.io.buffer.DataBufferUtils in project spring-framework by spring-projects.

the class ServerSentEventHttpMessageWriter method encode.

private Flux<Publisher<DataBuffer>> encode(Publisher<?> input, ResolvableType elementType, MediaType mediaType, DataBufferFactory factory, Map<String, Object> hints) {
    ResolvableType dataType = (ServerSentEvent.class.isAssignableFrom(elementType.toClass()) ? elementType.getGeneric() : elementType);
    return Flux.from(input).map(element -> {
        ServerSentEvent<?> sse = (element instanceof ServerSentEvent ? (ServerSentEvent<?>) element : ServerSentEvent.builder().data(element).build());
        StringBuilder sb = new StringBuilder();
        String id = sse.id();
        String event = sse.event();
        Duration retry = sse.retry();
        String comment = sse.comment();
        Object data = sse.data();
        if (id != null) {
            writeField("id", id, sb);
        }
        if (event != null) {
            writeField("event", event, sb);
        }
        if (retry != null) {
            writeField("retry", retry.toMillis(), sb);
        }
        if (comment != null) {
            sb.append(':').append(StringUtils.replace(comment, "\n", "\n:")).append('\n');
        }
        if (data != null) {
            sb.append("data:");
        }
        Flux<DataBuffer> result;
        if (data == null) {
            result = Flux.just(encodeText(sb + "\n", mediaType, factory));
        } else if (data instanceof String) {
            data = StringUtils.replace((String) data, "\n", "\ndata:");
            result = Flux.just(encodeText(sb + (String) data + "\n\n", mediaType, factory));
        } else {
            result = encodeEvent(sb, data, dataType, mediaType, factory, hints);
        }
        return result.doOnDiscard(PooledDataBuffer.class, DataBufferUtils::release);
    });
}