Modifying data sources, adding, deleting, and querying

This commit is contained in:
JiaJu Zhuang
2023-08-19 17:03:08 +08:00
parent 7a53957e3f
commit c387de4d64
12 changed files with 174 additions and 25 deletions

View File

@ -0,0 +1,34 @@
package ai.chat2db.server.domain.api.param.datasource.access;
import ai.chat2db.server.domain.api.enums.AccessObjectTypeEnum;
import ai.chat2db.server.tools.base.wrapper.param.PageQueryParam;
import jakarta.validation.constraints.NotNull;
import lombok.Data;
/**
* Data Source Access
*
* @author Jiaju Zhuang
*/
@Data
public class DataSourceAccessPageQueryParam extends PageQueryParam {
/**
* 数据源id
*/
@NotNull
private Long dataSourceId;
/**
* 授权类型
*
* @see AccessObjectTypeEnum
*/
@NotNull
private String accessObjectType;
/**
* 授权id,根据类型区分是用户还是团队
*/
@NotNull
private Long accessObjectId;
}

View File

@ -1,5 +1,6 @@
package ai.chat2db.server.domain.api.param.datasource.access;
import ai.chat2db.server.domain.api.param.datasource.DataSourceSelector;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
@ -21,9 +22,13 @@ public class DataSourceAccessSelector {
*/
private Boolean accessObject;
/**
* 数据源
*/
private Boolean dataSource;
/**
* 数据源
*/
private DataSourceSelector dataSourceSelector;
}

View File

@ -3,6 +3,7 @@ package ai.chat2db.server.domain.api.service;
import ai.chat2db.server.domain.api.model.DataSourceAccess;
import ai.chat2db.server.domain.api.param.datasource.access.DataSourceAccessComprehensivePageQueryParam;
import ai.chat2db.server.domain.api.param.datasource.access.DataSourceAccessCreatParam;
import ai.chat2db.server.domain.api.param.datasource.access.DataSourceAccessPageQueryParam;
import ai.chat2db.server.domain.api.param.datasource.access.DataSourceAccessSelector;
import ai.chat2db.server.tools.base.wrapper.result.ActionResult;
import ai.chat2db.server.tools.base.wrapper.result.DataResult;
@ -16,6 +17,15 @@ import jakarta.validation.constraints.NotNull;
*/
public interface DataSourceAccessService {
/**
* Comprehensive Paging Query Data
*
* @param param
* @param selector
* @return
*/
PageResult<DataSourceAccess> pageQuery(DataSourceAccessPageQueryParam param, DataSourceAccessSelector selector);
/**
* Paging Query Data
*

View File

@ -32,7 +32,6 @@ public interface DataSourceService {
* @return
*/
DataResult<Long> createWithPermission(DataSourceCreateParam param);
/**
* 更新数据源连接
@ -100,9 +99,18 @@ public interface DataSourceService {
*
* @param ids
* @return
* @deprecated Use {@link #listQuery(List, DataSourceSelector)}
*/
ListResult<DataSource> queryByIds(List<Long> ids);
/**
* 通过ID列表查询数据源
*
* @param idList
* @return
*/
ListResult<DataSource> listQuery(List<Long> idList, DataSourceSelector selector);
/**
* 数据源连接测试
*