From 65996f5be5ac43ed523ef47a77e4b328d92e536b6fd5667a2d63b095611d55f4 Mon Sep 17 00:00:00 2001
From: attiya <2413103649@qq.com>
Date: Thu, 18 Dec 2025 16:48:46 +0800
Subject: [PATCH] =?UTF-8?q?=E8=BF=90=E8=90=A5=E5=8C=BA=E7=AB=99=E7=82=B9?=
 =?UTF-8?q?=E6=95=B0=E9=87=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/EbikeSiteServiceImpl.java    | 27 ++++++++++++++++++-
 1 file changed, 26 insertions(+), 1 deletion(-)

diff --git a/ebike-operations/src/main/java/com/cdzy/operations/service/impl/EbikeSiteServiceImpl.java b/ebike-operations/src/main/java/com/cdzy/operations/service/impl/EbikeSiteServiceImpl.java
index 10bbb80..a2ec79d 100644
--- a/ebike-operations/src/main/java/com/cdzy/operations/service/impl/EbikeSiteServiceImpl.java
+++ b/ebike-operations/src/main/java/com/cdzy/operations/service/impl/EbikeSiteServiceImpl.java
@@ -2,12 +2,19 @@ package com.cdzy.operations.service.impl;
 
 import cn.dev33.satoken.stp.StpUtil;
 import com.cdzy.common.ex.EbikeException;
+import com.cdzy.operations.mapper.EbikeRegionMapper;
+import com.cdzy.operations.model.entity.EbikeRegion;
 import com.cdzy.operations.model.vo.EbikeSiteVo;
+import com.mybatisflex.core.query.QueryWrapper;
 import com.mybatisflex.spring.service.impl.ServiceImpl;
 import com.cdzy.operations.model.entity.EbikeSite;
 import com.cdzy.operations.mapper.EbikeSiteMapper;
 import com.cdzy.operations.service.EbikeSiteService;
+import jakarta.annotation.Resource;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import static com.cdzy.operations.model.entity.table.EbikeRegionTableDef.EBIKE_REGION;
 
 /**
  * 服务层实现。
@@ -18,9 +25,20 @@ import org.springframework.stereotype.Service;
 @Service
 public class EbikeSiteServiceImpl extends ServiceImpl<EbikeSiteMapper, EbikeSite> implements EbikeSiteService {
 
+    @Resource
+    private EbikeRegionMapper regionMapper;
+
+    @Transactional
     @Override
     public void saveSite(EbikeSiteVo ebikeSite) {
-                     EbikeSite site = EbikeSite.builder()
+        QueryWrapper queryWrapper = QueryWrapper.create()
+                .where(EBIKE_REGION.REGION_ID.eq(ebikeSite.getRegionId()));
+        EbikeRegion region = regionMapper.selectOneByQuery(queryWrapper);
+        if (region == null) {
+            throw new EbikeException("所选运营区不存在");
+        }
+        region.setSiteNum(region.getSiteNum() + 1);
+        EbikeSite site = EbikeSite.builder()
                 .siteName(ebikeSite.getSiteName())
                 .sitePolygon(ebikeSite.getSitePolygon())
                 .operatorId(ebikeSite.getOperatorId())
@@ -29,6 +47,7 @@ public class EbikeSiteServiceImpl extends ServiceImpl<EbikeSiteMapper, EbikeSite
                 .createdBy(StpUtil.getLoginIdAsLong())
                 .build();
         this.mapper.insert(site);
+        regionMapper.update(region);
     }
 
     @Override
@@ -38,6 +57,12 @@ public class EbikeSiteServiceImpl extends ServiceImpl<EbikeSiteMapper, EbikeSite
         if (site == null) {
             throw new EbikeException("站点不存在");
         }
+        QueryWrapper queryWrapper = QueryWrapper.create()
+                .where(EBIKE_REGION.REGION_ID.eq(ebikeSite.getRegionId()));
+        EbikeRegion region = regionMapper.selectOneByQuery(queryWrapper);
+        if (region == null) {
+            throw new EbikeException("所选运营区不存在");
+        }
         site.setSiteName(ebikeSite.getSiteName());
         site.setSitePolygon(ebikeSite.getSitePolygon());
         site.setOperatorId(ebikeSite.getOperatorId());