feat：【framework】增加对 LocalDateTime 类型格式化不生效问题 #1019

YunaiV
1 parent 1a7a2182
Showing 1 changed file with 17 additions and 1 deletions
yudao-framework/yudao-common/src/main/java/cn/iocoder/yudao/framework/common/util/json/databind/TimestampLocalDateTimeSerializer.java
 package cn.iocoder.yudao.framework.common.util.json.databind;
  
+import com.fasterxml.jackson.annotation.JsonFormat;
 import com.fasterxml.jackson.core.JsonGenerator;
 import com.fasterxml.jackson.databind.JsonSerializer;
 import com.fasterxml.jackson.databind.SerializerProvider;
+import org.apache.commons.lang3.reflect.FieldUtils;
  
 import java.io.IOException;
+import java.lang.reflect.Field;
 import java.time.LocalDateTime;
 import java.time.ZoneId;
+import java.time.format.DateTimeFormatter;
  
 /**
  * 基于时间戳的 LocalDateTime 序列化器
@@ -19,7 +23,19 @@ public class TimestampLocalDateTimeSerializer extends JsonSerializer&lt;LocalDateTi
  
     @Override
     public void serialize(LocalDateTime value, JsonGenerator gen, SerializerProvider serializers) throws IOException {
-        // 将 LocalDateTime 对象，转换为 Long 时间戳
+        String fieldName = gen.getOutputContext().getCurrentName();
+        Class<?> clazz = gen.getOutputContext().getCurrentValue().getClass();
+        Field field = FieldUtils.getField(clazz, fieldName, true);
+        // 情况一：有 JsonFormat 自定义注解，则使用它。https://github.com/YunaiV/ruoyi-vue-pro/pull/1019
+        JsonFormat[] jsonFormats = field.getAnnotationsByType(JsonFormat.class);
+        if (jsonFormats.length > 0) {
+            String pattern = jsonFormats[0].pattern();
+            DateTimeFormatter formatter = DateTimeFormatter.ofPattern(pattern);
+            gen.writeString(formatter.format(value));
+            return;
+        }
+
+        // 情况二：默认将 LocalDateTime 对象，转换为 Long 时间戳
         gen.writeNumber(value.atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());
     }