一、簡單的Excel地址導入與樹狀結構生成

時間  2019-11-10
標籤 簡單 excel 地址 導入 樹狀 結構 生成 欄目 Microsoft Office 简体版
原文   原文鏈接

1、地區表結構java

DROP TABLE IF EXISTS `pro_area`;
/*!40101 SET @saved_cs_client     = @@character_set_client */;
/*!40101 SET character_set_client = utf8 */;
CREATE TABLE `pro_area` (
  `area_id` bigint(30) NOT NULL AUTO_INCREMENT COMMENT '主鍵ID',
  `area_code` varchar(64) NOT NULL COMMENT '地區編號',
  `parent_code` varchar(64) NOT NULL COMMENT '父分類編號 一級地區父地區編號=-1',
  `area_name` varchar(100) NOT NULL COMMENT '地區名稱',
  `area_state` int(2) DEFAULT NULL COMMENT '狀態:1:未啓用 2:已啓用 9:刪除',
  `create_time` timestamp NULL DEFAULT NULL COMMENT '建立時間',
  `update_time` timestamp NULL DEFAULT NULL COMMENT '修改時間',
  PRIMARY KEY (`area_id`)
) ENGINE=InnoDB AUTO_INCREMENT=3424 DEFAULT CHARSET=utf8 COMMENT='地區表';

2、導入的excel表結構node

3、導入本地數據到庫sql

public class MyBatisTest {
	
	
	
	public SqlSessionFactory getSqlSessionFactory() throws IOException {  
        String resource = "mybatis-config.xml";  
        InputStream inputStream = Resources.getResourceAsStream(resource);  
        return new SqlSessionFactoryBuilder().build(inputStream);  
    } 
	
	
	
