From 4ee86eb0f4984bf3ede3196ad8c5fe95e8c5a504 Mon Sep 17 00:00:00 2001
From: java110 <928255095@qq.com>
Date: 星期三, 15 六月 2022 19:05:02 +0800
Subject: [PATCH] 优化代码

---
 service-api/src/main/java/com/java110/api/bmo/unit/impl/UnitBMOImpl.java |    6 +++++-
 1 files changed, 5 insertions(+), 1 deletions(-)

diff --git a/service-api/src/main/java/com/java110/api/bmo/unit/impl/UnitBMOImpl.java b/service-api/src/main/java/com/java110/api/bmo/unit/impl/UnitBMOImpl.java
old mode 100644
new mode 100755
index 6b396bf..cdbe354
--- a/service-api/src/main/java/com/java110/api/bmo/unit/impl/UnitBMOImpl.java
+++ b/service-api/src/main/java/com/java110/api/bmo/unit/impl/UnitBMOImpl.java
@@ -4,6 +4,7 @@
 import com.java110.api.bmo.ApiBaseBMO;
 import com.java110.api.bmo.unit.IUnitBMO;
 import com.java110.core.context.DataFlowContext;
+import com.java110.core.factory.GenerateCodeFactory;
 import com.java110.po.unit.UnitPo;
 import com.java110.utils.constant.BusinessTypeConstant;
 import com.java110.utils.constant.CommonConstant;
@@ -48,10 +49,12 @@
         JSONObject businessUnit = new JSONObject();
         businessUnit.put("floorId", paramInJson.getString("floorId"));
         businessUnit.put("layerCount", paramInJson.getString("layerCount"));
-        businessUnit.put("unitId", "-1");
+        businessUnit.put("unitId", !paramInJson.containsKey("unitId") ? GenerateCodeFactory.getGeneratorId(GenerateCodeFactory.CODE_PREFIX_unitId)
+                : paramInJson.getString("unitId"));
         businessUnit.put("unitNum", paramInJson.getString("unitNum"));
         businessUnit.put("lift", paramInJson.getString("lift"));
         businessUnit.put("remark", paramInJson.getString("remark"));
+        businessUnit.put("unitArea", paramInJson.getString("unitArea"));
         businessUnit.put("userId", dataFlowContext.getRequestCurrentHeaders().get(CommonConstant.HTTP_USER_ID));
         UnitPo unitPo = BeanConvertUtil.covertBean(businessUnit, UnitPo.class);
 
@@ -74,6 +77,7 @@
         businessUnit.put("unitNum", paramInJson.getString("unitNum"));
         businessUnit.put("lift", paramInJson.getString("lift"));
         businessUnit.put("remark", paramInJson.getString("remark"));
+        businessUnit.put("unitArea", paramInJson.getString("unitArea"));
         businessUnit.put("userId", dataFlowContext.getRequestCurrentHeaders().get(CommonConstant.HTTP_USER_ID));
         UnitPo unitPo = BeanConvertUtil.covertBean(businessUnit, UnitPo.class);
 

--
Gitblit v1.8.0