做了几个组件

klp-ui/src/components/WarehouseSelect/index.vue

@@ -0,0 +1,95 @@
+<template>
+  <treeselect
+    v-model="innerValue"
+    :options="warehouseOptions"
+    :normalizer="normalizer"
+    :placeholder="placeholder"
+    :clearable="clearable"
+    @input="onInput"
+  />
+</template>
+
+<script>
+import Treeselect from '@riophae/vue-treeselect';
+import '@riophae/vue-treeselect/dist/vue-treeselect.css';
+import { listWarehouse } from '@/api/wms/warehouse';
+
+export default {
+  name: 'WarehouseSelect',
+  components: { Treeselect },
+  props: {
+    value: {
+      type: [Number, String, null],
+      default: null
+    },
+    placeholder: {
+      type: String,
+      default: '请选择库区/仓库/库位'
+    },
+    clearable: {
+      type: Boolean,
+      default: true
+    },
+    showTop: {
+      type: Boolean,
+      default: true // 是否显示顶级节点
+    }
+  },
+  data() {
+    return {
+      warehouseOptions: [],
+      innerValue: this.value
+    };
+  },
+  watch: {
+    value(val) {
+      this.innerValue = val;
+    }
+  },
+  mounted() {
+    this.loadOptions();
+  },
+  methods: {
+    loadOptions() {
+      listWarehouse().then(response => {
+        const options = [];
+        if (this.showTop) {
+          const top = { warehouseId: 0, warehouseName: '顶级节点', children: [] };
+          top.children = this.handleTree(response.data, 'warehouseId', 'parentId');
+          options.push(top);
+        } else {
+          options.push(...this.handleTree(response.data, 'warehouseId', 'parentId'));
+        }
+        this.warehouseOptions = options;
+      });
+    },
+    handleTree(data, id, parentId) {
+      const cloneData = JSON.parse(JSON.stringify(data));
+      return cloneData.filter(father => {
+        const branchArr = cloneData.filter(child => father[id] === child[parentId]);
+        if (branchArr.length > 0) father.children = branchArr;
+        return father[parentId] === 0 || father[parentId] === null;
+      });
+    },
+    normalizer(node) {
+      if (node.children && !node.children.length) {
+        delete node.children;
+      }
+      return {
+        id: node.warehouseId,
+        label: node.warehouseName,
+        children: node.children
+      };
+    },
+    onInput(val) {
+      // 选中顶级节点时，返回null
+      if (val === 0) {
+        this.$emit('input', null);
+        this.innerValue = null;
+      } else {
+        this.$emit('input', val);
+      }
+    }
+  }
+};
+</script>