Merge pull request #1019 from wunai-pl/patch-2

增加对LocalDateTime类型格式化不生效问题

Author: YunaiV

yudao-framework/yudao-common/src/main/java/cn/iocoder/yudao/framework/common/util/json/databind/TimestampLocalDateTimeSerializer.java

@@ -1,12 +1,16 @@
 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,8 +23,18 @@ public class TimestampLocalDateTimeSerializer extends JsonSerializer<LocalDateTi
 
     @Override
     public void serialize(LocalDateTime value, JsonGenerator gen, SerializerProvider serializers) throws IOException {
-        // 将 LocalDateTime 对象，转换为 Long 时间戳
-        gen.writeNumber(value.atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());
+        String fieldName = gen.getOutputContext().getCurrentName();
+        Class<?> clazz = gen.getOutputContext().getCurrentValue().getClass();
+        Field field =  FieldUtils.getField(clazz, fieldName, true);
+        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));
+        }else{
+            // 将 LocalDateTime 对象，转换为 Long 时间戳
+            gen.writeNumber(value.atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());
+        }
     }
 
 }