com.treasure_data.td_import.source.Source Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of td-import Show documentation
Treasure Data Import Tool in Java.
There is a newer version: 0.5.10
//
// Treasure Data Bulk-Import Tool in Java
//
// Copyright (C) 2012 - 2013 Muga Nishizawa
//
//    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.treasure_data.td_import.source;

import java.io.IOException;
import java.io.InputStream;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class Source {

    public static enum Type {
        FILE("file") {
            @Override
            public List createSources(SourceDesc desc) {
                return LocalFileSource.createSources(desc);
            }
        },
        S3("s3") {
            @Override
            public List createSources(SourceDesc desc) {
                return S3Source.createSources(desc);
            }
        },
        MYSQL("mysql") {
            @Override
            public List createSources(SourceDesc desc) {
                return MysqlSource.createSources(desc);
            }
        };

        private String type;

        Type(String type) {
            this.type = type;
        }

        public String type() {
            return type;
        }

        public abstract List createSources(SourceDesc desc);

        public static Type fromString(String src) {
            return StringToType.get(src);
        }

        private static class StringToType {
            private static final Map REVERSE_DICTIONARY;

            static {
                Map map = new HashMap();
                for (Type elem : Type.values()) {
                    map.put(elem.type(), elem);
                }
                REVERSE_DICTIONARY = Collections.unmodifiableMap(map);
            }

            static Type get(String key) {
                return REVERSE_DICTIONARY.get(key);
            }
        }
    }

    public static class Factory {
        public static List createSources(SourceDesc desc) {
            Type type = Source.Type.fromString(desc.getType());
            if (type == null) {
                return null;
            }
            return type.createSources(desc);
        }
    }

    protected String path;

    public Source(String path) {
        this.path = path;
    }

    public String getPath() {
        return path;
    }

    public long getSize() {
        throw new UnsupportedOperationException(
                "this method should be declared in sub-class");
    }

    public InputStream getInputStream() throws IOException {
        throw new UnsupportedOperationException(
                "this method should be declared in sub-class");
    }

    @Override
    public boolean equals(Object o) {
        if (! (o instanceof Source)) {
            return false;
        }

        Source other = (Source) o;
        return path.equals(other.path);
    }
}