From 880b395fab6a2f38209438733b0c4e54959f8964 Mon Sep 17 00:00:00 2001
From: lihongjie0209 <lihongjie0209@gmail.com>
Date: Tue, 12 Sep 2023 21:09:39 +0800
Subject: [PATCH] =?UTF-8?q?=E5=90=8C=E6=AD=A5=E7=85=A4=E6=BA=90=E4=BB=B7?=
 =?UTF-8?q?=E6=A0=BC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../coalInfo/service/CoalInfoService.java     | 23 +++++++++++++++----
 1 file changed, 18 insertions(+), 5 deletions(-)

diff --git a/src/main/java/cn/lihongjie/coal/coalInfo/service/CoalInfoService.java b/src/main/java/cn/lihongjie/coal/coalInfo/service/CoalInfoService.java
index ae3c61cd..62ee9283 100644
--- a/src/main/java/cn/lihongjie/coal/coalInfo/service/CoalInfoService.java
+++ b/src/main/java/cn/lihongjie/coal/coalInfo/service/CoalInfoService.java
@@ -9,7 +9,8 @@ import cn.lihongjie.coal.coalInfo.dto.UpdateCoalInfoDto;
 import cn.lihongjie.coal.coalInfo.entity.CoalInfoEntity;
 import cn.lihongjie.coal.coalInfo.mapper.CoalInfoMapper;
 import cn.lihongjie.coal.coalInfo.repository.CoalInfoRepository;
-import java.lang.String;
+import cn.lihongjie.coal.coalPrice.entity.CoalPriceEntity;
+import cn.lihongjie.coal.coalPrice.service.CoalPriceService;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.core.convert.ConversionService;
@@ -30,21 +31,33 @@ public class CoalInfoService extends BaseService<CoalInfoEntity, CoalInfoReposit
   @Autowired
   private ConversionService conversionService;
 
+  @Autowired
+  CoalPriceService coalPriceService;
   public CoalInfoDto create(CreateCoalInfoDto request) {
     CoalInfoEntity entity = mapper.toEntity(request);
 
 
-        this.repository.save(entity);
-        return getById(entity.getId())
+      this.repository.save(entity);
+
+      updatePrice(entity);
+      return getById(entity.getId())
         ;
   }
 
-  public CoalInfoDto update(UpdateCoalInfoDto request) {
+    private void updatePrice(CoalInfoEntity entity) {
+        CoalPriceEntity coalPrice = new CoalPriceEntity();
+        coalPrice.setCoalInfo(entity);
+        coalPrice.setPrice(entity.getInitPrice());
+
+        coalPriceService.save(coalPrice);
+    }
+
+    public CoalInfoDto update(UpdateCoalInfoDto request) {
     CoalInfoEntity entity = this.repository.get(request.getId());
         this.mapper.updateEntity(entity, request);
 
         this.repository.save(entity);
-
+        updatePrice(entity);
         return getById(entity.getId())
         ;
   }