From 873344c752b9ac94a6dbbb0ce38df0578ad3ec47 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 23 Sep 2025 13:57:50 +0800
Subject: [PATCH] ``` feat(ai):管理 新增工作流功能
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java | 67 +++++++++++++++++++++++++++++++++
1 files changed, 66 insertions(+), 1 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java
index ffeb9e1..f68864b 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java
@@ -1,8 +1,10 @@
package cc.mrbird.febs.ai.service.impl;
import cc.mrbird.febs.ai.entity.AiCompany;
+import cc.mrbird.febs.ai.entity.AiCompanyWorkflow;
import cc.mrbird.febs.ai.entity.AiMemberRole;
import cc.mrbird.febs.ai.mapper.AiCompanyMapper;
+import cc.mrbird.febs.ai.mapper.AiCompanyWorkflowMapper;
import cc.mrbird.febs.ai.service.AiCompanyService;
import cc.mrbird.febs.ai.util.UUID;
import cc.mrbird.febs.common.entity.FebsResponse;
@@ -25,6 +27,7 @@
public class AiCompanyServiceImpl extends ServiceImpl<AiCompanyMapper, AiCompany> implements AiCompanyService {
private final AiCompanyMapper aiCompanyMapper;
+ private final AiCompanyWorkflowMapper aiCompanyWorkflowMapper;
@Override
public AiCompany getById(String id) {
@@ -33,11 +36,21 @@
@Override
public FebsResponse add(AiCompany dto) {
+ Date createTime = new Date();
AiCompany entity = new AiCompany();
entity.setId(UUID.getSimpleUUIDString());
entity.setName(dto.getName());
- entity.setCreatedTime(new Date());
+ entity.setCreatedTime(createTime);
this.baseMapper.insert( entity);
+
+
+
+ this.addWorkflow(entity.getId(),1,null,createTime);
+ this.addWorkflow(entity.getId(),2,null,createTime);
+ this.addWorkflow(entity.getId(),3,null,createTime);
+ this.addWorkflow(entity.getId(),4,null,createTime);
+ this.addWorkflow(entity.getId(),6,null,createTime);
+
return new FebsResponse().success().message("操作成功");
}
@@ -64,4 +77,56 @@
Page<AiCompany> pages = this.baseMapper.selectPage(page, query);
return pages;
}
+
+ @Override
+ public AiCompanyWorkflow getWorkflowById(String id) {
+ return aiCompanyWorkflowMapper.selectById( id);
+ }
+
+ @Override
+ public IPage<AiCompanyWorkflow> workflowListInPage(AiCompanyWorkflow dto, QueryRequest request) {
+ Page<AiCompanyWorkflow> page = new Page<>(request.getPageNum(), request.getPageSize());
+ LambdaQueryWrapper<AiCompanyWorkflow> query = Wrappers.lambdaQuery(AiCompanyWorkflow.class);
+ query.orderByDesc(AiCompanyWorkflow::getCreatedTime);
+ Page<AiCompanyWorkflow> pages = aiCompanyWorkflowMapper.selectPage(page, query);
+ return pages;
+ }
+
+ @Override
+ public FebsResponse addWorkflow(AiCompanyWorkflow dto) {
+ AiCompanyWorkflow entity = new AiCompanyWorkflow();
+ entity.setId(UUID.getSimpleUUIDString());
+ entity.setType(dto.getType());
+ entity.setCode(dto.getCode());
+ entity.setCreatedTime(new Date());
+ aiCompanyWorkflowMapper.insert( entity);
+ return new FebsResponse().success().message("操作成功");
+ }
+
+ @Override
+ public FebsResponse updateWorkflow(AiCompanyWorkflow dto) {
+ String id = dto.getId();
+ AiCompanyWorkflow entity = this.getWorkflowById(id);
+ if(ObjectUtil.isNotNull( entity)){
+ aiCompanyWorkflowMapper.update(null,
+ Wrappers.lambdaUpdate(AiCompanyWorkflow.class)
+ .set(AiCompanyWorkflow::getType, dto.getType())
+ .set(AiCompanyWorkflow::getCode, dto.getCode())
+ .set(AiCompanyWorkflow::getUpdatedTime, new Date())
+ .eq(AiCompanyWorkflow::getId, id));
+
+ }
+ return new FebsResponse().success().message("操作成功");
+ }
+
+ @Override
+ public void addWorkflow(String companyId, int type,String code, Date createTime) {
+ AiCompanyWorkflow entity = new AiCompanyWorkflow();
+ entity.setId(UUID.getSimpleUUIDString());
+ entity.setCompanyId(companyId);
+ entity.setType(type);
+ entity.setCode(code);
+ entity.setCreatedTime(createTime);
+ aiCompanyWorkflowMapper.insert( entity);
+ }
}
--
Gitblit v1.9.1