	/**
	 * 三、導入區縣
	 * @throws Exception
	 */
	@Test
	public void insertCountyBatch()throws Exception{
		SqlSessionFactory sqlSessionFactory = getSqlSessionFactory();  
		SqlSession openSession = sqlSessionFactory.openSession();
		
		InputStream inp = new FileInputStream(new File("C:\\Users\\Administrator\\Desktop\\地址1.xlsx"));
		Workbook workbook=new XSSFWorkbook(inp);
		Sheet sheet = workbook.getSheetAt(0);
		
		Map<String,String> currentMap = new HashMap<>();
		for(int i=sheet.getFirstRowNum();i<=sheet.getLastRowNum();i++) {
			Row currRow = sheet.getRow(i);
			Cell currentCell = currRow.getCell(2);
			Cell parentCell = currRow.getCell(1);
			Cell ancestorCell = currRow.getCell(0);
			
			
			if(currentCell.getCellType()!=Cell.CELL_TYPE_BLANK) {
				currentMap.put(ancestorCell.getStringCellValue()+"_"+parentCell.getStringCellValue()+"-"+currentCell.getStringCellValue(), currentCell.getStringCellValue());
			}
		}
		
		
		List<ProArea> list = new ArrayList<>();
		Iterator it = currentMap.entrySet().iterator();
		ProArea proArea = null;
		while(it.hasNext()) {
			Map.Entry<String, String> node = (Entry<String, String>) it.next();
			proArea = new ProArea();
			proArea.setAreaCode(UUID.randomUUID().toString());
			proArea.setAreaName(node.getValue());
			proArea.setAreaState(2);
			proArea.setCreateTime(new Date(System.currentTimeMillis()));
			
			//查詢parentCode
			Map<String,String> params = new HashMap<>();
			String ancestorName = node.getKey().substring(0, node.getKey().indexOf("_"));
			String parantName = node.getKey().substring(node.getKey().indexOf("_")+1, node.getKey().indexOf("-"));
			params.put("parentName", parantName);
			params.put("ancestorName", ancestorName);
			String parentCode = openSession.selectOne("com.lee.poi.AreaMapper.selectAreaCodeByPNameAndAName",params);
			
			proArea.setParentCode(parentCode);
			list.add(proArea);
		}
		
		
		Integer insertCountyRes = openSession.insert("com.lee.poi.AreaMapper.insertAreaBatch", list);
		
		//關閉鏈接
		openSession.commit();
		openSession.close();
		
	}
	
	
	/**
	 * 二、導入市
	 */
	@Test
	public void insertCityBatch()throws Exception {
		SqlSessionFactory sqlSessionFactory = getSqlSessionFactory();  
		SqlSession openSession = sqlSessionFactory.openSession();
		
		InputStream inp = new FileInputStream(new File("C:\\Users\\Administrator\\Desktop\\地址1.xlsx"));
		Workbook workbook=new XSSFWorkbook(inp);
		Sheet sheet = workbook.getSheetAt(0);
		
		Map<String,String> currentMap = new HashMap<>();
		
		for(int i=sheet.getFirstRowNum();i<=sheet.getLastRowNum();i++) {
			Row currRow = sheet.getRow(i);
			Cell currentCell = currRow.getCell(1);
			Cell parentCell = currRow.getCell(0);
			
			if(currentCell.getCellType()!=Cell.CELL_TYPE_BLANK) {
				currentMap.put(parentCell.getStringCellValue()+"-"+currentCell.getStringCellValue(), currentCell.getStringCellValue());
			}
			
		}
		
		List<ProArea> list = new ArrayList<>();
		Iterator it = currentMap.entrySet().iterator();
		ProArea proArea = null;
		while(it.hasNext()) {
			Map.Entry<String, String> node = (Entry<String, String>) it.next();
			proArea = new ProArea();
			proArea.setAreaCode(UUID.randomUUID().toString());
			proArea.setAreaName(node.getValue());
			proArea.setAreaState(2);
			proArea.setCreateTime(new Date(System.currentTimeMillis()));
			
			//查詢parentCode
			String parentCode = openSession.selectOne("com.lee.poi.AreaMapper.selectAreaCodeByName",node.getKey().substring(0, node.getKey().indexOf("-")));
			
			proArea.setParentCode(parentCode);
			list.add(proArea);
		}
		
		
		Integer insertCityRes = openSession.insert("com.lee.poi.AreaMapper.insertAreaBatch", list);
		
		//關閉鏈接
		openSession.commit();
		openSession.close();
		
	}
	
	
	/**
	 * 一、導入省
	 */
	@Test
	public void insertProvinceBatch()throws Exception {
		SqlSessionFactory sqlSessionFactory = getSqlSessionFactory();  
		SqlSession openSession = sqlSessionFactory.openSession();
		
		InputStream inp = new FileInputStream(new File("C:\\Users\\Administrator\\Desktop\\地址1.xlsx"));
		Workbook workbook=new XSSFWorkbook(inp);
		Sheet sheet = workbook.getSheetAt(0);
		
		Map<String,String> currentMap = new HashMap<>();
		for(int i=sheet.getFirstRowNum();i<=sheet.getLastRowNum();i++) {
			Row currRow = sheet.getRow(i);
			Cell currentCell = currRow.getCell(0);
			if(currentCell.getCellType()!=Cell.CELL_TYPE_BLANK) {
				currentMap.put(currentCell.getStringCellValue(), currentCell.getStringCellValue());
			}
		}
		
		
		List<ProArea> list = new ArrayList<>();
		Iterator it = currentMap.entrySet().iterator();
		ProArea proArea = null;
		while(it.hasNext()) {
			Map.Entry<String, String> node = (Entry<String, String>) it.next();
			proArea = new ProArea();
			proArea.setAreaCode(UUID.randomUUID().toString());
			proArea.setAreaName(node.getValue());
			proArea.setAreaState(2);
			proArea.setCreateTime(new Date(System.currentTimeMillis()));
			proArea.setParentCode("-1");
			list.add(proArea);
		}
		
		Integer insertProvinceRes = openSession.insert("com.lee.poi.AreaMapper.insertAreaBatch", list);
		
		//關閉鏈接
		openSession.commit();
		openSession.close();
	}
	
	
	

}

