Storageで生成されたURLのFireStoreへの保存

Question

Firebase Auth/Storage/FireStore を使った新規登録の処理を作っています。

以下は新規登録(まずAuthで認証し、帰ってきたUserをFireStoreに登録、画像があるならばそのURLも一緒に保存する)処理です。

static func signUp(name: String, email: String, password: String, image: UIImage?, onSuccess: @escaping () -> Void, onError: @escaping (_ errorMessage: String?) -> Void) {
        Auth.auth().createUser(withEmail: email, password: password, completion: { user, error in
            if error != nil {
                onError(error)
                return
            }
            guard let uid = user?.user.uid else { return }
            var dict: [String: Any] = [
                "name": name,
                "email": email
            ]
            // If Image is Set
            if let image = image {
                StorageService.storage(image: image, path: .icon, id: uid) { (imageUrl) in
                    dict["iconUrl"] = imageUrl
                }
            }
            Firestore.firestore().collection("users").document(uid).setData(dict) { (error) in
                if let error = error {
                    print(error)
                    return
                }
            }
            onSuccess()
        })
    }

以下はStorageのUIImageを引数にとり、URLを返す処理を関数にしたものです

class StorageService {

    // Upload Image to Storage
    static func storage(image: UIImage?, path: PathType, id: String, completion: @escaping (_ imageUrl: String?) -> ()) {
        guard let image = image, let imageData = UIImageJPEGRepresentation(image, 0.1) else {
            print("Non Image")
            completion(nil)
            return
        }
        let storageRef = Storage.storage().reference().child(path.rawValue).child(id)
        storageRef.putData(imageData, metadata: nil, completion: { (metaData, error) in
            if let error = error {
                print("Fail to Put Data in Storage : (error)")
                completion(nil)
                return
            }
            storageRef.downloadURL { (imageUrl, error) in
                if let error = error {
                    print("Fail to Download Url : (error)")
                    completion(nil)
                    return
                }
                if let imageUrl = imageUrl?.absoluteString {
                    completion(imageUrl)
                }
            }
        })
    }
}

Authの登録と、FireStoreへの保存は成功するのですが、画像がある時、
Storageには保存されているのに、FireStoreには画像のURlが保存されません。

storage()クロージャの書き方などに問題があるのでしょうか？

Answer 1

非同期処理とコールバックが呼ばれるタイミングについて意識してコードを書くようにしてみましょう。

if let image = image {
    StorageService.storage(image: image, path: .icon, id: uid) { (imageUrl) in
        dict["iconUrl"] = imageUrl
    }
}
Firestore.firestore().collection("users").document(uid).setData(dict) { (error) in
    if let error = error {
        print(error)
        return
    }
}

こちらの処理ですが、StorageService.storageに渡しているコールバックの中身は画像のアップロードが終わった時点で呼ばれるように実装されていますね。つまり、下のsetDataの処理のほうが先に呼ばれてしまうので、画像のURLが含まれない状態でFirestoreに保存されてしまいます。

以下のように、アップロードが終わった時点でFirestoreに保存するよう修正しましょう。imageがnilの場合の処理はstorageメソッド内で完結していますので、 if let image = image { } による場合分けは不要です。

StorageService.storage(image: image, path: .icon, id: uid) { (imageUrl) in
    // 画像のアップロードが終わったらURLをセット
    // （画像がなかったりエラーの場合はnilが渡されるので何も設定しない）
    dict["iconUrl"] = imageUrl

    // Firestoreに保存する
    Firestore.firestore().collection("users").document(uid).setData(dict) { (error) in
        ...
    }
}