react 表格扩展与编辑

项目里有个需求是点击表格某行的工料机，显示对应定额下的工料机的表格数据，并能对两个表格进行增删改查，效果如下：

代码如下：

// 引入 Component 组件
import React, { Component } from 'react';

//定义子表显示
const EditableContext = React.createContext();
const EditableRow = ({ form, index, ...props }) => (
  <EditableContext.Provider value={form}>
    <tr {...props} />
  </EditableContext.Provider>
);

const EditableFormRow = Form.create()(EditableRow);
class EditableCell extends React.Component {
  getInput = () => {
    // 这里也可以通过其他值判断显示下拉框（比如dataIndex）
    if (this.props.inputType === 'number') {
      return <InputNumber />;
    }
    return <Input />;
  };

  render() {
    const {
      editing,
      dataIndex,
      title,
      inputType,
      record,
      index,
      ...restProps
    } = this.props;
    return (
      <EditableContext.Consumer>
        {(form) => {
          const { getFieldDecorator } = form;
          return (
            <td {...restProps}>
              {editing ? (
                <FormItem style={{ margin: 0 }}>
                  {getFieldDecorator(dataIndex, {
                    rules: [{
                      required: true,
                      message: `请填写${title}!`,
                    }],
                    initialValue: record[dataIndex],
                  })(this.getInput())}
                </FormItem>
              ) : restProps.children}
            </td>
          );
        }}
      </EditableContext.Consumer>
    );
  }
} 

// 将上面的配置应用到子表中
const components = {
      body: {
        row: EditableFormRow,
        cell: EditableCell,
      },
    };

// 原始colums要把想要编辑的editable设置为true
const columns = OriginalColumns.map((col) => {
      if (!col.editable) {
        return col;
      }
      return {
        ...col,
        onCell: record => ({
          record,
          inputType: col.dataIndex === 'age' ? 'number' : 'text',
          dataIndex: col.dataIndex,
          title: col.title,
          editing: this.isEditing(record),   // 判断编辑哪一行
        }),
      };
    });

     // 判断编辑哪一行的函数
     isEditing = record => record.id === this.state.editingKey;

    <Table 
      rowClassName="tableInput" 
      dataSource={tableData} 
      columns={columns} 
      pagination={false} 
      rowKey={record=>record.id} 
      getTableData={this.getTableData} 
      components={components}
      rowSelection={rowSelection}
      expandedRowRender={this.expandedRowRender}    // 字列表扩展函数
      expandIconColumnIndex={-1}                    // 隐藏默认展开样式 
      expandIconAsCell={false}                      // 隐藏默认展开样式  
      expandedRowKeys={expandedRowKeys}             // 展开的行的id
      defaultExpandedRowKeys={expandedRowKeys}
    />
    
  // 点击工料机，获取对应数据并展开，收起上一个展开的
  expandTable=(record,index)=>{
    this.setState({childTabData:[],parentTabId:record.id})
    // 如果已经展开则收回，否则展开
    let expandedRows = this.state.expandedRowKeys; 
    if(expandedRows.includes(record.id)){
      this.setState({expandedRowKeys: []})
    }else{
      // 获取数据并展开当前行
      let arr = []
      arr.push(record.id)
      this.getStuffListById(record.id)
      this.setState({expandedRowKeys:arr}) 
    }
  } 

　　

// 子列表
expandedRowRender=()=>{
// 子列表也要编辑（字列表不用编辑可以不用，省略了一些子列表编辑代码）
const childComponents = {
body: {
 row: EdiChildTableFormRow, 
 cell: EdiChildTableCell,
},
};


    const  childColumns = childOriColumns.map((col) => {
      if (!col.editable) {
        return col;
      }
      return {
        ...col,
        onCell: record => ({
          record,
          inputType: col.dataIndex === 'age' ? 'number' : 'text',
          dataIndex: col.dataIndex,
          title: col.title,
          editing: this.isEditing(record),
        }),
      };
    });



     const { countKey, childTabData ,editingKey} = this.state;



    return (
      <div className="childTab">
        <Table
          rowClassName="tableInput" 
          columns={childColumns}
          dataSource={this.state.childTabData}
          rowKey={record=>record.id} 
          components={childComponents}
          rowSelection={childRowSelection}
        />
        <div className="childBtnBox">
          <Button className="mr20" onClick={addGlj} disabled={editingKey && editingKey.includes('new')? true:false}>新增</Button>
          <Button disabled={this.state.onDeleteChildBtn} onClick={delChild}>删除</Button>
        </div>
      </div> 
    );



  }