vue实现 可展开 且 可多选table 组件封装

基于 网上代码进行优化, 实现 实际开发功能
效果如图:

在父组件可以拿到 当前点击所有行数组:就可以把二层设备的数组id,愉快的传给后端了!

当然也可以增加一些功能
比如: 点击行 当前行改变样式 ,localstorage 记住 当前状态 等功能

创建 TreeTable 组件:

<template>
  <el-table
    ref="multipleTable"
    :data="formatData"
    :row-style="showRow"
    v-bind="$attrs"
  >
    <!-- @header-click="chooseall" -->
    <el-table-column :render-header="renderHeader" width="50" align="center">
      <template slot-scope="scope">
        <el-checkbox
          v-model="scope.row.checks"
          @change="toselect(scope.row)"
        ></el-checkbox>
      </template>
    </el-table-column>
    <el-table-column v-if="columns.length === 0" width="150">
      <template slot-scope="scope">
        <span
          v-for="space in scope.row._level"
          :key="space"
          class="ms-tree-space"
        />
        <span
          v-if="iconShow(0, scope.row)"
          class="tree-ctrl"
          @click="toggleExpanded(scope.$index)"
        >
          <i v-if="!scope.row._expanded" class="el-icon-plus" />
          <i v-else class="el-icon-minus" />
        </span>
        {{ scope.$index }}
      </template>
    </el-table-column>
    <el-table-column
      v-for="(column, index) in columns"
      v-else
      :key="column.value"
      :label="column.text"
      :width="column.width"
    >
      <template slot-scope="scope">
        <!-- Todo -->
        <!-- eslint-disable-next-line vue/no-confusing-v-for-v-if -->
        <span
          v-for="space in scope.row._level"
          v-if="index === 0"
          :key="space"
          class="ms-tree-space"
        />
        <span
          v-if="iconShow(index, scope.row)"
          class="tree-ctrl"
          @click="toggleExpanded(scope.$index)"
        >
          <i v-if="!scope.row._expanded" class="el-icon-plus" />
          <i v-else class="el-icon-minus" />
        </span>
        {{ scope.row[column.value] }}
      </template>
    </el-table-column>
    <slot />
  </el-table>
