[improvement][Headless] Supports getting domain by bizName

This commit is contained in:
lxwcodemonkey
2024-11-30 23:17:49 +08:00
parent 6ecc5a9362
commit 5837a5b3ae
5 changed files with 21 additions and 2 deletions

View File

@@ -15,4 +15,6 @@ public interface DomainRepository {
List<DomainDO> getDomainList(); List<DomainDO> getDomainList();
DomainDO getDomainById(Long id); DomainDO getDomainById(Long id);
List<DomainDO> getDomainByBizName(String bizName);
} }

View File

@@ -1,5 +1,6 @@
package com.tencent.supersonic.headless.server.persistence.repository.impl; package com.tencent.supersonic.headless.server.persistence.repository.impl;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.tencent.supersonic.headless.server.persistence.dataobject.DomainDO; import com.tencent.supersonic.headless.server.persistence.dataobject.DomainDO;
import com.tencent.supersonic.headless.server.persistence.mapper.DomainDOMapper; import com.tencent.supersonic.headless.server.persistence.mapper.DomainDOMapper;
@@ -43,4 +44,12 @@ public class DomainRepositoryImpl implements DomainRepository {
public DomainDO getDomainById(Long id) { public DomainDO getDomainById(Long id) {
return domainDOMapper.selectById(id); return domainDOMapper.selectById(id);
} }
@Override
public List<DomainDO> getDomainByBizName(String bizName) {
QueryWrapper<DomainDO> queryWrapper = new QueryWrapper<>();
queryWrapper.lambda().eq(DomainDO::getBizName, bizName);
return domainDOMapper.selectList(queryWrapper);
}
} }

View File

@@ -51,8 +51,8 @@ public class ModelController {
} }
@PostMapping("/createModelBatch") @PostMapping("/createModelBatch")
public Boolean createModelBatch(@RequestBody ModelBuildReq modelBuildReq, HttpServletRequest request, public Boolean createModelBatch(@RequestBody ModelBuildReq modelBuildReq,
HttpServletResponse response) throws Exception { HttpServletRequest request, HttpServletResponse response) throws Exception {
User user = UserHolder.findUser(request, response); User user = UserHolder.findUser(request, response);
modelService.createModel(modelBuildReq, user); modelService.createModel(modelBuildReq, user);
return true; return true;

View File

@@ -5,6 +5,7 @@ import com.tencent.supersonic.common.pojo.enums.AuthType;
import com.tencent.supersonic.headless.api.pojo.request.DomainReq; import com.tencent.supersonic.headless.api.pojo.request.DomainReq;
import com.tencent.supersonic.headless.api.pojo.request.DomainUpdateReq; import com.tencent.supersonic.headless.api.pojo.request.DomainUpdateReq;
import com.tencent.supersonic.headless.api.pojo.response.DomainResp; import com.tencent.supersonic.headless.api.pojo.response.DomainResp;
import com.tencent.supersonic.headless.server.persistence.dataobject.DomainDO;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
@@ -32,5 +33,7 @@ public interface DomainService {
Set<DomainResp> getDomainAuthSet(User user, AuthType authTypeEnum); Set<DomainResp> getDomainAuthSet(User user, AuthType authTypeEnum);
List<DomainDO> getDomainByBizName(String bizName);
Set<DomainResp> getDomainChildren(List<Long> domainId); Set<DomainResp> getDomainChildren(List<Long> domainId);
} }

View File

@@ -179,6 +179,11 @@ public class DomainServiceImpl implements DomainService {
.collect(Collectors.toMap(DomainResp::getId, a -> a, (k1, k2) -> k1)); .collect(Collectors.toMap(DomainResp::getId, a -> a, (k1, k2) -> k1));
} }
@Override
public List<DomainDO> getDomainByBizName(String bizName) {
return domainRepository.getDomainByBizName(bizName);
}
@Override @Override
public Set<DomainResp> getDomainChildren(List<Long> domainIds) { public Set<DomainResp> getDomainChildren(List<Long> domainIds) {
Set<DomainResp> childDomains = new HashSet<>(); Set<DomainResp> childDomains = new HashSet<>();