[React] React Hooks - Custom Hooks

Hannahhh·2022년 9월 29일

React

목록 보기

26/30

🔍 Custom Hooks

개발자가 스스로 커스텀한 Hook을 의미하며, Custom Hook을 이용해 반복되는 로직을 함수로 뽑아내어 재사용할 수 있다.

주로, 여러 url을 fetch할 때, 여러 input에 의한 상태 변경 등 반복되는 로직을 동일한 함수에서 작동하게 하고 싶을 때 사용한다.

이때, Custom Hook을 이용한다면 아래의 장점이 있다.

상태관리 로직 재활용 가능
Class Component보다 적은 양의 코드로 동일한 로직 구현 가능
함수형으로 작성하기 때문에 명료함.

👀 사용 방법(예시)

//FriendStatus : 친구가 online인지 offline인지 return하는 컴포넌트
function FriendStatus(props) {
  const [isOnline, setIsOnline] = useState(null);
  useEffect(() => {
    function handleStatusChange(status) {
      setIsOnline(status.isOnline);
    }
    ChatAPI.subscribeToFriendStatus(props.friend.id, handleStatusChange);
    return () => {
      ChatAPI.unsubscribeFromFriendStatus(props.friend.id, handleStatusChange);
    };
  });

  if (isOnline === null) {
    return 'Loading...';
  }
  return isOnline ? 'Online' : 'Offline';
}

//FriendListItem : 친구가 online일 때 초록색으로 표시하는 컴포넌트
function FriendListItem(props) {
  const [isOnline, setIsOnline] = useState(null);
  useEffect(() => {
    function handleStatusChange(status) {
      setIsOnline(status.isOnline);
    }
    ChatAPI.subscribeToFriendStatus(props.friend.id, handleStatusChange);
    return () => {
      ChatAPI.unsubscribeFromFriendStatus(props.friend.id, handleStatusChange);
    };
  });

  return (
    <li style={{ color: isOnline ? 'green' : 'black' }}>
      {props.friend.name}
    </li>
  );
}

위에 예시에서 FriendStatus 컴포넌트는 사용자들이 온라인인지 오프라인인지 확인하고, FriendListItem 컴포넌트는 사용자들의 상태에 따라 온라인이라면 초록색으로 표시하는 컴포넌트이다.

이 두 컴포넌트는 정확하게 똑같이 쓰이는 로직이 존재하고 있는데,이 로직을 빼내서 두 컴포넌트에서 공유하기 위해 Custom Hook을 사용해보자.

1. Custom Hook 추출/정의

먼저, 동일하게 사용되고 있는 로직을 분리한다.

이때, Custom Hook을 정의하는 데 아래와 같은 일종의 규칙이 존재한다.

Custom Hook은 use로 시작하는 JavaScript 함수로, 다른 Hook(ex.useState)을 호출할 수 있다.(일반 함수에서는 불가능)
프로젝트 내의 Hooks 디렉토리에 Custom Hook을 위치시킨다.
함수는 조건부 함수가 아니어야 한다.
즉, return 값이 조건부가 아니어야 한다.

예시에서는 Custom Hook의 이름을 useFriendStatus 로 설정했다.

function useFriendStatus(friendID) {// useFriendStatus
  const [isOnline, setIsOnline] = useState(null);

  useEffect(() => {
    function handleStatusChange(status) {
      setIsOnline(status.isOnline);
    }

    ChatAPI.subscribeToFriendStatus(friendID, handleStatusChange);
    return () => {
      ChatAPI.unsubscribeFromFriendStatus(friendID, handleStatusChange);
    };
  });

  return isOnline;
}

2. Custom Hook 이용

위에서 추출한 Custom Hook을 적용하면, 아래와 같이 작성할 수 있다.

function FriendStatus(props) {
  const isOnline = useFriendStatus(props.friend.id);

  if (isOnline === null) {
    return 'Loading...';
  }
  return isOnline ? 'Online' : 'Offline';
}

function FriendListItem(props) {
  const isOnline = useFriendStatus(props.friend.id);

  return (
    <li style={{ color: isOnline ? 'green' : 'black' }}>
      {props.friend.name}
    </li>
  );
}

