diff --git a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/common/CommonAdminController.java b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/common/CommonAdminController.java index d5859640..c0d5d521 100644 --- a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/common/CommonAdminController.java +++ b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/common/CommonAdminController.java @@ -5,11 +5,14 @@ import java.util.List; import ai.chat2db.server.admin.api.controller.common.converter.CommonAdminConverter; import ai.chat2db.server.admin.api.controller.common.vo.TeamUserListVO; +import ai.chat2db.server.admin.api.controller.datasource.vo.SimpleDataSourceVO; import ai.chat2db.server.admin.api.controller.team.vo.SimpleTeamVO; import ai.chat2db.server.admin.api.controller.user.vo.SimpleUserVO; import ai.chat2db.server.common.api.controller.request.CommonQueryRequest; +import ai.chat2db.server.domain.api.param.datasource.DataSourceSelector; import ai.chat2db.server.domain.api.param.team.TeamPageQueryParam; import ai.chat2db.server.domain.api.param.user.UserPageQueryParam; +import ai.chat2db.server.domain.api.service.DataSourceService; import ai.chat2db.server.domain.api.service.TeamService; import ai.chat2db.server.domain.api.service.UserService; import ai.chat2db.server.tools.base.wrapper.result.ListResult; @@ -28,12 +31,16 @@ import org.springframework.web.bind.annotation.RestController; @RequestMapping("/api/admin/common") @RestController public class CommonAdminController { - + private static final DataSourceSelector DATA_SOURCE_SELECTOR = DataSourceSelector.builder() + .environment(Boolean.TRUE) + .build(); @Resource private UserService userService; @Resource private TeamService teamService; @Resource + private DataSourceService dataSourceService; + @Resource private CommonAdminConverter commonAdminConverter; /** @@ -84,4 +91,17 @@ public class CommonAdminController { .mapToList(commonAdminConverter::dto2voTeam); } + /** + * Fuzzy query of data source + * + * @param request + * @return + * @version 2.1.0 + */ + @GetMapping("/data_source/list") + public ListResult dataSourceList(@Valid CommonQueryRequest request) { + return dataSourceService.queryPageWithPermission(commonAdminConverter.request2paramDataSource(request), + DATA_SOURCE_SELECTOR) + .mapToList(commonAdminConverter::dto2voDataSource); + } } diff --git a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/common/converter/CommonAdminConverter.java b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/common/converter/CommonAdminConverter.java index 41b28795..24034a69 100644 --- a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/common/converter/CommonAdminConverter.java +++ b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/common/converter/CommonAdminConverter.java @@ -1,12 +1,15 @@ package ai.chat2db.server.admin.api.controller.common.converter; import ai.chat2db.server.admin.api.controller.common.vo.TeamUserListVO; +import ai.chat2db.server.admin.api.controller.datasource.vo.SimpleDataSourceVO; import ai.chat2db.server.admin.api.controller.team.vo.SimpleTeamVO; import ai.chat2db.server.admin.api.controller.user.vo.SimpleUserVO; import ai.chat2db.server.common.api.controller.request.CommonQueryRequest; import ai.chat2db.server.domain.api.enums.AccessObjectTypeEnum; +import ai.chat2db.server.domain.api.model.DataSource; import ai.chat2db.server.domain.api.model.Team; import ai.chat2db.server.domain.api.model.User; +import ai.chat2db.server.domain.api.param.datasource.DataSourcePageQueryParam; import ai.chat2db.server.domain.api.param.team.TeamPageQueryParam; import ai.chat2db.server.domain.api.param.user.UserPageQueryParam; import org.mapstruct.Mapper; @@ -43,6 +46,17 @@ public abstract class CommonAdminConverter { }) public abstract UserPageQueryParam request2paramUser(CommonQueryRequest request); + /** + * conversion + * + * @param request + * @return + */ + @Mappings({ + @Mapping(target = "pageSize", expression = "java(10)"), + }) + public abstract DataSourcePageQueryParam request2paramDataSource(CommonQueryRequest request); + /** * conversion * @@ -51,6 +65,14 @@ public abstract class CommonAdminConverter { */ public abstract SimpleTeamVO dto2voTeam(Team dto); + /** + * conversion + * + * @param dto + * @return + */ + public abstract SimpleDataSourceVO dto2voDataSource(DataSource dto); + /** * conversion * diff --git a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/datasource/converter/DataSourceAccessAdminConverter.java b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/datasource/converter/DataSourceAccessAdminConverter.java index 19839a21..f97e2098 100644 --- a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/datasource/converter/DataSourceAccessAdminConverter.java +++ b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/datasource/converter/DataSourceAccessAdminConverter.java @@ -27,6 +27,7 @@ public abstract class DataSourceAccessAdminConverter { */ @Mappings({ @Mapping(source = "searchKey", target = "userOrTeamSearchKey"), + @Mapping(target = "enableReturnCount", expression = "java(true)"), }) public abstract DataSourceAccessComprehensivePageQueryParam request2param(DataSourceAccessPageQueryRequest request); diff --git a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/datasource/converter/DataSourceAdminConverter.java b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/datasource/converter/DataSourceAdminConverter.java index 9a1ebd6e..5236de60 100644 --- a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/datasource/converter/DataSourceAdminConverter.java +++ b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/datasource/converter/DataSourceAdminConverter.java @@ -27,6 +27,9 @@ public abstract class DataSourceAdminConverter { * @param request * @return */ + @Mappings({ + @Mapping(target = "enableReturnCount", expression = "java(true)"), + }) public abstract DataSourcePageQueryParam request2param(CommonPageQueryRequest request); /** @@ -35,6 +38,9 @@ public abstract class DataSourceAdminConverter { * @param request * @return */ + @Mappings({ + @Mapping(target = "enableReturnCount", expression = "java(true)"), + }) public abstract DataSourcePageQueryParam request2paramAccess(CommonPageQueryRequest request); /** diff --git a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/team/converter/TeamAdminConverter.java b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/team/converter/TeamAdminConverter.java index 9cbd9a16..a6eaf129 100644 --- a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/team/converter/TeamAdminConverter.java +++ b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/team/converter/TeamAdminConverter.java @@ -10,6 +10,8 @@ import ai.chat2db.server.domain.api.param.team.TeamCreateParam; import ai.chat2db.server.domain.api.param.team.TeamPageQueryParam; import ai.chat2db.server.domain.api.param.team.TeamUpdateParam; import org.mapstruct.Mapper; +import org.mapstruct.Mapping; +import org.mapstruct.Mappings; /** * converter @@ -26,6 +28,9 @@ public abstract class TeamAdminConverter { * @param request * @return */ + @Mappings({ + @Mapping(target = "enableReturnCount", expression = "java(true)"), + }) public abstract TeamPageQueryParam request2param(CommonPageQueryRequest request); diff --git a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/team/converter/TeamDataSourcesAdminConverter.java b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/team/converter/TeamDataSourcesAdminConverter.java index 451bf7e0..d3dc0ad5 100644 --- a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/team/converter/TeamDataSourcesAdminConverter.java +++ b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/team/converter/TeamDataSourcesAdminConverter.java @@ -27,6 +27,7 @@ public abstract class TeamDataSourcesAdminConverter { */ @Mappings({ @Mapping(source = "searchKey", target = "dataSourceSearchKey"), + @Mapping(target = "enableReturnCount", expression = "java(true)"), }) public abstract DataSourceAccessComprehensivePageQueryParam request2param(CommonPageQueryRequest request); diff --git a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/team/converter/TeamUserAdminConverter.java b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/team/converter/TeamUserAdminConverter.java index 4327e2e4..e7ed55b1 100644 --- a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/team/converter/TeamUserAdminConverter.java +++ b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/team/converter/TeamUserAdminConverter.java @@ -26,6 +26,7 @@ public abstract class TeamUserAdminConverter { */ @Mappings({ @Mapping(source = "searchKey", target = "userSearchKey"), + @Mapping(target = "enableReturnCount", expression = "java(true)"), }) public abstract TeamUserComprehensivePageQueryParam request2param(CommonPageQueryRequest request); diff --git a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/user/converter/UserAdminConverter.java b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/user/converter/UserAdminConverter.java index c3c750c5..28bc43b6 100644 --- a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/user/converter/UserAdminConverter.java +++ b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/user/converter/UserAdminConverter.java @@ -9,6 +9,8 @@ import ai.chat2db.server.domain.api.param.user.UserCreateParam; import ai.chat2db.server.domain.api.param.user.UserPageQueryParam; import ai.chat2db.server.domain.api.param.user.UserUpdateParam; import org.mapstruct.Mapper; +import org.mapstruct.Mapping; +import org.mapstruct.Mappings; /** * converter @@ -24,9 +26,11 @@ public abstract class UserAdminConverter { * @param request * @return */ + @Mappings({ + @Mapping(target = "enableReturnCount", expression = "java(true)"), + }) public abstract UserPageQueryParam request2param(CommonPageQueryRequest request); - /** * conversion * @@ -35,7 +39,6 @@ public abstract class UserAdminConverter { */ public abstract UserPageQueryVO dto2vo(User dto); - /** * conversion * @@ -44,7 +47,6 @@ public abstract class UserAdminConverter { */ public abstract UserCreateParam request2param(UserCreateRequest request); - /** * conversion * diff --git a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/user/converter/UserDataSourcesAdminConverter.java b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/user/converter/UserDataSourcesAdminConverter.java index e9de9d50..6e2eaf36 100644 --- a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/user/converter/UserDataSourcesAdminConverter.java +++ b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/user/converter/UserDataSourcesAdminConverter.java @@ -27,6 +27,7 @@ public abstract class UserDataSourcesAdminConverter { */ @Mappings({ @Mapping(source = "searchKey", target = "userOrTeamSearchKey"), + @Mapping(target = "enableReturnCount", expression = "java(true)"), }) public abstract DataSourceAccessComprehensivePageQueryParam request2param(CommonPageQueryRequest request); diff --git a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/user/converter/UserTeamAdminConverter.java b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/user/converter/UserTeamAdminConverter.java index 850e718a..41a0e139 100644 --- a/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/user/converter/UserTeamAdminConverter.java +++ b/chat2db-server/chat2db-server-web/chat2db-server-admin-api/src/main/java/ai/chat2db/server/admin/api/controller/user/converter/UserTeamAdminConverter.java @@ -24,6 +24,7 @@ public abstract class UserTeamAdminConverter { */ @Mappings({ @Mapping(source = "searchKey", target = "teamSearchKey"), + @Mapping(target = "enableReturnCount", expression = "java(true)"), }) public abstract TeamUserComprehensivePageQueryParam request2param(CommonPageQueryRequest request);