ProtoTextMessageWriter.java

package net.morimekta.proto.sio;

import com.google.protobuf.Message;
import com.google.protobuf.TextFormat;

import java.io.IOException;
import java.io.Writer;

import static java.util.Objects.requireNonNull;

/**
 * Writer for writing messages using the proto text format.
 */
public class ProtoTextMessageWriter implements MessageWriter {
    private final Writer writer;

    /**
     * @param writer Writer to write message to.
     */
    public ProtoTextMessageWriter(Writer writer) {
        this.writer = requireNonNull(writer, "out == null");
    }

    @Override
    public void write(Message message) throws IOException {
        requireNonNull(message, "message == null");
        TextFormat.printer().print(message, writer);
        writer.flush();
    }
}