如何在Firestore中使用电子邮件和密码创建用户时从OnCompleteListener返回值？

FirebaseAuth mAuth = FirebaseAuth.getInstance();
FirebaseFirestore db = FirebaseFirestore.getInstance();

String email = "example@example.com";
String password = "password123";

mAuth.createUserWithEmailAndPassword(email, password)
    .addOnCompleteListener(new OnCompleteListener<AuthResult>() {
        @Override
        public void onComplete(@NonNull Task<AuthResult> task) {
            if (task.isSuccessful()) {
                // 用户创建成功
                FirebaseUser user = mAuth.getCurrentUser();
                String userId = user.getUid();

                // 在Firestore中创建用户文档
                Map<String, Object> userData = new HashMap<>();
                userData.put("email", email);
                userData.put("userId", userId);

                db.collection("users").document(userId)
                    .set(userData)
                    .addOnSuccessListener(new OnSuccessListener<Void>() {
                        @Override
                        public void onSuccess(Void aVoid) {
                            // 用户文档创建成功
                            Log.d(TAG, "User document created successfully");
                        }
                    })
                    .addOnFailureListener(new OnFailureListener() {
                        @Override
                        public void onFailure(@NonNull Exception e) {
                            // 用户文档创建失败
                            Log.e(TAG, "Error creating user document", e);
                        }
                    });
            } else {
                // 用户创建失败
                Log.e(TAG, "Error creating user", task.getException());
            }
        }
    });