From 5837a5b3aef0c6344a853eb0a19f81a6eead9568 Mon Sep 17 00:00:00 2001
From: lxwcodemonkey <jolunoluo@tencent.com>
Date: Sat, 30 Nov 2024 23:17:49 +0800
Subject: [PATCH] [improvement][Headless] Supports getting domain by bizName

---
 .../server/persistence/repository/DomainRepository.java  | 2 ++
 .../repository/impl/DomainRepositoryImpl.java            | 9 +++++++++
 .../supersonic/headless/server/rest/ModelController.java | 4 ++--
 .../headless/server/service/DomainService.java           | 3 +++
 .../headless/server/service/impl/DomainServiceImpl.java  | 5 +++++
 5 files changed, 21 insertions(+), 2 deletions(-)

diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/DomainRepository.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/DomainRepository.java
index 4ee3f9db7..169efad58 100644
--- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/DomainRepository.java
+++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/DomainRepository.java
@@ -15,4 +15,6 @@ public interface DomainRepository {
     List<DomainDO> getDomainList();
 
     DomainDO getDomainById(Long id);
+
+    List<DomainDO> getDomainByBizName(String bizName);
 }
diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/impl/DomainRepositoryImpl.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/impl/DomainRepositoryImpl.java
index 30267b57a..4783b7f80 100644
--- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/impl/DomainRepositoryImpl.java
+++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/impl/DomainRepositoryImpl.java
@@ -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);
+    }
+
 }
diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/rest/ModelController.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/rest/ModelController.java
index dd552636b..e535b4c54 100644
--- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/rest/ModelController.java
+++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/rest/ModelController.java
@@ -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;
diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/DomainService.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/DomainService.java
index 793b66bf7..2175f3627 100644
--- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/DomainService.java
+++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/DomainService.java
@@ -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);
 }
diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/DomainServiceImpl.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/DomainServiceImpl.java
index be75ac482..8199db64d 100644
--- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/DomainServiceImpl.java
+++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/DomainServiceImpl.java
@@ -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<>();