opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-08-02 05:20:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Complete team user code

Browse Source
This commit is contained in:
JiaJu Zhuang
2023-08-06 17:46:34 +08:00
parent 29155703c4
commit f492ad9a14
13 changed files with 467 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
chat2db-server/chat2db-server-domain/chat2db-server-domain-core/src/main/java/ai/chat2db/server/domain/core/converter/DataSourceAccessConverter.java Normal file
View File

@ -0,0 +1,56 @@
package ai.chat2db.server.domain.core.converter;
import java.util.List;
import ai.chat2db.server.domain.api.model.DataSourceAccess;
import ai.chat2db.server.domain.repository.entity.DataSourceAccessDO;
import lombok.extern.slf4j.Slf4j;
import org.mapstruct.Mapper;
import org.mapstruct.Mapping;
import org.mapstruct.Mappings;
/**
* converter
*
* @author Jiaju Zhuang
*/
@Slf4j
@Mapper(componentModel = "spring")
public abstract class DataSourceAccessConverter {
/**
* convert
*
* @param data
* @return
*/
@Mappings({
@Mapping(target = "accessObject.id", source = "accessObjectId"),
@Mapping(target = "accessObject.type", source = "accessObjectType"),
})
public abstract DataSourceAccess do2dto(DataSourceAccessDO data);
/**
* convert
*
* @param dataSourceId
* @param accessObjectId
* @param accessObjectType
* @param userId
* @return
*/
@Mappings({
@Mapping(target = "createUserId", source = "userId"),
@Mapping(target = "modifiedUserId", source = "userId"),
})
public abstract DataSourceAccessDO param2do(Long dataSourceId, Long accessObjectId, String accessObjectType,
Long userId);
/**
* convert
*
* @param list
* @return
*/
public abstract List<DataSourceAccess> do2dto(List<DataSourceAccessDO> list);
}

147
chat2db-server/chat2db-server-domain/chat2db-server-domain-core/src/main/java/ai/chat2db/server/domain/core/impl/DataSourceAccessServiceImpl.java Normal file
View File

