org.codelibs.elasticsearch.idxproxy.stream.IndexingProxyStreamInput Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of elasticsearch-indexing-proxy Show documentation

This plugin provides a proxy for indexing.

There is a newer version: 5.6.5

Show newest version

package org.codelibs.elasticsearch.idxproxy.stream;

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

import org.elasticsearch.common.io.stream.InputStreamStreamInput;
import org.elasticsearch.common.io.stream.NamedWriteable;
import org.elasticsearch.common.io.stream.NamedWriteableRegistry;
import org.elasticsearch.common.io.stream.Writeable;

public class IndexingProxyStreamInput extends InputStreamStreamInput {

    private final NamedWriteableRegistry namedWriteableRegistry;

    public IndexingProxyStreamInput(final InputStream is, final NamedWriteableRegistry namedWriteableRegistry) {
        super(is);
        this.namedWriteableRegistry = namedWriteableRegistry;
    }

    @Override
    public  C readNamedWriteable(final Class categoryClass) throws IOException {
        final String name = readString();
        return readNamedWriteable(categoryClass, name);
    }

    @Override
    public  C readNamedWriteable(final Class categoryClass, final String name) throws IOException {
        final Writeable.Reader reader = namedWriteableRegistry.getReader(categoryClass, name);
        final C c = reader.read(this);
        if (c == null) {
            throw new IOException(
                    "Writeable.Reader [" + reader + "] returned null which is not allowed and probably means it screwed up the stream.");
        }
        assert name.equals(c.getWriteableName()) : c + " claims to have a different name [" + c.getWriteableName()
                + "] than it was read from [" + name + "].";
        return c;
    }
}