一、功能实现
1、节点树查询：按ID查询树2、节点新增：http://host/tree_data/node/${treeId}in: {node: {key: ..., ...}, parent: ${key}, sequ: ${sequ}}3、节点修改http://host/tree_data/node/${treeId}/${key}in: {node: {key: ..., ...}}4、节点删除http://host/tree_data/node/${treeId}/${key} 二、数据表结构
CREATE TABLE `catagory_tree` ( `id` bigint(11) NOT NULL AUTO_INCREMENT, `name` varchar(32) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '分类树名称：PRODUCT：产品分类 GOODS：商品分类', `json_tree` text CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '直接存储ANTD树表的数据结构', `modify_staff` varchar(32) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL, `modify_time` datetime(0) NOT NULL, PRIMARY KEY (`id`) USING BTREE, INDEX `AK_uni_catagory_tree_name`(`name`) USING BTREE) ENGINE = InnoDB AUTO_INCREMENT = 2 CHARACTER SET = utf8 COLLATE = utf8_general_ci ROW_FORMAT = Compact;
三、json字符串
[{ "id": 1, "code": "FLOW_NODE_1", "name": "环节A", "children": [{ "id": 2, "code": "RULE_NODE_1", "name": "规则A" }, { "id": 3, "code": "RULE_NODE_2", "name": "规则B" }, { "id": 4, "code": "PARALLEL_NODE_2", "name": "并行节点", "children": [{ "id": 5, "code": "RULE_NODE_3", "name": "规则C1" }, { "id": 6, "code": "RULE_COLLECTION_1", "name": "规则集1", "children": [{ "id": 7, "code": "RULE_NODE_4", "name": "规则C21" }, { "id": 8, "code": "RULE_NODE_5", "name": "规则C22" }] }] }, { "id": 9, "code": "MUTUAL_NODE_1", "name": "互斥节点", "children": [{ "id": 10, "code": "RULE_NODE_6", "name": "规则D1" }, { "id": 11, "code": "RULE_NODE_7", "name": "规则D2" }] }]}, { "id": 12, "code": "FLOW_NODE_2", "name": "环节B"}] 四、pom文件依赖
com.alibaba fastjson 1.2.49
五、controller层
package cn.chinaunicom.srigz.goods.admin.controller;import cn.chinaunicom.srigz.goods.admin.service.CatagoryTreeService;import cn.chinaunicom.srigz.goods.admin.utils.ActionHelper;import io.swagger.annotations.ApiOperation;import io.swagger.annotations.ApiParam;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.web.bind.annotation.*;import java.util.Map;@RestControllerpublic class CatagoryTreeController { @Autowired private CatagoryTreeService catagoryTreeService; @ApiOperation(value="查询分类树信息", notes="输入分类树Id") @RequestMapping(value = "/api/v1/tree_data/node/{treeId}", method = RequestMethod.GET) public Map getOne(@ApiParam(required = true, value = "分类树ID") @PathVariable Integer treeId){ return ActionHelper.responseOk(catagoryTreeService.getOne(treeId)); } @ApiOperation(value="新建节点", notes="根据输入参数创建节点") @RequestMapping(value = "/tree_data/node/{treeId}", method = RequestMethod.POST) public Map addOne(@ApiParam(required = true, value = "分类树ID")@PathVariable Integer treeId, @ApiParam(required = true, value = "节点信息、父类ID、数组位置") @RequestBody Map map){ return ActionHelper.responseOk(catagoryTreeService.addOne(treeId,map)); } @ApiOperation(value="更新节点信息", notes="更新节点信息") @RequestMapping(value = "/tree_data/node/{treeId}/{id}", method = RequestMethod.PATCH) public Map updateOne(@ApiParam(required = true, value = "分类树ID")@PathVariable Integer treeId, @ApiParam(required = true, value = "节点ID")@PathVariable Integer id, @ApiParam(required = true, value = "节点信息") @RequestBody Map map){ return ActionHelper.responseOk(catagoryTreeService.updateOne(treeId,id,map)); } @ApiOperation(value="删除节点详情", notes="删除节点详情") @RequestMapping(method = RequestMethod.DELETE,value = "/tree_data/node/{treeId}/{id}") public Map remove(@ApiParam(required = true, value = "分类树ID")@PathVariable Integer treeId, @ApiParam(required = true, value = "节点ID")@PathVariable Integer id){ return ActionHelper.responseOk(catagoryTreeService.remove(treeId,id)); }} 六、service层
JSONArray jsonArray = JSON.parseArray(jsonTree); //由json字符串变成json数组对象
jsonArray.toJSONString() //由json数组对象变成json字符串package cn.chinaunicom.srigz.goods.admin.service;import cn.chinaunicom.srigz.goods.admin.database.dao.CatagoryTreeRepository;import cn.chinaunicom.srigz.goods.admin.database.model.CatagoryTree;import cn.chinaunicom.srigz.goods.admin.utils.Recursive;import cn.chinaunicom.srigz.goods.common.exception.AlreadyExistException;import cn.chinaunicom.srigz.goods.common.exception.BadRequestException;import cn.chinaunicom.srigz.goods.common.exception.NotFoundException;import com.alibaba.fastjson.JSON;import com.alibaba.fastjson.JSONArray;import com.alibaba.fastjson.JSONObject;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.stereotype.Service;import org.springframework.transaction.annotation.Transactional;import java.util.*;@Servicepublic class CatagoryTreeService { @Autowired private CatagoryTreeRepository catagoryTreeRepository; /** * 查询单个分类树 * @param treeId * @return */ public JSON getOne(Integer treeId) { CatagoryTree catagoryTree = catagoryTreeRepository.findById(treeId.longValue()).get(); return JSON.parseArray(catagoryTree.getJsonTree()); } /** * 增加单个节点 * @param treeId * @param map * @return */ @Transactional public JSON addOne(Integer treeId, Map map) { CatagoryTree catagoryTree = catagoryTreeRepository.findById(treeId.longValue()).get(); String jsonTree = catagoryTree.getJsonTree(); JSONArray jsonArray = JSON.parseArray(jsonTree); Object parentId = map.get("parentId"); Integer sequ = (Integer) map.get("sequ"); if(sequ <= 0 ){ throw new BadRequestException("数组位置不正确"); } Map nodeMap = (Map) map.get("node"); JSONObject node = new JSONObject(nodeMap); List list = new ArrayList(); List keyList = Recursive.recursive(jsonArray,"id",list); for (Object key : keyList) { if (nodeMap.get("id").toString().equals(key.toString())) { throw new AlreadyExistException("节点ID为" + key); } } Recursive.addRecursive(jsonArray,node,sequ,"id",parentId); catagoryTree.setJsonTree(jsonArray.toJSONString()); catagoryTreeRepository.save(catagoryTree); return jsonArray; } /** * 修改单个节点 * @param treeId * @param id * @param map * @return */ @Transactional public JSON updateOne(Integer treeId, Integer id, Map map) { CatagoryTree catagoryTree = catagoryTreeRepository.findById(treeId.longValue()).get(); String jsonTree = catagoryTree.getJsonTree(); JSONArray jsonArray = JSON.parseArray(jsonTree); List list = new ArrayList(); List keyList = Recursive.recursive(jsonArray,"id",list); JSONObject node = new JSONObject(map); if (id.toString().equals(map.get("id").toString())) { for (Object key : keyList) { if (id.toString().equals(key.toString())) { Recursive.updateRecursive(jsonArray,"id",id,node); catagoryTree.setJsonTree(jsonArray.toJSONString()); catagoryTree.setModifyTime(new Date()); catagoryTreeRepository.save(catagoryTree); return jsonArray; } } }else { throw new BadRequestException("id is not allowed to be modified"); } throw new NotFoundException("节点ID为" + id); } /** * 删除节点以及子节点 * @param treeId * @param id * @return */ @Transactional public JSON remove(Integer treeId, Integer id) { CatagoryTree catagoryTree = catagoryTreeRepository.findById(treeId.longValue()).get(); String jsonTree = catagoryTree.getJsonTree(); JSONArray jsonArray = JSON.parseArray(jsonTree); List list = new ArrayList(); List keyList = Recursive.recursive(jsonArray,"id",list); for (Object key : keyList) { if (id.toString().equals(key.toString())) { Recursive.removeRecursive(jsonArray,"id",id); catagoryTree.setJsonTree(jsonArray.toJSONString()); catagoryTreeRepository.save(catagoryTree); return jsonArray; } } throw new NotFoundException("节点ID为" + id); }} 七、把增删查改的递归方法写成工具类
package cn.chinaunicom.srigz.goods.admin.utils;import com.alibaba.fastjson.JSONArray;import com.alibaba.fastjson.JSONObject;import java.util.List;public class Recursive { /** * 递归取出所有Key对应的value值 * @param jsonArray 需要查询的目标主体 * @param key 需要查询的字段名key * @param list 存储value值 * @return */ public static List recursive(JSONArray jsonArray, String key, List list) { for (Object obj : jsonArray) { JSONObject jsonObject = (JSONObject) obj; JSONArray children = jsonObject.getJSONArray("children"); if (children != null) { list.add(jsonObject.get(key)); recursive(children,key,list); }else { list.add(jsonObject.get(key)); } } return list; } /** * 增加节点 * @param jsonArray 需要更新的目标主体 * @param node 增加节点信息 * @param sequ 数组的位置 * @param key 目标主体中的字段名key * @param value 节点信息与key对应的value */ public static void addRecursive(JSONArray jsonArray, JSONObject node, Integer sequ, String key, Object value) { for (Object obj : jsonArray) { JSONObject jsonObject = (JSONObject) obj; JSONArray children = jsonObject.getJSONArray("children"); if (children != null) { if (value.toString().equals(jsonObject.get(key).toString())) { if(sequ > children.size()){ children.add(children.size(), node); }else{ children.add(sequ-1, node); } return; } addRecursive(children,node,sequ,key,value); } else { if (value.toString().equals(jsonObject.get(key).toString())) { JSONArray array = new JSONArray(); array.add(node); jsonObject.put("children", array); return; } } } } /** * 根据条件更新节点 * @param jsonArray 需要更新的目标主体 * @param key 目标主体中的字段名key * @param value 更新节点信息与key对应的value * @param node 更新节点信息 */ public static void updateRecursive(JSONArray jsonArray, String key, Object value, JSONObject node) { for (int i = 0; i < jsonArray.size() ; i++) { JSONObject jsonObject = jsonArray.getJSONObject(i); JSONArray children = jsonObject.getJSONArray("children"); if (value.toString().equals(jsonObject.get(key).toString())) { if(children != null){ node.put("children",children); } jsonArray.set(i,node); return; }else if(children != null){ updateRecursive(children,key,value,node); } } } /** * 删除节点 * @param jsonArray 需要更新的目标主体 * @param key 目标主体中的字段名key * @param value 节点信息与key对应的value */ public static void removeRecursive(JSONArray jsonArray,String key,Object value) { for (Object obj : jsonArray) { JSONObject jsonObject = (JSONObject) obj; if (value.toString().equals(jsonObject.get(key).toString())) { jsonArray.remove(jsonObject); return; } JSONArray children = jsonObject.getJSONArray("children"); if (children != null) { removeRecursive(children,key,value); } } }}