Search in sources :

Example 1 with ObjectToChunks

use of io.helidon.media.jsonb.JsonbBodyWriter.ObjectToChunks in project helidon by oracle.

the class JsonbBodyStreamWriter method write.

@Override
public Multi<DataChunk> write(Flow.Publisher<?> publisher, GenericType<?> type, MessageBodyWriterContext context) {
    MediaType contentType = context.findAccepted(MediaType.JSON_PREDICATE, MediaType.APPLICATION_JSON);
    context.contentType(contentType);
    AtomicBoolean first = new AtomicBoolean(true);
    ObjectToChunks jsonToChunks = new ObjectToChunks(jsonb, context.charset());
    return Multi.create(publisher).flatMap(jsonToChunks).flatMap(it -> {
        if (first.getAndSet(false)) {
            // first record, do not prepend a comma
            return Multi.just(DataChunk.create(ARRAY_JSON_BEGIN_BYTES), it);
        } else {
            // any subsequent record starts with a comma
            return Multi.just(DataChunk.create(COMMA_BYTES), it);
        }
    }).onCompleteResume(DataChunk.create(ARRAY_JSON_END_BYTES));
}
Also used : Objects(java.util.Objects) Jsonb(jakarta.json.bind.Jsonb) Flow(java.util.concurrent.Flow) DataChunk(io.helidon.common.http.DataChunk) AtomicBoolean(java.util.concurrent.atomic.AtomicBoolean) MessageBodyStreamWriter(io.helidon.media.common.MessageBodyStreamWriter) ObjectToChunks(io.helidon.media.jsonb.JsonbBodyWriter.ObjectToChunks) GenericType(io.helidon.common.GenericType) StandardCharsets(java.nio.charset.StandardCharsets) Multi(io.helidon.common.reactive.Multi) MessageBodyWriterContext(io.helidon.media.common.MessageBodyWriterContext) MediaType(io.helidon.common.http.MediaType) AtomicBoolean(java.util.concurrent.atomic.AtomicBoolean) ObjectToChunks(io.helidon.media.jsonb.JsonbBodyWriter.ObjectToChunks) MediaType(io.helidon.common.http.MediaType)

Aggregations

GenericType (io.helidon.common.GenericType)1 DataChunk (io.helidon.common.http.DataChunk)1 MediaType (io.helidon.common.http.MediaType)1 Multi (io.helidon.common.reactive.Multi)1 MessageBodyStreamWriter (io.helidon.media.common.MessageBodyStreamWriter)1 MessageBodyWriterContext (io.helidon.media.common.MessageBodyWriterContext)1 ObjectToChunks (io.helidon.media.jsonb.JsonbBodyWriter.ObjectToChunks)1 Jsonb (jakarta.json.bind.Jsonb)1 StandardCharsets (java.nio.charset.StandardCharsets)1 Objects (java.util.Objects)1 Flow (java.util.concurrent.Flow)1 AtomicBoolean (java.util.concurrent.atomic.AtomicBoolean)1