From 62e0ed647d0393cdaa8d91c878c805fa9db6f07b Mon Sep 17 00:00:00 2001 From: WendyYang Date: Wed, 28 Feb 2024 10:50:01 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E5=B1=9E=E6=80=A7=E5=80=BC?= =?UTF-8?q?=E5=A1=AB=E5=85=85=E5=A4=84=E7=90=86=E5=99=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../api/common/handler/BaseFieldFillHandler.java | 26 ++++++++++++++++------ 1 file changed, 19 insertions(+), 7 deletions(-) diff --git a/hz-pm-api/src/main/java/com/hz/pm/api/common/handler/BaseFieldFillHandler.java b/hz-pm-api/src/main/java/com/hz/pm/api/common/handler/BaseFieldFillHandler.java index 7b2513f..f3bac19 100644 --- a/hz-pm-api/src/main/java/com/hz/pm/api/common/handler/BaseFieldFillHandler.java +++ b/hz-pm-api/src/main/java/com/hz/pm/api/common/handler/BaseFieldFillHandler.java @@ -31,18 +31,30 @@ public class BaseFieldFillHandler implements MetaObjectHandler { @Override public void insertFill(MetaObject metaObject) { - long userId = getUserId(); + fillVal(CREATE_BY, metaObject, getUserId()); + fillVal(UPDATE_BY, metaObject, getUserId()); LocalDateTime now = LocalDateTime.now(); - this.fillStrategy(metaObject, CREATE_BY, userId); - this.fillStrategy(metaObject, CREATE_ON, now); - this.fillStrategy(metaObject, UPDATE_BY, userId); - this.fillStrategy(metaObject, UPDATE_ON, now); + fillVal(CREATE_ON, metaObject, now); + fillVal(UPDATE_ON, metaObject, now); } @Override public void updateFill(MetaObject metaObject) { - this.fillStrategy(metaObject, UPDATE_BY, getUserId()); - this.fillStrategy(metaObject, UPDATE_ON, LocalDateTime.now()); + fillVal(UPDATE_BY, metaObject, getUserId()); + fillVal(UPDATE_ON, metaObject, LocalDateTime.now()); + } + + private void fillVal(String fieldName, MetaObject metaObject, Object val) { + if (metaObject.hasSetter(fieldName)) { + Class setterType = metaObject.getSetterType(fieldName); + if (setterType != null) { + if (setterType.equals(String.class)) { + this.fillStrategy(metaObject, fieldName, val.toString()); + } else { + this.fillStrategy(metaObject, fieldName, val); + } + } + } } } \ No newline at end of file