antd為Tree組件標(biāo)題附加操作按鈕功能

更新時間：2022年08月16日 11:45:59 作者：曲鳥???????

這篇文章主要介紹了antd為Tree組件標(biāo)題附加操作按鈕功能，文章圍繞主題展開詳細(xì)的內(nèi)容介紹，具有一定的參考價值，需要的小伙伴可以參考一下

一、前言

使用antd的tree組件實現(xiàn)下面這樣的模塊樹，點擊標(biāo)題請求其下列表的數(shù)據(jù)，點擊標(biāo)題旁邊的操作圖標(biāo)則執(zhí)行對應(yīng)的增刪改功能：

二、實現(xiàn)方案

1.封裝一個設(shè)置樹標(biāo)題的方法，通過開關(guān)改變state來控制圖標(biāo)按鈕是否可見：

處理樹數(shù)據(jù)（name、children）

  const setTree = (module_data: any) => {
    return module_data.map((item: any) => {
      let _json = { ...item };
      _json.name = setTreeTitle(item);
      _json.children = item.children ? setTree(item.children) : [];
      return _json;
    });
  };

賦值給Tree的treeData：

<Tree
   		...
        treeData={setTree(treeData)} 
        ...
        fieldNames={{ title: 'name', key: 'id', children: 'children' }}
      />

2.樹標(biāo)題被選中時，則調(diào)用刷新列表的方法:

onSelect={(value: any, e: any) => {
      if (e.selected) {
        treeSelectFunc(value[0]);
      }
    }}

3.但需要注意的坑是，由于我增加了刪除功能，當(dāng)執(zhí)行刪除操作后，樹處于選中狀態(tài)的話就會報錯，因為處于選中狀態(tài)就會調(diào)用刷新列表的方法，但該數(shù)據(jù)已經(jīng)被我刪除了，可能因此報錯：

所以我們需要創(chuàng)建一個state來存儲當(dāng)前選中的數(shù)據(jù)：

const [selectId, setSelectId] = useState<any>(null);

然后在onSelect方法中判斷新選中的數(shù)據(jù)是否是當(dāng)前數(shù)據(jù)，如果是則不再執(zhí)行請求方法：

onSelect={(value: any, e: any) => {
  const selectValue = value[0]
  if (e.selected && selectValue !== selectId) {
    treeSelectFunc(selectValue);
    setSelectId(selectValue)
  }
}}

這樣就解決了刪除數(shù)據(jù)可能報錯的問題。