初始化货品单位

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);
+        }
+    }
 }