Search code examples
reactjsantdreact-class-based-component

How to convern ANTD class-based component to react function-based component


Can anyone give a suggestion on how can I convert Ant Design class-based component into function-based? I am new to class-based, so it is pretty confusing for me to convert class components. Now, my application is a functional-based component. Any suggestion will be appreciated!

Transfer Component

import { Transfer, Button } from 'antd';

class App extends React.Component {
  state = {
    mockData: [],
    targetKeys: [],
  };

  componentDidMount() {
    this.getMock();
  }

  getMock = () => {
    const targetKeys = [];
    const mockData = [];
    for (let i = 0; i < 20; i++) {
      const data = {
        key: i.toString(),
        title: `content${i + 1}`,
        description: `description of content${i + 1}`,
        chosen: Math.random() * 2 > 1,
      };
      if (data.chosen) {
        targetKeys.push(data.key);
      }
      mockData.push(data);
    }
    this.setState({ mockData, targetKeys });
  };

  handleChange = targetKeys => {
    this.setState({ targetKeys });
  };

  renderFooter = (props, { direction }) => {
    if (direction === 'left') {
      return (
        <Button size="small" style={{ float: 'left', margin: 5 }} onClick={this.getMock}>
          Left button reload
        </Button>
      );
    }
    return (
      <Button size="small" style={{ float: 'right', margin: 5 }} onClick={this.getMock}>
        Right button reload
      </Button>
    );
  };

  render() {
    return (
      <Transfer
        dataSource={this.state.mockData}
        showSearch
        listStyle={{
          width: 250,
          height: 300,
        }}
        operations={['to right', 'to left']}
        targetKeys={this.state.targetKeys}
        onChange={this.handleChange}
        render={item => `${item.title}-${item.description}`}
        footer={this.renderFooter}
      />
    );
  }
}

ReactDOM.render(<App />, mountNode);

Solution

  • in function component to define state you can use useState hook, and also instead of lifecycles such as componentDidMount, componentDidUpdate, ... you can use useEffect hook, liek this:

    import {useState, useEffect} from 'react';
    import { Transfer, Button } from 'antd';
    
    function App() {
      const [state, setState] = useState({
        mockData: [],
        targetKeys: [],
      });
    
      const getMock = () => {
        const targetKeys = [];
        const mockData = [];
        for (let i = 0; i < 20; i++) {
          const data = {
            key: i.toString(),
            title: `content${i + 1}`,
            description: `description of content${i + 1}`,
            chosen: Math.random() * 2 > 1,
          };
          if (data.chosen) {
            targetKeys.push(data.key);
          }
          mockData.push(data);
        }
        setState({ mockData, targetKeys });
      };
    
      useEffect(()=>{
        getMock();
      }, [])
    
      const handleChange = targetKeys => {
        setState(prevState => ({ ...prevState, targetKeys }));
      };
    
      const renderFooter = (props, { direction }) => {
        if (direction === 'left') {
          return (
            <Button size="small" style={{ float: 'left', margin: 5 }} onClick={getMock}>
              Left button reload
            </Button>
          );
        }
        return (
          <Button size="small" style={{ float: 'right', margin: 5 }} onClick={getMock}>
            Right button reload
          </Button>
        );
      };
    
      return (
        <Transfer
          dataSource={state.mockData}
          showSearch
          listStyle={{
            width: 250,
            height: 300,
          }}
          operations={['to right', 'to left']}
          targetKeys={state.targetKeys}
          onChange={handleChange}
          render={item => `${item.title}-${item.description}`}
          footer={renderFooter}
        />
      );
    }