react+antd 遞歸實(shí)現(xiàn)樹狀目錄操作

更新時(shí)間：2020年11月02日 09:50:36 作者：宋三步

這篇文章主要介紹了react+antd 遞歸實(shí)現(xiàn)樹狀目錄操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧

1.寫在前面

作為前端小白的我一直對(duì)算法和數(shù)據(jù)結(jié)構(gòu)淺嘗輒止，噥，吃虧了。使用多次遞歸實(shí)現(xiàn)數(shù)據(jù)格式化后將數(shù)據(jù)進(jìn)行樹狀展示的目的，分享一下我這次撓頭的經(jīng)歷~

2.數(shù)據(jù)

后臺(tái)傳過來的數(shù)據(jù)大概是這樣的

{
 "data":[
 {
 "id":1,
 "name":"一級(jí)節(jié)點(diǎn)",
 "parentId":0,
 "isValid":true,
 "canAddChild":true,
 "parent":null,
 "children":[]
 },{
 "id":3,
 "name":"二級(jí)節(jié)點(diǎn)",
 "parentId":1,
 "isValid":true,
 "canAddChild":true,
 "parent":null,
 "children":[]
 }
 ],
 "status":1
}

3.數(shù)據(jù)格式

data里面每個(gè)元素的parentId指向是父級(jí)元素的id，parentId為0的是結(jié)構(gòu)樹的頂級(jí)元素，但現(xiàn)在是個(gè)平面的數(shù)組不好處理，而我們要做的是樹狀的結(jié)構(gòu)，所以首先要對(duì)數(shù)據(jù)進(jìn)行格式化，將一個(gè)元素的所有子元素放到該元素的children屬性中去。那么，遞歸就來了。

 createTree = data => {
 let treeArr = [];
 //獲取頂級(jí)父元素集合
 let roots = data.filter(elemt => elemt.parentId === 0);
 treeArr.push(...roots);
 //從頂級(jí)元素開始，獲取每個(gè)元素的子元素放到該元素的children屬性中
 const getChildren = (resultarr,data) => {
  resultarr.forEach((elemt,index) => {
   elemt.children = data.filter((item,index) => item.parentId === elemt.id);
   //判斷當(dāng)前元素是不是有子元素被添加，如果有，再在子元素這一層循環(huán)
   if(elemt.children.length > 0){
    getChildren(elemt.children,data);
   }
  });
 }
 getChildren(treeArr,data);
 //最后更新一下數(shù)據(jù)
 this.setState({
  treeArr
 })

4.組件格式

因?yàn)閁I組件用的是antd，使用Tree和TreeNode做樹結(jié)構(gòu)。因?yàn)閿?shù)據(jù)已經(jīng)是樹狀的了，而且深度我們不確定，想要按層級(jí)輸出UI控件，那么，遞歸又來了。

renderTree = jsonTree => jsonTree.map( value => {
 //遍歷樹狀數(shù)組，如果發(fā)現(xiàn)他有children則先套上<TreeNode>,再對(duì)他children中的元素做相同的操縱，直到children為空的元素停止，說明他們已經(jīng)是最深的那一層了。 
  if(value.children){
   return(
    <TreeNode title={
     <span>
     {value.name}
     <Icon type="plus" onClick={this.showModal.bind(this,2,value.id)} />
     <Icon type="edit" onClick={this.showModal.bind(this,1,value.id)} />
     <Icon type="delete" onClick={this.showModal.bind(this,0,value.id)} />
     </span>
    } key={value.id}>
     //對(duì)children中的每個(gè)元素進(jìn)行遞歸
     {this.renderTree(value.children)} 
    </TreeNode>  
   )
  }  
 })

至此，就基本完成對(duì)數(shù)據(jù)的格式和對(duì)UI樹的格式啦，最后在樹控件中調(diào)用它，OK~

5.瘋狂輸出

 render(){
 return(
 <Tree showLine={true}>
 {this.renderTree(this.state.treeArr)}
 </Tree>
 )
 }

運(yùn)行，Bingo~

tips

因?yàn)槟夸洏涞拿恳豁?xiàng)都是可編輯的，而原始的UI組件也沒有可用配置，后來查閱文檔竟然應(yīng)該在TreeNode的title樹形中添加樹的自定義元素，可以，很強(qiáng)勢(shì)，官方文檔，看就完了，哈哈。

補(bǔ)充知識(shí)：antd的tree樹形組件異步加載數(shù)據(jù)小案例

前不久，做業(yè)務(wù)需求時(shí)遇到一個(gè)樹形選擇的問題，子節(jié)點(diǎn)的數(shù)據(jù)要通過點(diǎn)擊展開才去加載，在antd給出的官方文檔中也有關(guān)于動(dòng)態(tài)加載數(shù)據(jù)的demo，然而不夠詳細(xì)，自己研究之后，整理出來共享給大家借鑒下。

view.jsx（純函數(shù)式聲明）

// 渲染樹的方法

