topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-10 19:51:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[improvement][Headless] Supports getting domain by bizName

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/DomainRepository.java
View File

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

9
headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/impl/DomainRepositoryImpl.java
View File

@@ -1,5 +1,6 @@
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.tencent.supersonic.headless.server.persistence.dataobject.DomainDO;
import com.tencent.supersonic.headless.server.persistence.mapper.DomainDOMapper;
@@ -43,4 +44,12 @@ public class DomainRepositoryImpl implements DomainRepository {
public DomainDO getDomainById(Long 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);
}
}

4
headless/server/src/main/java/com/tencent/supersonic/headless/server/rest/ModelController.java
View File

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

3
headless/server/src/main/java/com/tencent/supersonic/headless/server/service/DomainService.java
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.DomainUpdateReq;
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.Map;
@@ -32,5 +33,7 @@ public interface DomainService {
Set<DomainResp> getDomainAuthSet(User user, AuthType authTypeEnum);
List<DomainDO> getDomainByBizName(String bizName);
Set<DomainResp> getDomainChildren(List<Long> domainId);
}

5
headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/DomainServiceImpl.java
View File

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