com.yomahub.liteflow.spi.solon.SolonPathContentParser Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of liteflow-solon-plugin Show documentation
liteflow solon plugin
The newest version!
package com.yomahub.liteflow.spi.solon;

import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.io.IoUtil;
import cn.hutool.core.stream.StreamUtil;
import cn.hutool.core.util.CharsetUtil;
import cn.hutool.core.util.StrUtil;
import com.yomahub.liteflow.exception.ConfigErrorException;
import com.yomahub.liteflow.spi.PathContentParser;
import com.yomahub.liteflow.util.PathMatchUtil;
import org.noear.solon.core.util.ResourceUtil;

import java.io.File;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

public class SolonPathContentParser implements PathContentParser {

	@Override
	public List parseContent(List pathList) throws Exception {
		List allResource = getUrls(pathList);

		// 转换成内容List
		List contentList = new ArrayList<>();
		for (URL resource : allResource) {
			String content = IoUtil.read(resource.openStream(), CharsetUtil.CHARSET_UTF_8);
			if (StrUtil.isNotBlank(content)) {
				contentList.add(content);
			}
		}

		return contentList;
	}

	@Override
	public List getFileAbsolutePath(List pathList) throws Exception {
		List allResource = getUrls(pathList);
		return StreamUtil.of(allResource).map(URL::getPath).filter(FileUtil::isFile).collect(Collectors.toList());
	}

	private static List getUrls(List pathList) throws MalformedURLException {
		if (CollectionUtil.isEmpty(pathList)) {
			throw new ConfigErrorException("rule source must not be null");
		}
		List absolutePathList = PathMatchUtil.searchAbsolutePath(pathList);
		List allResource = new ArrayList<>();

		for (String path : absolutePathList) {
			// 如果 path 是绝对路径且这个文件存在时，我们认为这是一个本地文件路径，而并非classpath路径
			if (FileUtil.isAbsolutePath(path) && FileUtil.isFile(path)) {
				allResource.add(new File(path).toURI().toURL());
			}
			else {
				if (path.startsWith(ResourceUtils.CLASSPATH_URL_PREFIX)) {
					path = path.substring(ResourceUtils.CLASSPATH_URL_PREFIX.length());
				}

				if (ResourceUtil.getResource(path) != null) {
					allResource.add(ResourceUtil.getResource(path));
				}
			}
		}

		// 如果有多个资源，检查资源都是同一个类型，如果出现不同类型的配置，则抛出错误提示
		Set fileTypeSet = new HashSet<>();
		allResource.forEach(resource -> fileTypeSet.add(FileUtil.extName(resource.getPath())));
		if (fileTypeSet.size() > 1) {
			throw new ConfigErrorException("config error,please use the same type of configuration");
		}
		return allResource;
	}

	@Override
	public int priority() {
		return 1;
	}

}