Browse Source

配置文件修改

lrj 5 months ago
parent
commit
e5b99fd39f

+ 6 - 4
PCS/src/main/java/cn/cslg/permission/controller/qiaobi/QiaoBiPersonController.java

@@ -17,10 +17,7 @@ import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import lombok.RequiredArgsConstructor;
 import org.springframework.context.annotation.Lazy;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestBody;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
 
 import javax.validation.Valid;
 import java.util.List;
@@ -77,4 +74,9 @@ public class QiaoBiPersonController {
     public String queryInvitationMessage(@RequestBody GetPersonRelationDTO getPersonRelationDTO) {
        return personRelationService.queryInvitationMessage(getPersonRelationDTO);
     }
+    @GetMapping("getAIApiKey")
+    @Operation(summary = "获取AiKey")
+    public String getAIApiKey() {
+        return qiaoBiPersonService.getAiApiKey();
+    }
 }

+ 27 - 3
PCS/src/main/java/cn/cslg/permission/service/qiaobi/QiaoBiPersonService.java

@@ -8,15 +8,15 @@ import cn.cslg.permission.common.model.qiaobi.QiaoBiVipTypeDTO;
 import cn.cslg.permission.common.model.qiaobi.tenant.CompanyPersonSignUpDTO;
 import cn.cslg.permission.common.model.vo.LoginVO;
 import cn.cslg.permission.common.model.vo.PersonnelVO;
-import cn.cslg.permission.common.utils.RedisUtil;
-import cn.cslg.permission.common.utils.RegexUtil;
-import cn.cslg.permission.common.utils.StringUtils;
+import cn.cslg.permission.common.utils.*;
 import cn.cslg.permission.domain.PersonInvitationCode;
 import cn.cslg.permission.domain.Personnel;
 import cn.cslg.permission.domain.qiaobi.BusinessTask;
 import cn.cslg.permission.domain.qiaobi.RegistInformation;
+import cn.cslg.permission.domain.sys.SysDictItem;
 import cn.cslg.permission.exception.ExceptionEnum;
 import cn.cslg.permission.exception.XiaoShiException;
+import cn.cslg.permission.mapper.SysDictItemMapper;
 import cn.cslg.permission.service.LoginService;
 import cn.cslg.permission.service.PersonInvitationCodeService;
 import cn.cslg.permission.service.PersonnelService;
@@ -57,6 +57,12 @@ public class QiaoBiPersonService {
     private MailSendService mailSendService;
     @Autowired
     private PersonRelationService personRelationService;
+    @Autowired
+    private LoginUtils loginUtils;
+    @Autowired
+    private CacheUtils cacheUtils;
+    @Autowired
+    private SysDictItemMapper sysDictItemMapper;
 
     @Transactional(rollbackFor = Exception.class)
     public QiaoBiPersonSignUpDTO personSignUp(QiaoBiPersonSignUpDTO qiaoBiPersonSignUpDTO) {
@@ -275,4 +281,22 @@ public class QiaoBiPersonService {
         }
 
     }
+
+    public String getAiApiKey() {
+        String re = null;
+        PersonnelVO personnelVO = cacheUtils.getLoginUser(loginUtils.getId());
+//        PersonnelVO personnelVO=new PersonnelVO();
+//        personnelVO.setTenantId(1);
+        Integer tenantId = personnelVO.getTenantId();
+        if (tenantId != null && tenantId.equals(1)) {
+            LambdaQueryWrapper<SysDictItem> queryWrapper = new LambdaQueryWrapper<>();
+            queryWrapper.eq(SysDictItem::getDictParentKey, "AI_API_KEY");
+            List<SysDictItem> sysDictItems = sysDictItemMapper.selectList(queryWrapper);
+            if (sysDictItems.size() > 0) {
+                SysDictItem sysDictItem = sysDictItems.get(0);
+                re = sysDictItem.getDictChildValue();
+            }
+        }
+        return re;
+    }
 }

+ 5 - 0
PCS/src/test/java/cn/cslg/permission/qiaobi/QiaobiPersonTest.java

@@ -60,4 +60,9 @@ public void CompanyPersonSignUp(){
      String  a=personRelationService.queryInvitationMessage(getPersonRelationDTO);
      System.out.println(a);
     }
+    @Test
+    public void getAiApiKey(){
+      String re= qiaoBiPersonService.getAiApiKey();
+      System.out.println(re);
+    }
 }