io.deephaven.server.util.UnaryInputStreamMarshaller Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of deephaven-server Show documentation

The 'server' project

The newest version!

//
// Copyright (c) 2016-2024 Deephaven Data Labs and Patent Pending
//
package io.deephaven.server.util;

import io.deephaven.extensions.barrage.util.ExposedByteArrayOutputStream;
import io.grpc.MethodDescriptor;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;

public class UnaryInputStreamMarshaller implements MethodDescriptor.Marshaller {
    public static UnaryInputStreamMarshaller INSTANCE = new UnaryInputStreamMarshaller();

    @Override
    public InputStream stream(InputStream value) {
        return value;
    }

    @Override
    public InputStream parse(InputStream stream) {
        try (final ExposedByteArrayOutputStream baos = new ExposedByteArrayOutputStream()) {
            final byte[] buffer = new byte[4096];
            while (stream.available() > 0) {
                int len = stream.read(buffer);
                if (len <= 0) {
                    throw new IOException("failed to read from stream");
                }
                baos.write(buffer, 0, len);
            }
            return new ByteArrayInputStream(baos.peekBuffer(), 0, baos.size());
        } catch (final IOException e) {
            throw new IllegalStateException("Failed to parse barrage message: ", e);
        }
    }
}