</template>
<script>
import treeToArray from "./eval";
export default {
  name: "TreeTable",
  data() {
    return {
      chooseson: true, //全选
      key: true, //单个点击直到全部选中
      tableChcekedList: [],
    };
  },
  props: {
    /* eslint-disable */
    data: {
      type: [Array, Object],
      required: true,
    },
    columns: {
      type: Array,
      default: () => [],
    },
    evalFunc: Function,
    evalArgs: Array,
    expandAll: {
      type: Boolean,
      default: false,
    },
  },
  computed: {
    // 格式化数据源
    formatData: function () {
      let tmp;
      if (!Array.isArray(this.data)) {
        tmp = [this.data];
      } else {
        tmp = this.data;
      }
      const func = this.evalFunc || treeToArray;
      const args = this.evalArgs
        ? [tmp, this.expandAll].concat(this.evalArgs)
        : [tmp, this.expandAll];
      return func.apply(null, args);
    },
  },
  methods: {
    showRow: function (row) {
      const show = row.row.parent
        ? row.row.parent._expanded && row.row.parent._show
        : true;
      row.row._show = show;
      return show
        ? "animation:treeTableShow 0s;-webkit-animation:treeTableShow 0s;"
        : "display:none;";
    },
    // 切换下级是否展开
    toggleExpanded: function (trIndex) {
      const record = this.formatData[trIndex];
      record._expanded = !record._expanded;
    },
    // 图标显示
    iconShow(index, record) {
      return index === 0 && record.child && record.child.length > 0;
    },

    //设置表头全选
    renderHeader(h, data) {
      return h(
        "span",
        {
          attrs: {
            style: "display:flex;align-items: center; justify-content: center;",
          },
        },
        [
          h("input", {
            attrs: {
              id: "chooseall",
              type: "checkbox",
            },
          }),
        ]
      );
    },
    //功能函数:选中部分子集
    setchildtobeselect(arr, key) {
      arr.forEach((v, i) => {
        // v.checks = key;
        this.$set(v, "checks", key);
        // v._expanded = key;//选中后展开子项
        if (v.child) {
          this.setchildtobeselect(v.child, v.checks);
        }
      });
      arr.forEach((item, index) => {
        if (item.checks) {
          if (item._level === 2) {
            let flag = false;
            this.tableChcekedList.forEach((tableItem, tableIndex) => {
              if (item._level === 2) {
                if (item.checks) {
                  if (item.id === tableItem.id) {
                    flag = true;
                  }
                } else {
                  if (item.id === tableItem.id) {
                    flag = true;
                    this.tableChcekedList.splice(tableIndex, 1);
                  }
                }
              }
            });
            if (!flag) {
              this.tableChcekedList.push(item);
            }
          }
        } else {
          this.tableChcekedList.splice(index, 1);
        }
        this.$emit("getCheckedRow", this.tableChcekedList);
      });
    },
    //是否所有的都被选中
    isallchecked(arr) {
      arr.forEach((v, i) => {
        if (!v.checks) {
          this.key = false;
          this.$set(this, "key", false);
        }
        if (v.child) {
          this.isallchecked(v.child);
        }
      });
    },
    //设置父级为 未选中状态（父级的父级没改变-有bug）
    setparentfalse(arr, id, level) {
      arr.forEach((v, i) => {
        if (v._level == level - 1 && v.child) {
          v.child.forEach((val, ind) => {
            if (val.id == id) {
              v.checks = false;
              return false; //终止此次循环，减少循环次数
            }
          });
        }
        if (v.child) {
          this.setparentfalse(v.child, id, level);
        }
      });
    },
    //设置父级为 选中状态
    setparenttrue(arr, id, level) {
      arr.forEach((v, i) => {
        if (v._level == level - 1 && v.child) {
          let key = true;
          let sameidkey = false;
          v.child.forEach((val, ind) => {
            if (val.id == id) {
              //确保当前点击的在该父级内
              sameidkey = true;
            }
            if (!val.checks) {
              key = false;
            }
          });
          if (key && sameidkey) {
            // v.checks = true;
            this.$set(v, "checks", true);
          }
        }
        if (v.child) {
          this.setparentfalse(v.child, id, level);
        }
      });
    },
    //某个复选框被点击时
    toselect(row) {
      // row._expanded = row.checks;//选中后是否展开
      //1、若有子集先让子选中
      if (row.child) {
        this.setchildtobeselect(row.child, row.checks);
      }
      //2、然后判断是否全选中
      this.key = true; //重置为true，防止上次已经是false的状态
      this.isallchecked(this.formatData);
      //3、设置多选框的状态
      if (!row.checks) {
        this.setparentfalse(this.formatData, row.id, row._level); //设置父级选中的状态为false
        document.getElementById("chooseall").checked = false; //设置全选框的状态
      } else {
        this.setparenttrue(this.formatData, row.id, row._level); //设置父级选中的状态为true
      }
      if (this.key) {
        document.getElementById("chooseall").checked = true; //设置全选框的状态
      }
      if (
        this.tableChcekedList.length === 0 &&
        row._level === 2 &&
        row.checks
      ) {
        // console.log("数组长度为0的时候 触发");
        this.tableChcekedList.push(row);
      } else {
        var flag = false;
        this.tableChcekedList.forEach((item, index) => {
          if (row._level === 2) {
            if (row.checks) {
              flag = true;
              // this.tableChcekedList.push(row);
            } else {
              if (item.id === row.id) {
                this.tableChcekedList.splice(index, 1);
              }
            }
          }
        });
        if (flag) {
          // console.log("单选行触发");
          this.tableChcekedList.push(row);
        }
      }
      this.$emit("getCheckedRow", this.tableChcekedList);
      // console.log(this.tableChcekedList, "this.tableChcekedList");
    },
  },
  mounted() {
    this.$nextTick(() => {
      var that = this;
      const all = document.getElementById("chooseall");
      all.onchange = function (e) {
        if (all.checked == true) {
          that.setchildtobeselect(that.formatData, true);
        } else {
          that.setchildtobeselect(that.formatData, false);
        }
      };
    });
  },
};
</script>
<style rel="stylesheet/css">
@keyframes treeTableShow {
  from {
    opacity: 0;
  }
  to {
    opacity: 1;
  }
}
@-webkit-keyframes treeTableShow {
  from {
    opacity: 0;
  }
  to {
    opacity: 1;
  }
}
</style>
  
