iailab-plat.git

houzhongyi

2024-07-11 e7c1260db32209a078a962aaa0ad5492c35774fb

提交 \| 用户 \| 时间
e7c126	1	package com.iailab.framework.ip.core.utils;
H	2
	3	import cn.hutool.core.io.resource.ResourceUtil;
	4	import cn.hutool.core.lang.Assert;
	5	import cn.hutool.core.text.csv.CsvRow;
	6	import cn.hutool.core.text.csv.CsvUtil;
	7	import com.iailab.framework.common.util.object.ObjectUtils;
	8	import com.iailab.framework.ip.core.Area;
	9	import com.iailab.framework.ip.core.enums.AreaTypeEnum;
	10	import lombok.NonNull;
	11	import lombok.extern.slf4j.Slf4j;
	12
	13	import java.util.ArrayList;
	14	import java.util.HashMap;
	15	import java.util.List;
	16	import java.util.Map;
	17	import java.util.function.Function;
	18
	19	import static com.iailab.framework.common.util.collection.CollectionUtils.convertList;
	20	import static com.iailab.framework.common.util.collection.CollectionUtils.findFirst;
	21
	22	/**
	23	* 区域工具类
	24	*
	25	* @author iailab
	26	*/
	27	@Slf4j
	28	public class AreaUtils {
	29
	30	/**
	31	* 初始化 SEARCHER
	32	*/
	33	@SuppressWarnings("InstantiationOfUtilityClass")
	34	private final static AreaUtils INSTANCE = new AreaUtils();
	35
	36	/**
	37	* Area 内存缓存，提升访问速度
	38	*/
	39	private static Map<Integer, Area> areas;
	40
	41	private AreaUtils() {
	42	long now = System.currentTimeMillis();
	43	areas = new HashMap<>();
	44	areas.put(Area.ID_GLOBAL, new Area(Area.ID_GLOBAL, "全球", 0,
	45	null, new ArrayList<>()));
	46	// 从 csv 中加载数据
	47	List<CsvRow> rows = CsvUtil.getReader().read(ResourceUtil.getUtf8Reader("area.csv")).getRows();
	48	rows.remove(0); // 删除 header
	49	for (CsvRow row : rows) {
	50	// 创建 Area 对象
	51	Area area = new Area(Integer.valueOf(row.get(0)), row.get(1), Integer.valueOf(row.get(2)),
	52	null, new ArrayList<>());
	53	// 添加到 areas 中
	54	areas.put(area.getId(), area);
	55	}
	56
	57	// 构建父子关系：因为 Area 中没有 parentId 字段，所以需要重复读取
	58	for (CsvRow row : rows) {
	59	Area area = areas.get(Integer.valueOf(row.get(0))); // 自己
	60	Area parent = areas.get(Integer.valueOf(row.get(3))); // 父
	61	Assert.isTrue(area != parent, "{}:父子节点相同", area.getName());
	62	area.setParent(parent);
	63	parent.getChildren().add(area);
	64	}
	65	log.info("启动加载 AreaUtils 成功，耗时 ({}) 毫秒", System.currentTimeMillis() - now);
	66	}
	67
	68	/**
	69	* 获得指定编号对应的区域
	70	*
	71	* @param id 区域编号
	72	* @return 区域
	73	*/
	74	public static Area getArea(Integer id) {
	75	return areas.get(id);
	76	}
	77
	78	/**
	79	* 获得指定区域对应的编号
	80	*
	81	* @param pathStr 区域路径，例如说：河南省/石家庄市/新华区
	82	* @return 区域
	83	*/
	84	public static Area parseArea(String pathStr) {
	85	String[] paths = pathStr.split("/");
	86	Area area = null;
	87	for (String path : paths) {
	88	if (area == null) {
	89	area = findFirst(areas.values(), item -> item.getName().equals(path));
	90	} else {
	91	area = findFirst(area.getChildren(), item -> item.getName().equals(path));
	92	}
	93	}
	94	return area;
	95	}
	96
	97	/**
	98	* 获取所有节点的全路径名称如：河南省/石家庄市/新华区
	99	*
	100	* @param areas 地区树
	101	* @return 所有节点的全路径名称
	102	*/
	103	public static List<String> getAreaNodePathList(List<Area> areas) {
	104	List<String> paths = new ArrayList<>();
	105	areas.forEach(area -> getAreaNodePathList(area, "", paths));
	106	return paths;
	107	}
	108
	109	/**
	110	* 构建一棵树的所有节点的全路径名称，并将其存储为 "祖先/父级/子级" 的形式
	111	*
	112	* @param node 父节点
	113	* @param path 全路径名称
	114	* @param paths 全路径名称列表，省份/城市/地区
	115	*/
	116	private static void getAreaNodePathList(Area node, String path, List<String> paths) {
	117	if (node == null) {
	118	return;
	119	}
	120	// 构建当前节点的路径
	121	String currentPath = path.isEmpty() ? node.getName() : path + "/" + node.getName();
	122	paths.add(currentPath);
	123	// 递归遍历子节点
	124	for (Area child : node.getChildren()) {
	125	getAreaNodePathList(child, currentPath, paths);
	126	}
	127	}
	128
	129	/**
	130	* 格式化区域
	131	*
	132	* @param id 区域编号
	133	* @return 格式化后的区域
	134	*/
	135	public static String format(Integer id) {
	136	return format(id, " ");
	137	}
	138
	139	/**
	140	* 格式化区域
	141	*
	142	* 例如说：
	143	* 1. id = “静安区”时：上海上海市静安区
	144	* 2. id = “上海市”时：上海上海市
	145	* 3. id = “上海”时：上海
	146	* 4. id = “美国”时：美国
	147	* 当区域在中国时，默认不显示中国
	148	*
	149	* @param id 区域编号
	150	* @param separator 分隔符
	151	* @return 格式化后的区域
	152	*/
	153	public static String format(Integer id, String separator) {
	154	// 获得区域
	155	Area area = areas.get(id);
	156	if (area == null) {
	157	return null;
	158	}
	159
	160	// 格式化
	161	StringBuilder sb = new StringBuilder();
	162	for (int i = 0; i < AreaTypeEnum.values().length; i++) { // 避免死循环
	163	sb.insert(0, area.getName());
	164	// “递归”父节点
	165	area = area.getParent();
	166	if (area == null
	167	\|\| ObjectUtils.equalsAny(area.getId(), Area.ID_GLOBAL, Area.ID_CHINA)) { // 跳过父节点为中国的情况
	168	break;
	169	}
	170	sb.insert(0, separator);
	171	}
	172	return sb.toString();
	173	}
	174
	175	/**
	176	* 获取指定类型的区域列表
	177	*
	178	* @param type 区域类型
	179	* @param func 转换函数
	180	* @param <T> 结果类型
	181	* @return 区域列表
	182	*/
	183	public static <T> List<T> getByType(AreaTypeEnum type, Function<Area, T> func) {
	184	return convertList(areas.values(), func, area -> type.getType().equals(area.getType()));
	185	}
	186
	187	/**
	188	* 根据区域编号、上级区域类型，获取上级区域编号
	189	*
	190	* @param id 区域编号
	191	* @param type 区域类型
	192	* @return 上级区域编号
	193	*/
	194	public static Integer getParentIdByType(Integer id, @NonNull AreaTypeEnum type) {
	195	for (int i = 0; i < Byte.MAX_VALUE; i++) {
	196	Area area = AreaUtils.getArea(id);
	197	if (area == null) {
	198	return null;
	199	}
	200	// 情况一：匹配到，返回它
	201	if (type.getType().equals(area.getType())) {
	202	return area.getId();
	203	}
	204	// 情况二：找到根节点，返回空
	205	if (area.getParent() == null \|\| area.getParent().getId() == null) {
	206	return null;
	207	}
	208	// 其它：继续向上查找
	209	id = area.getParent().getId();
	210	}
	211	return null;
	212	}
	213
	214	}