timi/timi-config
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

v0.0.3 #3

Merged
timi merged 1 commits from dev into master 2026-05-09 06:23:03 +00:00
Conversation 0 Commits 1 Files Changed 2 +166 -39
2 changed files with 166 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
pom.xml
View File

@@ -6,7 +6,7 @@
<groupId>com.imyeyu.config</groupId>
<artifactId>timi-config</artifactId>
<version>0.0.2</version>
<version>0.0.3</version>
<properties>
<maven.test.skip>true</maven.test.skip>
@@ -27,10 +27,42 @@
<artifactId>maven-source-plugin</artifactId>
<version>3.3.1</version>
</plugin>
<plugin>
<groupId>org.projectlombok</groupId>
<artifactId>lombok-maven-plugin</artifactId>
<version>1.18.20.0</version>
<configuration>
<sourceDirectory>${project.basedir}/src/main/java</sourceDirectory>
<outputDirectory>${project.build.directory}/delombok</outputDirectory>
<addOutputDirectory>false</addOutputDirectory>
<encoding>UTF-8</encoding>
</configuration>
<executions>
<execution>
<phase>generate-sources</phase>
<goals>
<goal>delombok</goal>
</goals>
</execution>
</executions>
<dependencies>
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
<version>1.18.36</version>
</dependency>
</dependencies>
</plugin>
<plugin>
<groupId>org.apache.maven.plugins</groupId>
<artifactId>maven-javadoc-plugin</artifactId>
<version>3.11.2</version>
<configuration>
<sourcepath>${project.build.directory}/delombok</sourcepath>
<encoding>UTF-8</encoding>
<charset>UTF-8</charset>
<docencoding>UTF-8</docencoding>
</configuration>
</plugin>
</plugins>
</build>
@@ -59,23 +91,23 @@
<dependency>
<groupId>org.yaml</groupId>
<artifactId>snakeyaml</artifactId>
<version>2.2</version>
<version>2.6</version>
</dependency>
<dependency>
<groupId>com.imyeyu.io</groupId>
<artifactId>timi-io</artifactId>
<version>0.0.2</version>
<version>0.0.5</version>
</dependency>
<dependency>
<groupId>org.junit.jupiter</groupId>
<artifactId>junit-jupiter</artifactId>
<version>5.10.3</version>
<version>6.0.3</version>
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
<version>1.18.34</version>
<version>1.18.46</version>
<scope>test</scope>
</dependency>
</dependencies>

161
src/main/java/com/imyeyu/config/ConfigLoader.java
View File

