spring.turbo.module.configuration.env.AbstractPropertySourceFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-turbo-module-configuration Show documentation

There is a newer version: 3.3.5-1

Show newest version

package spring.turbo.module.configuration.env;

import org.springframework.boot.env.PropertySourceLoader;
import org.springframework.core.env.PropertySource;
import org.springframework.core.io.support.EncodedResource;
import org.springframework.core.io.support.PropertySourceFactory;
import org.springframework.lang.Nullable;
import spring.turbo.util.StringUtils;
import spring.turbo.util.UUIDUtils;

import java.io.IOException;

/**
 * (内部使用)
 *
 * @author 应卓
 * @since 2.1.3
 */
public abstract class AbstractPropertySourceFactory implements PropertySourceFactory {

    private final PropertySourceLoader loader;

    public AbstractPropertySourceFactory(PropertySourceLoader loader) {
        this.loader = loader;
    }

    @Override
    public final PropertySource createPropertySource(@Nullable String name, EncodedResource resource)
            throws IOException {
        final var propertySourceName = enforcePropertySourceName(name, resource);
        final var list = loader.load(propertySourceName, resource.getResource());

        if (list.isEmpty()) {
            return EmptyPropertySource.of(propertySourceName);
        } else if (list.size() == 1) {
            return list.get(0);
        } else {
            throw new IOException("multiple document is NOT supported yet.");
        }
    }

    private String enforcePropertySourceName(@Nullable String name, EncodedResource resource) {
        if (name == null) {
            name = resource.getResource().getFilename();
        }

        if (StringUtils.isBlank(name)) {
            return UUIDUtils.uuid36();
        }
        return name;
    }

}