v0.0.3 #3

Merged
timi merged 1 commits from dev into master 2026-05-09 06:23:03 +00:00
2 changed files with 166 additions and 39 deletions
Showing only changes of commit 62f35bdb78 - Show all commits

42
pom.xml
View File

@@ -6,7 +6,7 @@
<groupId>com.imyeyu.config</groupId> <groupId>com.imyeyu.config</groupId>
<artifactId>timi-config</artifactId> <artifactId>timi-config</artifactId>
<version>0.0.2</version> <version>0.0.3</version>
<properties> <properties>
<maven.test.skip>true</maven.test.skip> <maven.test.skip>true</maven.test.skip>
@@ -27,10 +27,42 @@
<artifactId>maven-source-plugin</artifactId> <artifactId>maven-source-plugin</artifactId>
<version>3.3.1</version> <version>3.3.1</version>
</plugin> </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> <plugin>
<groupId>org.apache.maven.plugins</groupId> <groupId>org.apache.maven.plugins</groupId>
<artifactId>maven-javadoc-plugin</artifactId> <artifactId>maven-javadoc-plugin</artifactId>
<version>3.11.2</version> <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> </plugin>
</plugins> </plugins>
</build> </build>
@@ -59,23 +91,23 @@
<dependency> <dependency>
<groupId>org.yaml</groupId> <groupId>org.yaml</groupId>
<artifactId>snakeyaml</artifactId> <artifactId>snakeyaml</artifactId>
<version>2.2</version> <version>2.6</version>
</dependency> </dependency>
<dependency> <dependency>
<groupId>com.imyeyu.io</groupId> <groupId>com.imyeyu.io</groupId>
<artifactId>timi-io</artifactId> <artifactId>timi-io</artifactId>
<version>0.0.2</version> <version>0.0.5</version>
</dependency> </dependency>
<dependency> <dependency>
<groupId>org.junit.jupiter</groupId> <groupId>org.junit.jupiter</groupId>
<artifactId>junit-jupiter</artifactId> <artifactId>junit-jupiter</artifactId>
<version>5.10.3</version> <version>6.0.3</version>
<scope>test</scope> <scope>test</scope>
</dependency> </dependency>
<dependency> <dependency>
<groupId>org.projectlombok</groupId> <groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId> <artifactId>lombok</artifactId>
<version>1.18.34</version> <version>1.18.46</version>
<scope>test</scope> <scope>test</scope>
</dependency> </dependency>
</dependencies> </dependencies>

View File

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