0

In my React hooks I defined two functions for setting variables: setProjectMiddleCode and setProjectToolCode.

I hope to call this two method in my react hooks to avoid duplicate code.

I would like to do it like this:

//variable define
let data;
let index = res.data.indexOf(res.code.value);

//call dynamic

if(some state ==='A'){
 data= "setProjectMiddleCode"
}else{
 data = "setProjectToolCode"
}

if (index < 0) {
  this[data](res.data.concat(res.code.value));
} else {
  this[data](res.data.filter((_, i) => i !== index));
}

My current code:

  const [projectMiddleCode, setProjectMiddleCode] = useState([]);
  const [projectToolCode, setProjectToolCode] = useState([]);

const ProjectWrite = memo(({}) => {
  let component;
  const dispatch = useDispatch();

  const [projectMiddleCode, setProjectMiddleCode] = useState([]);
  const [projectToolCode, setProjectToolCode] = useState([]);

  const callbackFromChild = useCallback(
    res => () => {
        let index = res.data.indexOf(res.code.value);

      if (res.codeName === 'PROJECT_MIDDLE_CODE') {
        if (index < 0) {
          setProjectMiddleCode(res.data.concat(res.code.value));
        } else {
          setProjectMiddleCode(res.data.filter((_, i) => i !== index));
        }
      } else if (res.codeName === 'TOOL_LIST') {
        if (index < 0) {
          setProjectToolCode(res.data.concat(res.code.value));
        } else {
          setProjectToolCode(res.data.filter((_, i) => i !== index));
        }
      }
    },
    []
  );
Laurenz Albe
  • 209,280
  • 17
  • 206
  • 263
dondon
  • 3
  • 2

1 Answers1

0

One way to do this is to create a map of res.codeName to your functions:

const { codeName, code, data } = res;
const index = data.indexOf(code.value);

const funcMap = {
  PROJECT_MIDDLE_CODE: setProjectMiddleCode,
  TOOL_LIST: setProjectToolCode
}


const newData = index < 0 ? data.concat(code.value) : data.filter((_, i) => i !== index);

const func = funcMap[codeName];

func(newData);
helloitsjoe
  • 6,264
  • 3
  • 19
  • 32