React/기본

React - useMemo와 useCallback

태기의삶 2020. 11. 15. 23:03

 

useMemo란?

함수형 컴포넌트 내부에서 발생하는 연산을 최적화시킬 수 있는 Hooks를 말한다.

예제를 통해 알아보자.

먼저 리스트에 숫자를 추가하면 추가된 숫자들의 평균을 보여주는 함수형 컴포넌트를 작성해보자.

Average.js

import React, { useState } from "react";

const getAverage = (numbers) => {
  console.log("평균값 계산 중...");
  if (numbers.length === 0) return 0;
  const sum = numbers.reduce((a, b) => a + b);
  return sum / numbers.length;
};

const Average = () => {
  const [list, setList] = useState([]);
  const [number, setNumber] = useState("");

  const onChange = (e) => {
    setNumber(e.target.value);
  };

  const onInsert = (e) => {
    const nextList = list.concat(parseInt(number));
    setList(nextList);
    setNumber("");
  };

  return (
    <div>
      <input value={number} onChange={onChange} />
      <button onClick={onInsert}>등록</button>
      <ul>
        {list.map((value, index) => {
          return <li key={index}>{value}</li>;
        })}
      </ul>
      <div>
        <b>평균값:</b>
        {getAverage(list)}
      </div>
    </div>
  );
};

export default Average;

 

그러고 나서, App에서 이 컴포넌트를 렌더링 시킨다.

App.js

import React from "react";
import Average from "./components/Average/Average";

const App = () => {
  return (
    <>
      <Average />
    </>
  );
};

export default App;

실행 화면

그런데 숫자를 등록할 때뿐만 아니라 input 창에 숫자를 입력할 때도, 콘솔 창에 계속 "평균값 계산 중..."이라는 문장이 출력되는 것을 볼 수 있다.

그 이유는 계속 input 창에 숫자를 입력할 때마다, 화면이 렌더링 되어 getAverage 함수가 호출되기 때문이다.

하지만, input 창의 내용이 바뀔 때마다 계속 평균값을 다시 계산하고 렌더링 된다면 엄청난 낭비가 될 것이다.

이럴 때 useMemo Hook을 사용하면 이러한 낭비되는 작업을 최적화할 수 있다.

작동 원리는 렌더링 하는 과정에서 특정 값이 바뀌었을 때만 연산을 실행하고, 원하는 값이 바뀌지 않았다면 이전에 연산했던 결과를 다시 사용하는 방식이다.

import React, { useState, useMemo } from "react";

const getAverage = (numbers) => {
  console.log("평균값 계산 중...");
  if (numbers.length === 0) return 0;
  const sum = numbers.reduce((a, b) => a + b);
  return sum / numbers.length;
};

const Average = () => {
  const [list, setList] = useState([]);
  const [number, setNumber] = useState("");

  const onChange = (e) => {
    setNumber(e.target.value);
  };

  const onInsert = (e) => {
    const nextList = list.concat(parseInt(number));
    setList(nextList);
    setNumber("");
  };

  const avg = useMemo(() => getAverage(list), [list]);

  return (
    <div>
      <input value={number} onChange={onChange} />
      <button onClick={onInsert}>등록</button>
      <ul>
        {list.map((value, index) => {
          return <li key={index}>{value}</li>;
        })}
      </ul>
      <div>
        <b>평균값:</b>
        {getAverage(list)}
      </div>
    </div>
  );
};

export default Average;

이렇게 useMemo를 사용해서 함수를 호출하면 list 배열의 내용이 변경될 때만 getAverage 함수가 호출되는 것을 확인할 수 있다.

실행 화면

useCallback이란?

useCallback은 useMemo와 상당히 비슷한 함수이다. 이 또한 주로 렌더링 성능을 최적화해야 하는 상황에서 사용한다.

대신 이 Hook은 이벤트 핸들러 함수를 필요할 때만 생성할 수 있다.

이전에 Average 컴포넌트에서 onChange와 onInsert라는 함수를 선언했었는데, 이렇게 선언된 함수들은 리 렌더링 될 때마다 이 함수들이 새로 생성된다.

대부분의 경우 이러한 방식에 문제가 없지만, 컴포넌트의 렌더링이 자주 발생하거나 렌더링해야 할 컴포넌트의 개수가 많아졌을 경우 이 부분을 최적화해 주는 것이 좋다.

예시를 통해서 확인해보자.

Average.js

import React, { useState, useMemo, useCallback } from "react";

const getAverage = (numbers) => {
  console.log("평균값 계산 중...");
  if (numbers.length === 0) return 0;
  const sum = numbers.reduce((a, b) => a + b);
  return sum / numbers.length;
};

const Average = () => {
  const [list, setList] = useState([]);
  const [number, setNumber] = useState("");

  // 컴포넌트가 처음 렌더링될 때만 함수 생성
  const onChange = useCallback((e) => {
    setNumber(e.target.value);
  }, []);

  // number 혹은 list가 바뀌었을 때만 함수 생성
  const onInsert = useCallback(
    (e) => {
      const nextList = list.concat(parseInt(number));
      setList(nextList);
      setNumber("");
    },
    [number, list]
  );

  const avg = useMemo(() => getAverage(list), [list]);

  return (
    <div>
      <input value={number} onChange={onChange} />
      <button onClick={onInsert}>등록</button>
      <ul>
        {list.map((value, index) => {
          return <li key={index}>{value}</li>;
        })}
      </ul>
      <div>
        <b>평균값:</b>
        {avg}
      </div>
    </div>
  );
};

export default Average;

작동 원리를 확인해보면, useCallback의 첫 번째 파라미터에는 생성하고 싶은 함수를 넣고, 두 번째 파라미터에는 배열을 넣으면 된다.

대신 이 배열 안에 어떤 값이 바뀌었을 때 새로 함수를 생성할 수 있게 명시해야 한다.

예제에서의 onChange 함수처럼 비어 있는 배열을 넣게 되면 컴포넌트가 렌더링 될 때마다 한 번만 함수가 생성되고, onInsert처럼 배열 안에 number와 list를 넣게 되면 input 창의 내용이 변경되거나 새로운 항목이 추가될 때마다 함수가 생성된다.

또, 함수 내부에서 상태 값에 의존해야 할 때는 그 값을 반드시 두 번째 파라미터 안에 포함시켜 주어야 한다.

예를 들어 onChange 함수의 경우 기존의 값을 조회하는 것이 아니라 바로 변경된 값을 설정만 하기 때문에 배열이 비어있어도 상관없지만, onInsert 함수의 경우 number와 list를 조회(의존)해서 nextList를 생성하기 때문에 배열 안에 number와 list를 꼭 넣어줘야 한다.

그렇다면 useMemo와 useCallback 어떻게 써야 할까?

useMemo 같은 경우에는 숫자나, 문자열, 객체처럼 일반 값을 재사용할 경우 사용하고, useCallback 같은 경우는 함수를 재사용할 경우 사용하면 된다.

이 부분이 아직 잘 구별이 안 가지만 아마 개발하다 보면 알아서 익히게 될 것이다.

[ 벨로퍼트 책을 기반으로 스스로 정리한 내용입니다! ]