mirror of
https://github.com/CodePhiliaX/Chat2DB.git
synced 2025-07-29 18:53:12 +08:00
Modify login interface
This commit is contained in:
@ -0,0 +1,27 @@
|
||||
package ai.chat2db.server.domain.api.param.team.user;
|
||||
|
||||
import ai.chat2db.server.tools.base.wrapper.param.PageQueryParam;
|
||||
import jakarta.validation.constraints.NotNull;
|
||||
import lombok.Data;
|
||||
|
||||
/**
|
||||
* Team User
|
||||
*
|
||||
* @author Jiaju Zhuang
|
||||
*/
|
||||
@Data
|
||||
public class TeamUserPageQueryParam extends PageQueryParam {
|
||||
|
||||
/**
|
||||
* 团队id
|
||||
*/
|
||||
@NotNull
|
||||
private Long teamId;
|
||||
|
||||
/**
|
||||
* 用户id
|
||||
*/
|
||||
@NotNull
|
||||
private Long userId;
|
||||
|
||||
}
|
@ -3,6 +3,7 @@ package ai.chat2db.server.domain.api.service;
|
||||
import ai.chat2db.server.domain.api.model.TeamUser;
|
||||
import ai.chat2db.server.domain.api.param.team.user.TeamUserComprehensivePageQueryParam;
|
||||
import ai.chat2db.server.domain.api.param.team.user.TeamUserCreatParam;
|
||||
import ai.chat2db.server.domain.api.param.team.user.TeamUserPageQueryParam;
|
||||
import ai.chat2db.server.domain.api.param.team.user.TeamUserSelector;
|
||||
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 TeamUserService {
|
||||
|
||||
/**
|
||||
* Comprehensive Paging Query Data
|
||||
*
|
||||
* @param param
|
||||
* @param selector
|
||||
* @return
|
||||
*/
|
||||
PageResult<TeamUser> pageQuery(TeamUserPageQueryParam param, TeamUserSelector selector);
|
||||
|
||||
/**
|
||||
* Comprehensive Paging Query Data
|
||||
*
|
||||
|
@ -5,6 +5,7 @@ import java.util.List;
|
||||
import ai.chat2db.server.domain.api.model.TeamUser;
|
||||
import ai.chat2db.server.domain.api.param.team.user.TeamUserComprehensivePageQueryParam;
|
||||
import ai.chat2db.server.domain.api.param.team.user.TeamUserCreatParam;
|
||||
import ai.chat2db.server.domain.api.param.team.user.TeamUserPageQueryParam;
|
||||
import ai.chat2db.server.domain.api.param.team.user.TeamUserSelector;
|
||||
import ai.chat2db.server.domain.api.service.TeamUserService;
|
||||
import ai.chat2db.server.domain.core.converter.TeamConverter;
|
||||
@ -18,6 +19,7 @@ import ai.chat2db.server.tools.base.wrapper.result.DataResult;
|
||||
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 jakarta.annotation.Resource;
|
||||
@ -46,6 +48,24 @@ public class TeamUserServiceImpl implements TeamUserService {
|
||||
@Resource
|
||||
private TeamConverter teamConverter;
|
||||
|
||||
@Override
|
||||
public PageResult<TeamUser> pageQuery(TeamUserPageQueryParam param, TeamUserSelector selector) {
|
||||
LambdaQueryWrapper<TeamUserDO> queryWrapper = new LambdaQueryWrapper<>();
|
||||
queryWrapper.eq(TeamUserDO::getTeamId, param.getTeamId())
|
||||
.eq(TeamUserDO::getUserId, param.getUserId())
|
||||
;
|
||||
|
||||
Page<TeamUserDO> page = new Page<>(param.getPageNo(), param.getPageSize());
|
||||
page.setSearchCount(param.getEnableReturnCount());
|
||||
IPage<TeamUserDO> iPage = teamUserMapper.selectPage(page, queryWrapper);
|
||||
|
||||
List<TeamUser> list = teamUserConverter.do2dto(iPage.getRecords());
|
||||
|
||||
fillData(list, selector);
|
||||
|
||||
return PageResult.of(list, iPage.getTotal(), param);
|
||||
}
|
||||
|
||||
@Override
|
||||
public PageResult<TeamUser> comprehensivePageQuery(TeamUserComprehensivePageQueryParam param,
|
||||
TeamUserSelector selector) {
|
||||
|
@ -30,7 +30,7 @@ import org.springframework.web.bind.annotation.RestController;
|
||||
* @author Jiaju Zhuang
|
||||
*/
|
||||
@RestController
|
||||
@RequestMapping("/oauth")
|
||||
@RequestMapping("/api/oauth")
|
||||
@Slf4j
|
||||
public class OauthController {
|
||||
|
||||
|
@ -5,7 +5,6 @@ import java.io.Serializable;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
|
||||
import ai.chat2db.server.tools.base.constant.EasyToolsConstant;
|
||||
import ai.chat2db.server.tools.base.enums.OrderByDirectionEnum;
|
||||
|
||||
@ -68,6 +67,14 @@ public class PageQueryParam implements Serializable {
|
||||
this.pageSize = Integer.MAX_VALUE;
|
||||
}
|
||||
|
||||
/**
|
||||
* 查询1条加速
|
||||
*/
|
||||
public void queryOne() {
|
||||
this.pageNo = 1;
|
||||
this.pageSize = 1;
|
||||
}
|
||||
|
||||
/**
|
||||
* 新增一个排序 并替换原有排序
|
||||
*
|
||||
|
@ -160,7 +160,6 @@ public class PageResult<T> implements Serializable, Result<List<T>> {
|
||||
return new PageResult<>(data, 0L, param.getPageNo(), param.getPageSize());
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 构建空的返回对象
|
||||
*
|
||||
@ -223,6 +222,15 @@ public class PageResult<T> implements Serializable, Result<List<T>> {
|
||||
return hasNextPage;
|
||||
}
|
||||
|
||||
/**
|
||||
* 判断是否存在数据
|
||||
*
|
||||
* @return 是否存在数据
|
||||
*/
|
||||
public boolean hasData() {
|
||||
return hasData(this);
|
||||
}
|
||||
|
||||
/**
|
||||
* 返回查询异常信息
|
||||
*
|
||||
|
@ -6,6 +6,7 @@ import ai.chat2db.server.admin.api.controller.user.request.UserTeamPageCommonQue
|
||||
import ai.chat2db.server.admin.api.controller.user.request.UserTeamBatchCreateRequest;
|
||||
import ai.chat2db.server.admin.api.controller.user.vo.UserTeamPageQueryVO;
|
||||
import ai.chat2db.server.domain.api.param.team.user.TeamUserCreatParam;
|
||||
import ai.chat2db.server.domain.api.param.team.user.TeamUserPageQueryParam;
|
||||
import ai.chat2db.server.domain.api.param.team.user.TeamUserSelector;
|
||||
import ai.chat2db.server.domain.api.service.TeamUserService;
|
||||
import ai.chat2db.server.tools.base.wrapper.result.ActionResult;
|
||||
@ -59,10 +60,19 @@ public class UserTeamAdminController {
|
||||
@PostMapping("/batch_create")
|
||||
public ActionResult bacthCreate(@Valid @RequestBody UserTeamBatchCreateRequest request) {
|
||||
request.getTeamIdList()
|
||||
.forEach(teamId -> teamUserService.create(TeamUserCreatParam.builder()
|
||||
.forEach(teamId -> {
|
||||
TeamUserPageQueryParam teamUserPageQueryParam = new TeamUserPageQueryParam();
|
||||
teamUserPageQueryParam.setTeamId(teamId);
|
||||
teamUserPageQueryParam.setUserId(request.getUserId());
|
||||
teamUserPageQueryParam.queryOne();
|
||||
if (teamUserService.pageQuery(teamUserPageQueryParam, null).hasData()) {
|
||||
return;
|
||||
}
|
||||
teamUserService.create(TeamUserCreatParam.builder()
|
||||
.teamId(teamId)
|
||||
.userId(request.getUserId())
|
||||
.build()));
|
||||
.build());
|
||||
});
|
||||
return ActionResult.isSuccess();
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user