<style>
#chooseall:checked {
  background: #1673ff !important;
  border: solid 1px #1673ff !important;
}
#chooseall {
  background-color: #ffffff;
  border-radius: 50%;
  font-size: 0.8rem;
  margin: 0;
  padding: 0;
  position: relative;
  display: inline-block;
  vertical-align: top;
  cursor: default;
  -webkit-appearance: none;
  -webkit-user-select: none;
  user-select: none;
  -webkit-transition: background-color ease 0.6s;
  transition: background-color ease 0.6s;
  border: 1px solid #dcdfe6;
  border-radius: 2px;
  box-sizing: border-box;
  width: 14px;
  height: 14px;
  cursor: pointer;
}
#chooseall:checked::after {
  content: "";
  top: 2px;
  left: 2px;
  position: absolute;
  background: transparent;
  border: #fff solid 1px;
  border-top: none;
  border-right: none;
  height: 4px;
  width: 9px;
  -webkit-transform: rotate(-45deg);
  transform: rotate(-45deg);
}
.ms-tree-space {
  position: relative;
  top: 1px;
  display: inline-block;
  font-style: normal;
  font-weight: 400;
  line-height: 1;
  width: 18px;
  height: 14px;
}
.ms-tree-space::before {
  content: "";
}
.processContainer {
  width: 100%;
  height: 100%;
}
table td {
  line-height: 26px;
}
.tree-ctrl {
  position: relative;
  cursor: pointer;
  color: #2196f3;
  margin-left: -18px;
}
</style>

在 .vue 父组件调用:

<template>
  <div>
    <tree-table
      :data="data"
      :columns="columns"
      :expandAll="true"
      border
      @getCheckedRow="getCheckedRow"
    />
  </div>
</template>

<script>
import treeTable from "../../components/TreeTable";

export default {
  name: "courseProgress",
  components: {
    treeTable,
  },
  data() {
    return {
      rowArrList: [],
      columns: [
        {
          text: "事件",
          value: "event",
          width: 200,
        },
        {
          text: "ID",
          value: "id",
        },
      ],
      data: [
        {
          id: 0,
          event: "区域1",
          child: [
            {
              id: 1,
              event: "设备11",
            },
            {
              id: 2,
              event: "设备12",
            },
          ],
        },

        {
          id: 1,
          event: "区域2",
          child: [
            {
              id: 3,
              event: "设备21",
            },
            {
              id: 4,
              event: "设备22",
            },
          ],
        },
      ],
    };
  },
  methods: {
    getCheckedRow(arr) {
      this.rowArrList = arr;
      console.log(arr, "父组件拿到 点击行数组");
    },
  },
};
</script>

<style scoped>
</style>

创建 eval.js

import Vue from "vue";
export default function treeToArray(
  data,
  expandAll,
  parent = null,
  level = null
) {
  let tmp = [];
  Array.from(data).forEach(function(record) {
    if (record._expanded === undefined) {
      Vue.set(record, "_expanded", expandAll);
    }
    let _level = 1;
    if (level !== undefined && level !== null) {
      _level = level + 1;
    }
    Vue.set(record, "_level", _level);
    // 如果有父元素
    if (parent) {
      Vue.set(record, "parent", parent);
    }
    tmp.push(record);
    if (record.child && record.child.length > 0) {
      const child = treeToArray(record.child, expandAll, record, _level);
      tmp = tmp.concat(child);
    }
  });
  return tmp;
}