@ -0,0 +1,147 @@
package ai.chat2db.server.domain.core.impl;
import java.util.List;
import java.util.Map;
import com.alibaba.fastjson2.JSON;
import ai.chat2db.server.domain.api.enums.AccessObjectTypeEnum;
import ai.chat2db.server.domain.api.model.DataSourceAccess;
import ai.chat2db.server.domain.api.model.DataSourceAccessObject;
import ai.chat2db.server.domain.api.model.Team;
import ai.chat2db.server.domain.api.model.User;
import ai.chat2db.server.domain.api.param.datasource.access.DataSourceAccessBatchCreatParam;
import ai.chat2db.server.domain.api.param.datasource.access.DataSourceAccessComprehensivePageQueryParam;
import ai.chat2db.server.domain.api.param.datasource.access.DataSourceAccessObjectParam;
import ai.chat2db.server.domain.api.param.datasource.access.DataSourceAccessSelector;
import ai.chat2db.server.domain.api.service.DataSourceAccessService;
import ai.chat2db.server.domain.api.service.TeamService;
import ai.chat2db.server.domain.api.service.UserService;
import ai.chat2db.server.domain.core.converter.DataSourceAccessConverter;
import ai.chat2db.server.domain.repository.entity.DataSourceAccessDO;
import ai.chat2db.server.domain.repository.mapper.DataSourceAccessCustomMapper;
import ai.chat2db.server.domain.repository.mapper.DataSourceAccessMapper;
import ai.chat2db.server.tools.base.wrapper.result.ActionResult;
import ai.chat2db.server.tools.base.wrapper.result.PageResult;
import ai.chat2db.server.tools.common.util.ContextUtils;
import ai.chat2db.server.tools.common.util.EasyCollectionUtils;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.google.common.collect.Lists;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.BooleanUtils;
import org.springframework.stereotype.Service;
/**
* Data Source Access
*
* @author Jiaju Zhuang
*/
@Slf4j
@Service
public class DataSourceAccessServiceImpl implements DataSourceAccessService {
@Resource
private DataSourceAccessCustomMapper dataSourceAccessCustomMapper;
@Resource
private DataSourceAccessMapper dataSourceAccessMapper;
@Resource
private DataSourceAccessConverter dataSourceAccessConverter;
@Resource
private UserService userService;
@Resource
private TeamService teamService;
@Override
public PageResult<DataSourceAccess> comprehensivePageQuery(DataSourceAccessComprehensivePageQueryParam param,
DataSourceAccessSelector selector) {
IPage<DataSourceAccessDO> iPage = dataSourceAccessCustomMapper.comprehensivePageQuery(
new Page<>(param.getPageNo(), param.getPageSize()), param.getDataSourceId(), param.getSearchKey());
List<DataSourceAccess> list = dataSourceAccessConverter.do2dto(iPage.getRecords());
fillData(list, selector);
return PageResult.of(list, iPage.getTotal(), param);
}
@Override
public ActionResult batchCreate(DataSourceAccessBatchCreatParam param) {
if (CollectionUtils.isEmpty(param.getAccessObjectList())) {
return ActionResult.isSuccess();
}
for (DataSourceAccessObjectParam dataSourceAccessObjectParam : param.getAccessObjectList()) {
LambdaQueryWrapper<DataSourceAccessDO> queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.eq(DataSourceAccessDO::getAccessObjectType, dataSourceAccessObjectParam.getType());
queryWrapper.eq(DataSourceAccessDO::getDataSourceId, dataSourceAccessObjectParam.getId());
DataSourceAccessDO query = dataSourceAccessMapper.selectOne(queryWrapper);
if (query != null) {
log.info("The data source already exists, no need to add it again,{}",
JSON.toJSONString(dataSourceAccessObjectParam));
continue;
}
dataSourceAccessMapper.insert(
dataSourceAccessConverter.param2do(param.getDataSourceId(), dataSourceAccessObjectParam.getId(),
dataSourceAccessObjectParam.getType(), ContextUtils.getUserId()));
}
return ActionResult.isSuccess();
}
@Override
public ActionResult delete(Long id) {
dataSourceAccessMapper.deleteById(id);
return ActionResult.isSuccess();
}
private void fillData(List<DataSourceAccess> list, DataSourceAccessSelector selector) {
if (CollectionUtils.isEmpty(list) || selector == null) {
return;
}
fillAccessObject(list, selector);
}
private void fillAccessObject(List<DataSourceAccess> list, DataSourceAccessSelector selector) {
if (BooleanUtils.isNotTrue(selector.getAccessObject())) {
return;
}
List<Long> userIdList = Lists.newArrayList();
List<Long> teamIdList = Lists.newArrayList();
for (DataSourceAccess data : list) {
if (AccessObjectTypeEnum.TEAM.getCode().equals(data.getAccessObjectType())) {
teamIdList.add(data.getAccessObjectId());
} else if (AccessObjectTypeEnum.USER.getCode().equals(data.getAccessObjectType())) {
userIdList.add(data.getAccessObjectId());
}
}
List<User> userList = userService.listQuery(userIdList).getData();
Map<Long, User> userMap = EasyCollectionUtils.toIdentityMap(userList, User::getId);
List<Team> teamList = teamService.listQuery(teamIdList).getData();
Map<Long, Team> teamMap = EasyCollectionUtils.toIdentityMap(teamList, Team::getId);
for (DataSourceAccess data : list) {
DataSourceAccessObject dataSourceAccessObject = data.getAccessObject();
if (dataSourceAccessObject == null) {
continue;
}
if (AccessObjectTypeEnum.TEAM.getCode().equals(data.getAccessObjectType())) {
Team team = teamMap.get(data.getAccessObjectId());
if (team == null) {
continue;
}
dataSourceAccessObject.setCode(team.getCode());
dataSourceAccessObject.setName(team.getName());
} else if (AccessObjectTypeEnum.USER.getCode().equals(data.getAccessObjectType())) {
User user = userMap.get(data.getAccessObjectId());
if (user == null) {
continue;
}
dataSourceAccessObject.setCode(user.getUserName());
dataSourceAccessObject.setName(user.getNickName());
}
}
}
}