const loop = data => data.map((item) => {
  const index = item.regionName.indexOf(modelObj.searchValue);
  const beforeStr = item.regionName.substr(0, index);
  const afterStr = item.regionName.substr(index + modelObj.searchValue.length);
  const title = index > -1 ? (
  <span>
   {beforeStr}
   <span style={{ color: '#f50' }}>{modelObj.searchValue}</span>
   {afterStr}
  </span>
  ) : <span>{item.regionName}</span>;
  if (item.children && item.children.length > 0) {
  return (
   <TreeNode key={item.regionCode} parentId={item.parentId} title={getTreeTitle(title, item)}>
   {loop(item.children)}
   </TreeNode>
  );
  }
  return <TreeNode key={item.regionCode} parentId={item.parentId} title={getTreeTitle(title, item)} isGetDealer={item.isGetDealer} isLeaf={item.isLeaf}/>; 
  });

//樹結(jié)構(gòu)展開觸發(fā)的方法

const onExpand = (expandedKeys) => {
  console.log('onExpand-->', expandedKeys);
  
  dispatch({
  type: `${namespace}/onExpand`,
  payload: {
   expandedKeys,
   autoExpandParent: false,
  }
  });
 }

//異步加載樹形結(jié)構(gòu)子節(jié)點(diǎn)的方法

 const onLoadData = (treeNode) => {
  return new Promise((resolve) => {
  resolve();
  if(treeNode.props.isGetDealer===1){
  let cityIds =treeNode.props.eventKey;
  
  dispatch({
   type: `${namespace}/queryDealers`,
   payload: {
   checkedKeys:cityIds,
   }
  });
  }
 })
 }

//節(jié)點(diǎn)被選中時(shí)觸發(fā)的方法

 const onCheck = (checkedKeys,e) => {
 console.log('checkedKeys',checkedKeys);
 
 if(checkedKeys.length > 0) {
  updateModel(checkedKeys,'checkedKeys');
 } else {
  updateModel([], 'sellers');
  updateModel([], 'checkedKeys');
 }
 }

//dom渲染

return (
 {/* 經(jīng)銷商省市選擇 */}
     {
     modelObj.designatSeller === 1 && <div style={{marginBottom:20}}>
      <div className = {styles.treeStyle} style={{ backgroundColor: '#FBFBFB', width: 343, paddingLeft: 8, height: 200, overflow: 'auto' }}>
      <Tree
       checkable
       onExpand={onExpand}
       expandedKeys={modelObj.expandedKeys}
       checkedKeys={modelObj.checkedKeys}
       //checkStrictly={true}
       autoExpandParent={modelObj.autoExpandParent}
       onCheck={onCheck}
       loadData={onLoadData}
      >
       {loop(modelObj.countryList)}
      </Tree>
      </div>
     </div>
     }
)

mod.js

//初始默認(rèn)狀態(tài)

const defaultState = {
 countryList:[], //省市樹
 expandedKeys: [], //樹數(shù)據(jù)
 autoExpandParent: true, 
 checkedKeys:[],//當(dāng)前選中的節(jié)點(diǎn)
}

// 方法列表

 effects: {
  //根據(jù)城市獲取經(jīng)銷商
 *queryDealers({payload}, { call, put, select }) {
  let {countryList} = yield select(e => e[tmpModule.namespace]);

  let {data, resultCode } = yield call(queryDealers, {cityCodes:payload.checkedKeys});

  if(resultCode === 0 && data.length > 0) {

  let sellers = data.map(x=>x.dealerId);

  yield put({
   type: 'store',
   payload: {
   sellers
   }
  });

  let gdata = groupBy(data, 'cityCode');

  setgData(countryList);

  yield put({
   type: 'store',
   payload: {countryList } 
  });

  function setgData(arr) {
   if(arr&&arr.length>0){
   arr.forEach(x=>{
    if(x.regionCode.split(',')[1] in gdata) {
    x.children = gdata[x.regionCode.split(',')[1]].map(x=>{
     return {
     children:[],
     regionName:x.dealerName,
     regionCode:x.provinceCode+','+x.cityCode+','+x.dealerId,
     regionId:x.dealerId,
     isLeaf:true
     };
    })
    } else {
    setgData(x.children);
    }
   })
   }
   
  }

  }
 },

//獲取省市樹

 *queryCountry({ }, { call, put, select }) {
  let { countryList,biz} = yield select(e => e[tmpModule.namespace]);
  let resp = yield call(queryCountry);
  // console.log('resp_country-->',resp)
  if(resp.resultCode === 0){
   let dataList = cloneDeep(countryList);
   dataList = [{
   children: resp.data, 
   regionName:'全國',
   regionId:'100000', 
   regionCode:'100000'}];
   dataList.map((one,first)=> {
   one.children.map((two,second)=> {
    two.children.map((three,third)=> {
    three.children.map((four,fouth)=>{
     four.regionCode = three.regionCode+','+four.regionCode;
     //是否為最后的子節(jié)點(diǎn)去獲取經(jīng)銷商
     four.isGetDealer=1;
     four.children = []; 
    })
    })
   })
   })
   yield put({
   type: 'store',
   payload: {countryList: dataList } 
   });
  }
 },
 //展開節(jié)點(diǎn)觸發(fā)
  *onExpand({ payload }, { call, put, select }){
  yield put({
   type: 'store',
   payload: {
   expandedKeys:payload.expandedKeys,
   autoExpandParent:payload.autoExpandParent
   }
  });
  },
 }

回顯時(shí)后端需要返回的數(shù)據(jù)格式如下：