From 149d04de35cf9f6adb2f568da141bfd5f5a27329 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Tue, 12 Mar 2024 12:07:55 +0800
Subject: [PATCH] 新增天地图标签导入

---
 src/main/java/org/springblade/modules/category/service/impl/CategoryServiceImpl.java |   30 ++++++++++++++++++++++++++++++
 1 files changed, 30 insertions(+), 0 deletions(-)

diff --git a/src/main/java/org/springblade/modules/category/service/impl/CategoryServiceImpl.java b/src/main/java/org/springblade/modules/category/service/impl/CategoryServiceImpl.java
index 8749263..704d5aa 100644
--- a/src/main/java/org/springblade/modules/category/service/impl/CategoryServiceImpl.java
+++ b/src/main/java/org/springblade/modules/category/service/impl/CategoryServiceImpl.java
@@ -16,11 +16,13 @@
  */
 package org.springblade.modules.category.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.springblade.common.node.TreeIntegerNode;
 import org.springblade.common.utils.NodeTreeUtil;
 import org.springblade.modules.category.dto.CategoryDTO;
 import org.springblade.modules.category.entity.CategoryEntity;
+import org.springblade.modules.category.excel.CategoryExcel;
 import org.springblade.modules.category.vo.CategoryVO;
 import org.springblade.modules.category.mapper.CategoryMapper;
 import org.springblade.modules.category.service.ICategoryService;
@@ -77,4 +79,32 @@
 			node.setChildren(null);
 		}
 	}
+
+	/**
+	 * 标签导入
+	 * @param data
+	 * @param isCovered
+	 */
+	@Override
+	public void importCategory(List<CategoryExcel> data, Boolean isCovered) {
+		for (CategoryExcel categoryExcel : data) {
+			// 查询是否已存在
+			QueryWrapper<CategoryEntity> wrapper = new QueryWrapper<>();
+			wrapper.eq("category_no",categoryExcel.getCategoryNo()).eq("is_deleted",0);
+			CategoryEntity one = getOne(wrapper);
+			if (null==one){
+				// 新增
+				CategoryEntity categoryEntity = new CategoryEntity();
+				categoryEntity.setCategoryNo(categoryExcel.getCategoryNo());
+				categoryEntity.setCategoryName(categoryExcel.getCategoryName());
+				categoryEntity.setLevel(categoryExcel.getLevel());
+				categoryEntity.setIsDeleted(0);
+				categoryEntity.setParentNo(categoryExcel.getCategoryNo().substring(0,4));
+				categoryEntity.setDescription(categoryExcel.getDescription());
+				categoryEntity.setRemark(categoryExcel.getRemark());
+				// 新增
+				save(categoryEntity);
+			}
+		}
+	}
 }

--
Gitblit v1.9.3