@@ -11,8 +11,12 @@ import org.yaml.snakeyaml.constructor.Constructor;
import org.yaml.snakeyaml.representer.Representer;
import java.io.File;
import java.lang.reflect.Array;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
/**
@@ -40,6 +44,9 @@ public class ConfigLoader<T> {
/**
* 创建配置加载器，源路径和目标路径相同
*
* @param path 源路径/目标路径
* @param clazz 配置类型
*/
public ConfigLoader(String path, Class<T> clazz) {
this(path, path, clazz);
@@ -69,7 +76,7 @@ public class ConfigLoader<T> {
LoaderOptions loaderOptions = new LoaderOptions();
loaderOptions.setAllowRecursiveKeys(false); // 禁止递归键
loaderOptions.setMaxAliasesForCollections(50); // 限制别名数量
loaderOptions.setCodePointLimit((int) (IOSize.MB * 3)); // 限制 3 MB
loaderOptions.setCodePointLimit((int) (IOSize.MB * 10)); // 限制 10 MB
DumperOptions dumperOptions = new DumperOptions();
dumperOptions.setIndent(4);
@@ -96,13 +103,7 @@ public class ConfigLoader<T> {
public T load() {
try {
String content = IO.toString(file);
String normalized = content.replaceAll("\t", " ");
if (converters.isEmpty()) {
// SnakeYAML 自动转换 - 使用指定类型构造
value = yaml.load(normalized);
return value;
}
Object raw = rawYaml.load(normalized);
Object raw = rawYaml.load(content);
if (raw == null) {
value = null;
return null;
@@ -129,9 +130,8 @@ public class ConfigLoader<T> {
Object toSerialize = applyConvertersOnDump(value);
// SnakeYAML 序列化
String yamlContent = yaml.dump(toSerialize);
// 空格转制表符
String toWrite = yamlContent.replaceAll("( ){4}", " ");
IO.toFile(file, toWrite);
IO.toFile(file, yamlContent);
} catch (Exception e) {
throw new TimiException(TimiCode.ERROR, "dump config error: " + file.getAbsolutePath(), e);
}
@@ -163,11 +163,10 @@ public class ConfigLoader<T> {
Class<?> currentClass = target.getClass();
while (currentClass != null && currentClass != Object.class) {
for (Field field : currentClass.getDeclaredFields()) {
String fieldName = field.getName();
if (!map.containsKey(fieldName)) {
Object rawValue = resolveMapValue(map, field.getName());
if (rawValue == null && !containsFieldKey(map, field.getName())) {
continue;
}
Object rawValue = map.get(fieldName);
Object converted = convertFieldValue(field, rawValue);
field.setAccessible(true);
field.set(target, converted);
@@ -176,6 +175,17 @@ public class ConfigLoader<T> {
}
}
private boolean containsFieldKey(Map<?, ?> map, String fieldName) {
return map.containsKey(fieldName) || map.containsKey(toSnakeCase(fieldName));
}
private Object resolveMapValue(Map<?, ?> map, String fieldName) {
if (map.containsKey(fieldName)) {
return map.get(fieldName);
}
return map.get(toSnakeCase(fieldName));
}
private Object convertFieldValue(Field field, Object rawValue) {
BaseConverter<?, ?> converter = converters.get(field.getType());
if (converter != null) {
@@ -188,12 +198,40 @@ public class ConfigLoader<T> {
if (fieldType.isEnum()) {
return toEnumValue(fieldType, rawValue.toString());
}
if (fieldType.isArray() && rawValue instanceof Iterable<?> rawIterable) {
return toArrayValue(fieldType.getComponentType(), rawIterable);
}
if (rawValue instanceof Map<?, ?> rawMap && shouldMapToBean(fieldType)) {
return mapToBean(rawMap, fieldType);
}
return convertScalarValue(fieldType, rawValue);
}
private Object toArrayValue(Class<?> componentType, Iterable<?> rawIterable) {
List<Object> values = new ArrayList<>();
for (Object rawElement : rawIterable) {
values.add(convertArrayElement(componentType, rawElement));
}
Object array = Array.newInstance(componentType, values.size());
for (int i = 0; i < values.size(); i++) {
Array.set(array, i, values.get(i));
}
return array;
}
private Object convertArrayElement(Class<?> componentType, Object rawElement) {
if (rawElement == null) {
return null;
}
if (componentType.isEnum()) {
return toEnumValue(componentType, rawElement.toString());
}
if (rawElement instanceof Map<?, ?> rawMap && shouldMapToBean(componentType)) {
return mapToBean(rawMap, componentType);
}
return convertScalarValue(componentType, rawElement);
}
private boolean shouldMapToBean(Class<?> fieldType) {
if (fieldType.isPrimitive() || fieldType.isEnum()) {
return false;
@@ -251,35 +289,92 @@ public class ConfigLoader<T> {
}
/** 序列化时应用转换器 */
@SuppressWarnings("unchecked")
private <K> K applyConvertersOnDump(K obj) {
if (obj == null || converters.isEmpty()) {
return obj;
private Object applyConvertersOnDump(Object obj) {
return toYamlValue(null, obj);
}
try {
// 创建副本
K copy = (K) obj.getClass().getDeclaredConstructor().newInstance();
Class<?> currentClass = obj.getClass();
while (currentClass != null && currentClass != Object.class) {
for (Field field : currentClass.getDeclaredFields()) {
field.setAccessible(true);
Object fieldValue = field.get(obj);
private Object toYamlValue(Field field, Object value) {
if (field != null) {
BaseConverter<?, ?> converter = converters.get(field.getType());
if (converter != null) {
// 应用转换器
Object yamlValue = converter.serialize0(field, fieldValue);
field.set(copy, yamlValue);
} else {
// 直接复制
field.set(copy, fieldValue);
return converter.serialize0(field, value);
}
}
if (value == null) {
return null;
}
Class<?> valueType = value.getClass();
if (valueType.isEnum()) {
return ((Enum<?>) value).name();
}
if (isScalarType(valueType)) {
return value;
}
if (valueType.isArray()) {
int length = Array.getLength(value);
List<Object> result = new ArrayList<>(length);
for (int i = 0; i < length; i++) {
result.add(toYamlValue(null, Array.get(value, i)));
}
return result;
}
if (value instanceof Iterable<?> iterable) {
List<Object> result = new ArrayList<>();
for (Object item : iterable) {
result.add(toYamlValue(null, item));
}
return result;
}
if (value instanceof Map<?, ?> mapValue) {
Map<Object, Object> result = new LinkedHashMap<>();
for (Map.Entry<?, ?> entry : mapValue.entrySet()) {
result.put(entry.getKey(), toYamlValue(null, entry.getValue()));
}
return result;
}
if (!shouldMapToBean(valueType)) {
return value;
}
try {
Map<String, Object> result = new LinkedHashMap<>();
Class<?> currentClass = valueType;
while (currentClass != null && currentClass != Object.class) {
for (Field declaredField : currentClass.getDeclaredFields()) {
declaredField.setAccessible(true);
Object fieldValue = toYamlValue(declaredField, declaredField.get(value));
if (fieldValue != null) {
result.put(toSnakeCase(declaredField.getName()), fieldValue);
}
}
currentClass = currentClass.getSuperclass();
}
return copy;
return result;
} catch (Exception e) {
throw new TimiException(TimiCode.ERROR, "converter error on dump", e);
}
}
private boolean isScalarType(Class<?> type) {
return type.isPrimitive()
|| type == String.class
|| Number.class.isAssignableFrom(type)
|| type == Boolean.class
|| type == Character.class;
}
private String toSnakeCase(String fieldName) {
StringBuilder builder = new StringBuilder(fieldName.length() + 4);
for (int i = 0; i < fieldName.length(); i++) {
char current = fieldName.charAt(i);
if (Character.isUpperCase(current)) {
if (i != 0) {
builder.append('_');
}
builder.append(Character.toLowerCase(current));
continue;
}
builder.append(current);
}
return builder.toString();
}
}