com.mongodb.hadoop.streaming.io.MongoIdentifierResolver Maven / Gradle / Ivy

Go to download
package com.mongodb.hadoop.streaming.io;

import com.mongodb.hadoop.io.BSONWritable;
import org.apache.hadoop.streaming.io.IdentifierResolver;

public class MongoIdentifierResolver extends IdentifierResolver {
    public static final String MONGODB_ID = "mongodb";
    public static final String MONGO_ID = "mongo";
    public static final String BSON_ID = "bson";

    @Override
    public void resolve(final String identifier) {
        if (identifier.equalsIgnoreCase(MONGODB_ID) || identifier.equalsIgnoreCase(MONGO_ID) || identifier.equalsIgnoreCase(BSON_ID)) {
            setInputWriterClass(MongoInputWriter.class);
            setOutputReaderClass(MongoOutputReader.class);
            setOutputKeyClass(BSONWritable.class);
            setOutputValueClass(BSONWritable.class);
        } else {
            super.resolve(identifier);
        }
    }
}