ru.yandex.qatools.properties.providers.DefaultPropertyProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of properties-loader Show documentation

The newest version!

package ru.yandex.qatools.properties.providers;

import ru.yandex.qatools.properties.annotations.Resource;

import java.lang.annotation.Annotation;
import java.util.Properties;

import static ru.yandex.qatools.properties.utils.PropertiesUtils.readProperties;


/**
 * User: lanwen
 * Date: 17.07.13
 * Time: 17:28
 */
public class DefaultPropertyProvider implements PropertyProvider {
    @Override
    public  Properties provide(T bean, Properties properties) {
        Class clazz = bean.getClass();

        if (have(clazz, Resource.Classpath.class)) {
            String[] paths = classpath(clazz, properties);
            for (String path : paths) {
                properties.putAll(readProperties(getClassLoader().getResourceAsStream(path)));
            }
        }

        if (have(clazz, Resource.File.class)) {
            String[] paths = filepath(clazz, properties);
            for (String path : paths) {
                properties.putAll(readProperties(new java.io.File(path)));
            }
        }

        properties.putAll(System.getProperties());
        return properties;
    }


    protected boolean have(Class clazz, Class anno) {
        return clazz.isAnnotationPresent(anno);
    }

    protected String[] filepath(Class clazz, Properties properties) {
        return clazz.getAnnotation(Resource.File.class).value();
    }

    protected String[] classpath(Class clazz, Properties properties) {
        return clazz.getAnnotation(Resource.Classpath.class).value();
    }

    private ClassLoader getClassLoader() {
        ClassLoader classLoader = null;
        try {
            classLoader = Thread.currentThread().getContextClassLoader();
        } catch (SecurityException ignored) {
            // do nothing
        } finally {
            if (classLoader == null) {
                classLoader = ClassLoader.getSystemClassLoader();
            }
        }
        return classLoader;
    }
}