com.intropro.prairie.format.avro.AvroFormat Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of format Show documentation

There is a newer version: 1.2.0

/**
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.intropro.prairie.format.avro;

import com.intropro.prairie.format.Format;
import com.intropro.prairie.format.InputFormatReader;
import com.intropro.prairie.format.OutputFormatWriter;
import com.intropro.prairie.format.exception.FormatException;
import org.apache.avro.Schema;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Map;

/**
 * Created by presidentio on 10/7/15.
 */
public class AvroFormat implements Format> {

    private Schema schema;

    public AvroFormat() {
    }

    public AvroFormat(String schemaStr) {
        this.schema = new Schema.Parser().parse(schemaStr);
    }

    public AvroFormat(InputStream inputStream) throws IOException {
        this.schema = new Schema.Parser().parse(inputStream);
    }

    public AvroFormat(File file) throws IOException {
        this.schema = new Schema.Parser().parse(file);
    }

    @Override
    public InputFormatReader> createReader(InputStream inputStream) throws FormatException {
        return new AvroFormatReader(inputStream, schema);
    }

    @Override
    public OutputFormatWriter> createWriter(OutputStream outputStream) throws FormatException {
        if (schema == null) {
            throw new FormatException("Avro fromat writer can't be created without schema");
        }
        return new AvroFormatWriter(outputStream, schema);
    }

}