From f500082ea6adfc0ae6c4bc6ae23fbf9a2f2b746a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=83=9D=E5=BD=AC?=
 <14779346+xswl_123@user.noreply.gitee.com>
Date: Wed, 13 Aug 2025 10:08:35 +0800
Subject: [PATCH] =?UTF-8?q?8-13=20=E9=A1=B9=E7=9B=AE=E7=AB=8B=E9=A1=B9?=
 =?UTF-8?q?=E6=97=B6=E4=BD=BF=E5=90=88=E5=90=8C=E5=90=8C=E6=AD=A5=E5=85=B3?=
 =?UTF-8?q?=E8=81=94=E8=AF=A5=E9=A1=B9=E7=9B=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../dite/znpt/service/impl/ProjectServiceImpl.java   | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/core/src/main/java/com/dite/znpt/service/impl/ProjectServiceImpl.java b/core/src/main/java/com/dite/znpt/service/impl/ProjectServiceImpl.java
index 38d8b40..198ec2e 100644
--- a/core/src/main/java/com/dite/znpt/service/impl/ProjectServiceImpl.java
+++ b/core/src/main/java/com/dite/znpt/service/impl/ProjectServiceImpl.java
@@ -3,9 +3,11 @@ package com.dite.znpt.service.impl;
 import cn.hutool.core.bean.BeanUtil;
 import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.util.StrUtil;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.dite.znpt.constant.Message;
 import com.dite.znpt.converts.Converts;
+import com.dite.znpt.domain.entity.ContractEntity;
 import com.dite.znpt.domain.entity.ProjectEntity;
 import com.dite.znpt.domain.entity.ProjectTaskEntity;
 import com.dite.znpt.domain.entity.UserEntity;
@@ -16,12 +18,14 @@ import com.dite.znpt.domain.vo.ProjectReq;
 import com.dite.znpt.domain.vo.ProjectResp;
 import com.dite.znpt.enums.ProjectStatusEnum;
 import com.dite.znpt.exception.ServiceException;
+import com.dite.znpt.mapper.ContractMapper;
 import com.dite.znpt.mapper.ProjectMapper;
 import com.dite.znpt.service.ProjectService;
 import com.dite.znpt.service.UserService;
 import com.dite.znpt.service.ProjectTaskService;
 import com.dite.znpt.util.PageUtil;
 import lombok.RequiredArgsConstructor;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
@@ -47,6 +51,9 @@ public class ProjectServiceImpl extends ServiceImpl<ProjectMapper, ProjectEntity
 
     @Resource
     private ProjectTaskService projectTaskService;
+    @Autowired
+    private ContractMapper contractMapper;
+
     /**
      * 功能描述：查询项目信息列表
      *
@@ -135,6 +142,11 @@ public class ProjectServiceImpl extends ServiceImpl<ProjectMapper, ProjectEntity
     public void save(ProjectReq req) {
         ProjectEntity entity = Converts.INSTANCE.toProjectEntity(req);
         this.save(entity);
+        QueryWrapper<ContractEntity> queryWrapper = new QueryWrapper<>();
+        queryWrapper.eq("contract_code", req.getProjectOrigin());
+        ContractEntity contractEntity = contractMapper.selectOne(queryWrapper);
+        contractEntity.setProjectId(entity.getProjectId());
+        contractMapper.updateById(contractEntity);
         for (ProjectInitTaskReq taskReq : req.getTasks()) {
             ProjectTaskEntity taskEntity = BeanUtil.copyProperties(taskReq, ProjectTaskEntity.class);
             if (taskEntity.getScales() == null) {