From 44e6152534e8f534368ae02e8523323219ea7a52 Mon Sep 17 00:00:00 2001
From: lihongjie0209 <lihongjie0209@gmail.com>
Date: Mon, 26 Feb 2024 19:56:53 +0800
Subject: [PATCH] =?UTF-8?q?=E5=88=9D=E5=A7=8B=E5=8C=96=E8=B4=A7=E5=93=81?=
 =?UTF-8?q?=E5=8D=95=E4=BD=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/WarehouseGoodsUnitService.java    | 25 +++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/src/main/java/cn/lihongjie/coal/warehouseGoodsUnit/service/WarehouseGoodsUnitService.java b/src/main/java/cn/lihongjie/coal/warehouseGoodsUnit/service/WarehouseGoodsUnitService.java
index 4165afa4..eb16ce9b 100644
--- a/src/main/java/cn/lihongjie/coal/warehouseGoodsUnit/service/WarehouseGoodsUnitService.java
+++ b/src/main/java/cn/lihongjie/coal/warehouseGoodsUnit/service/WarehouseGoodsUnitService.java
@@ -20,6 +20,9 @@ import org.springframework.data.domain.Sort;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import java.util.List;
+import java.util.Set;
+
 @Service
 @Slf4j
 @Transactional
@@ -68,4 +71,26 @@ public class WarehouseGoodsUnitService
 
         return page.map(this.mapper::toDto);
     }
+
+
+    public void init(String orgId) {
+
+        List<WarehouseGoodsUnitEntity> all = this.repository.findByOrganizationId(orgId);
+
+        if (!all.isEmpty()) {
+            return;
+        }
+
+        Set<String> units = Set.of(
+                "吨", "千克", "克", "升", "毫升", "个", "件", "台", "套", "箱", "包", "袋", "桶", "瓶", "罐", "卷",
+                "支", "根", "条", "片", "块", "颗", "粒", "对", "双", "只", "张", "本", "盒");
+
+
+        for (String unit : units) {
+            WarehouseGoodsUnitEntity entity = new WarehouseGoodsUnitEntity();
+            entity.setName(unit);
+            entity.setOrganizationId(orgId);
+            this.repository.save(entity);
+        }
+    }
 }