Examples with Msg cn.taketoday.protobuf.Msg used on opensource projects

Example 6 with Msg

use of cn.taketoday.protobuf.Msg in project today-framework by TAKETODAY.

the class ProtobufDecoderTests method decodeSplitMessageSize.

// SPR-17429
@Test
public void decodeSplitMessageSize() {
    this.decoder.setMaxMessageSize(100009);
    StringBuilder builder = new StringBuilder();
    for (int i = 0; i < 10000; i++) {
        builder.append("azertyuiop");
    }
    Msg bigMessage = Msg.newBuilder().setFoo(builder.toString()).setBlah(secondMsg2).build();
    Flux<DataBuffer> input = Flux.just(bigMessage, bigMessage).flatMap(msg -> Mono.defer(() -> {
        DataBuffer buffer = this.bufferFactory.allocateBuffer();
        try {
            msg.writeDelimitedTo(buffer.asOutputStream());
            return Mono.just(buffer);
        } catch (IOException e) {
            release(buffer);
            return Mono.error(e);
        }
    })).flatMap(buffer -> {
        int len = 2;
        Flux<DataBuffer> result = Flux.just(DataBufferUtils.retain(buffer.slice(0, len)), DataBufferUtils.retain(buffer.slice(len, buffer.readableByteCount() - len)));
        release(buffer);
        return result;
    });
    testDecode(input, Msg.class, step -> step.expectNext(bigMessage).expectNext(bigMessage).verifyComplete());
}

Example 7 with Msg

use of cn.taketoday.protobuf.Msg in project today-framework by TAKETODAY.

the class ProtobufDecoderTests method decodeSplitChunks.

@Test
public void decodeSplitChunks() {
    Flux<DataBuffer> input = Flux.just(this.testMsg1, this.testMsg2).flatMap(msg -> Mono.defer(() -> {
        DataBuffer buffer = this.bufferFactory.allocateBuffer();
        try {
            msg.writeDelimitedTo(buffer.asOutputStream());
            return Mono.just(buffer);
        } catch (IOException e) {
            release(buffer);
            return Mono.error(e);
        }
    })).flatMap(buffer -> {
        int len = buffer.readableByteCount() / 2;
        Flux<DataBuffer> result = Flux.just(DataBufferUtils.retain(buffer.slice(0, len)), DataBufferUtils.retain(buffer.slice(len, buffer.readableByteCount() - len)));
        release(buffer);
        return result;
    });
    testDecode(input, Msg.class, step -> step.expectNext(this.testMsg1).expectNext(this.testMsg2).verifyComplete());
}

Example 8 with Msg

use of cn.taketoday.protobuf.Msg in project today-infrastructure by TAKETODAY.

the class ProtobufDecoderTests method decodeSplitChunks.

@Test
public void decodeSplitChunks() {
    Flux<DataBuffer> input = Flux.just(this.testMsg1, this.testMsg2).flatMap(msg -> Mono.defer(() -> {
        DataBuffer buffer = this.bufferFactory.allocateBuffer();
        try {
            msg.writeDelimitedTo(buffer.asOutputStream());
            return Mono.just(buffer);
        } catch (IOException e) {
            release(buffer);
            return Mono.error(e);
        }
    })).flatMap(buffer -> {
        int len = buffer.readableByteCount() / 2;
        Flux<DataBuffer> result = Flux.just(DataBufferUtils.retain(buffer.slice(0, len)), DataBufferUtils.retain(buffer.slice(len, buffer.readableByteCount() - len)));
        release(buffer);
        return result;
    });
    testDecode(input, Msg.class, step -> step.expectNext(this.testMsg1).expectNext(this.testMsg2).verifyComplete());
}