[ISSUE-3393][Fix]fix mybatis-plus enum annotation (#3395)

apache · Dec 13, 2023 · eceae64 · eceae64
1 parent fdb7c34
commit eceae64
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/...onsole-service/src/main/java/org/apache/streampark/console/base/config/MybatisConfig.java b/...onsole-service/src/main/java/org/apache/streampark/console/base/config/MybatisConfig.java
@@ -77,6 +77,7 @@ public PostgreSQLPrepareInterceptor postgreSQLPrepareInterceptor() {
   public MybatisPlusPropertiesCustomizer mybatisPlusPropertiesCustomizer() {
     return properties -> {
       properties.setTypeAliasesPackage("org.apache.streampark.console.*.entity");
+      properties.setTypeEnumsPackage("org.apache.streampark.console.*.enums");
       properties.setMapperLocations(new String[] {"classpath:mapper/*/*.xml"});
       MybatisConfiguration mybatisConfiguration = new MybatisConfiguration();
       mybatisConfiguration.setJdbcTypeForNull(JdbcType.NULL);

diff --git a/...park-console-service/src/main/java/org/apache/streampark/console/core/entity/Message.java b/...park-console-service/src/main/java/org/apache/streampark/console/core/entity/Message.java
@@ -40,7 +40,7 @@ public class Message {
   private String title;
 
   /** 1) build failure report 2) task monitoring exception */
-  private Integer type;
+  private NoticeTypeEnum type;
 
   private String context;
 
@@ -54,7 +54,7 @@ public Message(
     this.appId = appId;
     this.title = title;
     this.context = context;
-    this.type = noticeTypeEnum.get();
+    this.type = noticeTypeEnum;
     this.createTime = new Date();
     this.isRead = false;
   }

diff --git a/...ice/src/main/java/org/apache/streampark/console/core/service/impl/MessageServiceImpl.java b/...ice/src/main/java/org/apache/streampark/console/core/service/impl/MessageServiceImpl.java
@@ -53,7 +53,7 @@ public IPage<Message> getUnReadPage(NoticeTypeEnum noticeTypeEnum, RestRequest r
         new LambdaQueryWrapper<Message>()
             .eq(Message::getIsRead, false)
             .orderByDesc(Message::getCreateTime)
-            .eq(Message::getType, noticeTypeEnum.get());
+            .eq(Message::getType, noticeTypeEnum);
     return this.baseMapper.selectPage(page, queryWrapper);
   }
 }