fix(table): resolve column width on multiple columns

Tencent · Dec 23, 2021 · 02898ac · 02898ac
1 parent a529df1
commit 02898ac
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/src/table/base-table/col-group.tsx b/src/table/base-table/col-group.tsx
@@ -10,8 +10,23 @@ export default defineComponent({
   methods: {
     renderColgroup(): Array<VNode> {
       const { columns } = this;
+      type ColumnType = typeof columns;
       const colgroup: Array<VNode> = [];
-      columns.forEach((column) => {
+      const flatArray = (arr: ColumnType) => {
+        const res: ColumnType = [];
+        arr.forEach((item: any) => {
+          if (item?.children?.length > 0) {
+            const val = flatArray(item.children);
+            res.push(...val);
+          } else {
+            res.push(item);
+          }
+        });
+        return res;
+      };
+      const flatCols = flatArray(this.columns);
+
+      flatCols.forEach((column) => {
         const { width, minWidth, colKey } = column;
         const style: any = {};
         if (width) {