4、對應的mapper文件mybatis

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper  PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"  "http://mybatis.org/dtd/mybatis-3-mapper.dtd">  

<mapper namespace="com.lee.poi.AreaMapper">  

    
    <select id="selectAreaByParentCode" parameterType="java.lang.String" resultType="com.lee.poi.ProArea">
    	select * from pro_area where parent_code = #{parentCode}
    </select>
    
    <!-- 根據父級名稱查詢code -->
    <select id="selectAreaCodeByName" parameterType="java.lang.String" resultType="java.lang.String">
    	select area_code from pro_area where area_name = #{areaName}
    </select>
    
    <!-- 根據父級名稱 和  父級的父級名稱查詢code -->
    <select id="selectAreaCodeByPNameAndAName" parameterType="java.util.Map" resultType="java.lang.String">
    	select p1.area_code 
		from pro_area p1
		inner join pro_area p2 on p1.parent_code = p2.area_code
		where p1.area_name=#{parentName} and p2.area_name=#{ancestorName}
    </select>
    
    <select id="selectProAreaTree" parameterType="java.util.Map" resultType="com.lee.poi.ProAreaTree">
    	select 
			area_code as value,
			parent_code as parentCode,
			area_name as text    	
    	from pro_area where parent_code = #{parentCode}
    	order by convert(text using gbk) asc
    </select>
    
    <insert id="insertAreaBatch" parameterType="java.util.List" useGeneratedKeys="true" keyProperty="areaId">
     	insert into pro_area
     		(area_code,parent_code,area_name,area_state,create_time)
     	values
     		<foreach collection="list" item="proArea" index="index" separator=",">
     			(
     				#{proArea.areaCode},
     				#{proArea.parentCode},
     				#{proArea.areaName},
     				#{proArea.areaState},
     				#{proArea.createTime}
     			)
     		</foreach>
    </insert>
    
    
</mapper>

 

 

5、導入後的表內容app

6、生成樹結構dom

public class MyBatisTest2 {
	
	
	
	public SqlSessionFactory getSqlSessionFactory() throws IOException {  
        String resource = "mybatis-config.xml";  
        InputStream inputStream = Resources.getResourceAsStream(resource);  
        return new SqlSessionFactoryBuilder().build(inputStream);  
    } 
	
	
	/**
	 * 四、地址樹狀結構
	 */
	@Test
	public void findProAreaTree()throws Exception {
		SqlSessionFactory sqlSessionFactory = getSqlSessionFactory();  
		SqlSession openSession = sqlSessionFactory.openSession();
		
		Map<String,Object> params = new HashMap<>();
		params.put("parentCode", "-1");
		List<ProAreaTree> nodes = openSession.selectList("com.lee.poi.AreaMapper.selectProAreaTree", params);
		findTree(nodes,openSession);
		
		System.out.println(JSON.toJSONString(nodes));
		
		//關閉鏈接
		openSession.commit();
		openSession.close();
	}
	
	private void findTree(List<ProAreaTree> parentNodes,SqlSession openSession) {
		if(parentNodes!=null && parentNodes.size()>0) {
			List<ProAreaTree> nodes = new ArrayList<>();
			Map<String,Object> params = null;
			for(int i=0;i<parentNodes.size();i++) {
				params = new HashMap<>();
				params.put("parentCode", parentNodes.get(i).getValue());
				nodes = openSession.selectList("com.lee.poi.AreaMapper.selectProAreaTree", params);
				parentNodes.get(i).setChildren(nodes);
				findTree(nodes,openSession);
			}
		}
	}
	
	
	
	

}

7、生成的樹結構ui

相關文章
相關標籤/搜索
每日一句
    每一个你不满意的现在，都有一个你没有努力的曾经。
本站公眾號
   歡迎關注本站公眾號,獲取更多信息
聯繫我們 最近搜索 最新文章 沪ICP备13005482号-10 MyBatis教程 SQL 教程 MySQL教程 Java 教程 Thymeleaf 教程 Hibernate教程 Spring教程 Redis教程