Google Firebase：膳食规划App首次用户启动的数据处理问询

问题分析 #

你的代码存在以下核心问题：

• 已登录但数据库无用户记录时，未初始化空的meals和mealIngredients节点，导致后续添加餐食逻辑出错
• 添加餐食时直接用userRef.setValue会覆盖整个用户节点，破坏数据结构
• 未登录场景（else分支）未做处理，无法引导用户进入登录流程

修复后的完整代码 #

private let database = Database.database().reference()
private let auth = Auth.auth()

var meals: [String] = []
var mealIngredients: [String] = []

override func viewDidLoad() {
    super.viewDidLoad()
    
    if let user = auth.currentUser {
        let userId = user.uid
        let userRef = database.child("users").child(userId)
        
        // 读取meals数据，处理首次登录无记录的情况
        userRef.child("meals").observeSingleEvent(of: .value, with: { snapshot in
            if let mealsData = snapshot.value as? [String] {
                self.meals = mealsData
            } else {
                // 首次登录，初始化空数组到数据库
                self.meals = []
                userRef.child("meals").setValue(self.meals)
            }
        })
        
        // 读取mealIngredients数据，处理首次登录无记录的情况
        userRef.child("mealIngredients").observeSingleEvent(of: .value, with: { snapshot in
            if let ingredientsData = snapshot.value as? [String] {
                self.mealIngredients = ingredientsData
            } else {
                // 首次登录，初始化空数组到数据库
                self.mealIngredients = []
                userRef.child("mealIngredients").setValue(self.mealIngredients)
            }
        })
    } else {
        /*User launching app for first time - 引导用户登录*/
        // 示例：跳转至登录页面（根据你的App路由调整）
        // let loginVC = LoginViewController()
        // navigationController?.pushViewController(loginVC, animated: true)
        print("用户未登录，请引导登录")
    }
}

@IBAction func addMeal(_ sender: Any) {
    guard let mealName = mealNameField.text, !mealName.isEmpty,
          let mealIngredientsText = mealIngredientsField.text, !mealIngredientsText.isEmpty else {
        return
    }
    
    // 更新本地数组
    self.meals.append(mealName)
    self.mealIngredients.append(mealIngredientsText)
    
    // 清空输入框
    mealNameField.text = ""
    mealIngredientsField.text = ""
    
    guard let currentUser = auth.currentUser else {
        print("用户未登录，无法添加餐食")
        return
    }
    
    let userId = currentUser.uid
    let userRef = database.child("users").child(userId)
    
    // 分别更新meals和mealIngredients子节点，避免覆盖整个用户节点
    userRef.child("meals").setValue(self.meals)
    userRef.child("mealIngredients").setValue(self.mealIngredients)
}

关键说明 #

• 自动创建节点：Firebase实时数据库会在写入数据时自动创建不存在的节点，无需手动判断用户记录是否存在后再创建
• 避免节点覆盖：使用child("meals").setValue而非直接在userRef上调用setValue，防止覆盖用户节点下的其他字段（比如后续可能添加的用户偏好等）
• 首次登录初始化：在读取数据时，如果发现节点不存在，主动写入空数组，确保本地数据和数据库数据一致，避免后续添加餐食时出现逻辑异常
• 未登录场景处理：在else分支中添加登录引导逻辑，确保只有已登录用户才能使用餐食管理功能

内容的提问来源于stack exchange，提问作者akshaysub27