package cn.ksource.test.test;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.transaction.annotation.Transactional;

import cn.ksource.core.dao.BaseDao;

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations={"/applicationContext_core.xml"})
@Transactional
public class TreeTest {

	@Autowired
	private BaseDao baseDao;
	
	
	@Test
	public void treeTest() {
		String sql = "SELECT * FROM CMS_CATEGORY WHERE USINGSTATE = 1 ORDER BY LEVEL,ORDERNUM";
		List<Map> treeList = baseDao.queryForList(sql);
		
		List<Map> resultList = new ArrayList<Map>();
		
		Map<String,Map> yjgnMap = new HashMap<String, Map>();
		Map<String,Map> ejgnMap = new HashMap<String, Map>();
		
		for(Map map : treeList) {
			if(map.get("LEVEL").toString().equals("1")) {
				yjgnMap.put(map.get("ID").toString(), map);
				List<Map> ejgnList = new LinkedList<Map>();
				map.put("ejgn", ejgnList);
				resultList.add(map);
				continue;
			}
			
			if(map.get("LEVEL").toString().equals("2")) {
				Map yjgn = yjgnMap.get(map.get("UP_ID").toString());
				List<Map> ejgn = (List<Map>)yjgn.get("ejgn");
				List<Map> sjgnList = new LinkedList<Map>();
				map.put("sjgn", sjgnList);
				ejgn.add(map);
				ejgnMap.put(map.get("ID").toString(), map);
				continue;
			}
		}
		System.out.println(resultList);
	}
}