[Firebase] Functions 로 사용자 삭제 시 데이터 삭제 구현

 

이번에는 Firebase Functions 를 이용해서 사용자를 삭제하면 그 사용자와 관련된 데이터도 삭제하도록 해보겠습니다.

firestore, storage 를 사용하기 때문에 이 둘을 타겟으로 하여 사용자 삭제 시 트리거될 functions 를 작성하면 됩니다.

 

* 모든 내용은 functions/index.js 에서 작성했습니다.

* cloud functions 설정은 이전 포스팅에 있습니다.

2022.01.20 - [PROGRAMMING/기타] - [Firebase] Firebase Functions CLI 사용하기, 함수 배포, 리전 변경

[Firebase] Firebase Functions CLI 사용하기, 함수 배포, 리전 변경

 

 

1) firebase 설정

functions 와 firebase 를 다음과 같이 설정해줍니다.

const functions = require("firebase-functions");

// The Firebase Admin SDK to access Firestore.
const admin = require("firebase-admin");
admin.initializeApp({
  storageBucket: "APP_ID.appspot.com",
});

저는 initializeApp 에 storageBucket 으로 bucket 을 지정해주었습니다.

(이는 이후에 지정할 수도 있습니다. 저는 storage 에서 1개의 bucket 만 사용해서 이렇게 했습니다.)

 

 

2) user 삭제 시 실행할 functions 작성

exports.clearData = functions
    .region("asia-northeast2")
    .auth.user().onDelete((event) => {
      const uid = event.uid;
      const promises = clearFirebaseData(uid);

      return Promise.all(promises)
          .then(() =>
            console.log(`Successfully removed data for user #${uid}.`));
    });

user 가 delete 될 때 실행하도록 하였고, 발생한 event 에서 uid 를 가져와 실제로 삭제를 해주는 함수 clearFirebaseData 에 넘겨주도록 했습니다.

 

clearFirebaseDate 는 다음과 같이, 프로젝트 구조에 맞게 삭제 로직을 작성해주면 됩니다.

저의 경우 COLLECTION/uid 의 경로에 해당 사용자의 데이터를 저장하고 있어서, 다음과 같이 작성했습니다.

const clearFirebaseData = (uid) => {
  const promises = [];
  const firestore = admin.firestore();

  // delete firestore data
  promises.push(firestore.collection("COLLECTION").doc(uid)
      .delete()).catch((err) => {
        console.error("Error deleting document:", err);
      }));

  // delete storage data
  const bucket = admin.storage().bucket();
  promises.push(
      bucket.deleteFiles({prefix: `FOLDER/${uid}/`})
          .catch((err) => {
            console.error("Error deleting group image:", err);
          }));
  return promises;
};

주의할 것은, document 내에 있는 collection 은 한번에 삭제가 안되다는 것입니다.

firebase collection 삭제하는 데에는 다음의 2가지 방법이 있습니다.

// 1. foreach
firebase.firestore().collection(path)
.listDocuments().then((val) => {
  val.map((doc) => doc.delete());
}).catch((err) => {
  console.error("Error deleting document:", err);
});

// 2. using batch
const batch = firebase.firestore().batch()
firebase.firestore().collection(path)
.listDocuments().then((val) => {
  val.map((doc) => batch.delete(val));
  batch.commit();
}).catch((err) => {
  console.error("Error deleting document:", err);
});

둘 중에서 프로젝트에 맞는 방법을 사용하면 됩니다.

 

참고로, firebase batch 는 일괄 쓰기 작업을 수행하는 것으로, 최대 500개까지 포함될 수 있으니 주의해야 합니다.

(https://firebase.google.com/docs/firestore/manage-data/transactions#batched-writes)

 

 

3) 배포

이제 작성한 functions 를 배포합니다. 다음과 같이 특정 함수만 배포할 수 있습니다.

$ firebase deploy --only functions:clearData

 

배포가 완료되면 실제로 console 에서 사용자를 삭제하고 functions 로그를 보면 functions 작동 내역을 확인할 수 있습니다.

 

log

 

이렇게 사용자 삭제가 될 때 트리거되는 함수를 작성해보았습니다.

 

clearFirebaseData 를 잘 작성해두면 이후 사용자가 "전체 데이터 삭제" 등의 옵션을 선택할 때도 활용할 수 있습니다.

firebase 에서는 앱에서 직접 collection 삭제를 권장하지 않기 때문에 (공식 예제도 없음) functions 를 활용해야 합니다.

 

전체 컬렉션을 삭제하려면 신뢰할 수 있는 서버 환경에서만 삭제하세요. 모바일 또는 웹 클라이언트에서도 컬렉션을 삭제할 수 있지만, 이 경우 보안 및 성능에 부정적인 영향을 미칩니다.

(https://firebase.google.com/docs/firestore/manage-data/delete-data?hl=ko#collections)

 

 

다음에는 Algolia 를 functions 로 연동해서 사용해보도록 하겠습니다.