代码生成器优化

src/main/java/com/aizuda/boot/modules/gen/entity/GenDatabase.java

@@ -4,14 +4,13 @@ import com.aizuda.core.bean.BaseEntity;
 import com.aizuda.core.validation.Create;
 import com.baomidou.mybatisplus.annotation.DbType;
 import com.baomidou.mybatisplus.annotation.TableName;
+import io.swagger.v3.oas.annotations.media.Schema;
 import jakarta.validation.constraints.NotBlank;
 import jakarta.validation.constraints.NotNull;
 import jakarta.validation.constraints.PositiveOrZero;
 import jakarta.validation.constraints.Size;
-
-import io.swagger.v3.oas.annotations.media.Schema;
-import lombok.Setter;
 import lombok.Getter;
+import lombok.Setter;
 
 /**
  * 代码生成数据源表
@@ -60,6 +59,10 @@ public class GenDatabase extends BaseEntity {
 	@Size(max = 50)
 	private String database;
 
+	@Schema(description = "备注")
+	@Size(max = 255)
+	private String remark;
+
 	public String driverClassName() {
 		return "org.postgresql.Driver";
 	}

src/main/java/com/aizuda/boot/modules/gen/service/impl/GenDatabaseServiceImpl.java

@@ -13,6 +13,7 @@ import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import org.springframework.stereotype.Service;
 
+import java.io.Serializable;
 import java.util.ArrayList;
 import java.util.List;
 
@@ -28,12 +29,17 @@ public class GenDatabaseServiceImpl extends BaseServiceImpl<GenDatabaseMapper, G
     @Override
     public Page<GenDatabase> page(Page<GenDatabase> page, GenDatabase genDatabase) {
         LambdaQueryWrapper<GenDatabase> lqw = Wrappers.lambdaQuery(genDatabase);
-        return super.page(page, lqw);
+        Page<GenDatabase> genDatabasePage = super.page(page, lqw);
+        if (CollectionUtils.isNotEmpty(genDatabasePage.getRecords())) {
+            // 屏蔽数据源密码
+            genDatabasePage.getRecords().forEach(t -> t.setPassword(null));
+        }
+        return genDatabasePage;
     }
 
     @Override
     public boolean save(GenDatabase genDatabase) {
-        this.checkDatabase(genDatabase);
+//        this.checkDatabase(genDatabase);
         return super.save(genDatabase);
     }
 
@@ -45,10 +51,20 @@ public class GenDatabaseServiceImpl extends BaseServiceImpl<GenDatabaseMapper, G
     @Override
     public boolean updateById(GenDatabase genDatabase) {
         ApiAssert.fail(null == genDatabase.getId(), "主键不存在无法更新");
-        this.checkDatabase(genDatabase);
+//        this.checkDatabase(genDatabase);
         return super.updateById(genDatabase);
     }
 
+    @Override
+    public GenDatabase getById(Serializable id) {
+        GenDatabase genDatabase = super.getById(id);
+        if (null != genDatabase) {
+            // 屏蔽数据源密码
+            genDatabase.setPassword(null);
+        }
+        return genDatabase;
+    }
+
     @Override
     public List<SelectOptionVO> listSelectOptions() {
         List<SelectOptionVO> voList = new ArrayList<>();