기존의 코드와 비교했을 때, 두 컴포넌트를 더 직관적으로 확인할 수 있다.

단, 같은 Custom Hook을 사용했다고 하더라도 두 컴포넌트는 같은 로직만 공유할 뿐, state는 컴포넌트 내에서 독립적으로 정의되어 있으므로 같은 state를 공유하는 것은 아니다.

✔ 예시2

const useFetch = ( initialUrl:string ) => {
	const [url, setUrl] = useState(initialUrl);
	const [value, setValue] = useState('');

	const fetchData = () => axios.get(url).then(({data}) => setValue(data));	

	useEffect(() => {
		fetchData();
	},[url]);

	return [value];
};

export default useFetch;

[코드] 여러 url을 fetch할 때 쓸 수 있는 useFetch Hook

import { useState, useCallback } from 'react';

function useInputs(initialForm) {
  const [form, setForm] = useState(initialForm);
  // change
  const onChange = useCallback(e => {
    const { name, value } = e.target;
    setForm(form => ({ ...form, [name]: value }));
  }, []);
  const reset = useCallback(() => setForm(initialForm), [initialForm]);
  return [form, onChange, reset];
}

export default useInputs;

[코드] 여러 input에 의한 상태 변경을 할 때 쓸 수 있는 useInputs Hooks

🔥 실습 예시

✔ custom hook을 이용하여 useEffect 로직 분리하기

// App.js

import useFetch from "./util/useFetch";
import "./styles.css";

//useFetch, useInput, useForm : custom hook
export default function App() {
  const fetchData = useFetch("data.json");

  return (
    <div className="App">
      <h1>To do List</h1>
      <div className="todo-list">
        {fetchData &&
          fetchData.todo.map((el) => {
            return <li key={el.id}>{el.todo}</li>;
          })}
      </div>
    </div>
  );
}

// .util/useFetch.js

import { useEffect, useState } from "react";

const useFetch = (fetchUrl) => {
  const [data, setData] = useState();

  useEffect(() => {
    fetch(fetchUrl, {
      headers: {
        "Content-Type": "application/json",
        Accept: "application/json"
      }
    })
      .then((response) => {
        return response.json();
      })
      .then((myJson) => {
        setData(myJson);
      })
      .catch((error) => {
        console.log(error);
      });
  }, [fetchUrl]);

  //loading, error
  return data;
};

export default useFetch;

[그림]파일 list

[그림]결과 화면

✔ custom hook을 이용하여 useEffect 로직 분리하기

// App.js

import { useState } from "react";
import useInput from "./util/useInput";
import Input from "./component/Input";
import "./styles.css";

export default function App() {
  
  const [fistValue, firstBind, firstReset] = useInput("");
  const [lastValue, lastBind, lastReset] = useInput("");
  const [nameArr, setNameArr] = useState([]);

  const handleSubmit = (e) => {
    e.preventDefault();
    setNameArr([...nameArr, `${fistValue} ${lastValue}`]);
    firstReset();
    lastReset();
  };

  return (
    <div className="App">
      <h1>Name List</h1>
      <div className="name-form">
        <form onSubmit={handleSubmit}>
          <Input labelText={"성"} value={firstBind} />
          <Input labelText={"이름"} value={lastBind} />
          <button>제출</button>
        </form>
      </div>
      <div className="name-list-wrap">
        <div className="name-list">
          {nameArr.map((el, idx) => {
            return <p key={idx}>{el}</p>;
          })}
        </div>
      </div>
    </div>
  );
}

// .component/Input.js

function Input({ labelText, value }) {
  return (
    <div className="name-input">
      <label>{labelText}</label>
      <input {...value} type="text" />
    </div>
  );
}

export default Input;

// .util/useInput.js

import { useState } from "react";

function useInput(initialValue) {
  const [value, setValue] = useState(initialValue);

  const bind = {
    value,
    onChange: (e) => {
      setValue(e.target.value);
    }
  };

  const reset = () => {
    setValue(initialValue);
  };
  
//return 해야 하는 값은 배열 형태의 값
  return [value, bind, reset];
}

export default useInput;