如何在ViewModel中传递ModelContext或更新SwiftData

针对你的需求，提供两种可行的实现方式，直接解决SwiftData数据更新的问题：

方案一：将ModelContext传入ViewModel直接操作 #

让ViewModel同时负责位置获取和SwiftData更新，逻辑更内聚。

步骤1：修改ViewModel，添加ModelContext支持 #

final class LocationTestViewModel: NSObject, ObservableObject, CLLocationManagerDelegate {
    let locationManager = CLLocationManager()
    var modelContext: ModelContext? // 存储SwiftData上下文

    override init() {
        super.init()
        locationManager.delegate = self
    }

    func getCurrentLocation() {
        locationManager.requestLocation()
    }

    func locationManager(_ manager: CLLocationManager, didUpdateLocations locations: [CLLocation]) {
        guard let latestLocation = locations.first, let context = modelContext else {
            return
        }

        DispatchQueue.main.async {
            // 创建UserLocation实例并插入到SwiftData
            let newLocation = UserLocation(
                latitude: latestLocation.coordinate.latitude,
                longitude: latestLocation.coordinate.longitude
                // 可根据你的模型补充其他属性，比如时间戳
            )
            context.insert(newLocation)
        }
    }

    func locationManager(_ manager: CLLocationManager, didFailWithError error: any Error) {
        print(error)
    }
}

步骤2：在视图中注入ModelContext到ViewModel #

struct LocationTestView: View {
    
    @Environment(\.modelContext) private var modelContext
    @Query private var userLocations: [UserLocation]
    @StateObject private var viewModel = LocationTestViewModel()
    
    var body: some View {
        ZStack(alignment: .bottom) {
            List {
                ForEach(userLocations) { userLocation in
                    Text("纬度：\(userLocation.latitude)，经度：\(userLocation.longitude)")
                }
            }
            Button(action: addUserLocation, label: {
                Text("获取位置")
            })
        }
        .onAppear {
            // 视图出现时将上下文传递给ViewModel
            viewModel.modelContext = modelContext
        }
    }
    
    private func addUserLocation() {
        withAnimation {
            viewModel.getCurrentLocation()
        }
    }
}

方案二：视图监听ViewModel的@Published变量更新SwiftData #

保持ViewModel只负责位置获取，视图层处理SwiftData写入，职责更分离。

修改视图代码，添加监听逻辑 #

struct LocationTestView: View {
    
    @Environment(\.modelContext) private var modelContext
    @Query private var userLocations: [UserLocation]
    @StateObject private var viewModel = LocationTestViewModel()
    
    var body: some View {
        ZStack(alignment: .bottom) {
            List {
                ForEach(userLocations) { userLocation in
                    Text("纬度：\(userLocation.latitude)，经度：\(userLocation.longitude)")
                }
            }
            Button(action: addUserLocation, label: {
                Text("获取位置")
            })
        }
        // 监听ViewModel的坐标数组变化
        .onReceive(viewModel.$coordinates) { newCoordinates in
            // 只处理最新添加的坐标，避免重复操作
            guard let latestCoord = newCoordinates.last else { return }
            let newLocation = UserLocation(
                latitude: latestCoord.latitude,
                longitude: latestCoord.longitude
            )
            modelContext.insert(newLocation)
        }
    }
    
    private func addUserLocation() {
        withAnimation {
            viewModel.getCurrentLocation()
        }
    }
}

补充：确保UserLocation模型正确定义 #

你的SwiftData模型需要符合@Model要求，示例如下：

import SwiftData

@Model
class UserLocation {
    var latitude: Double
    var longitude: Double
    var timestamp: Date // 可选，记录位置获取时间

    init(latitude: Double, longitude: Double, timestamp: Date = .now) {
        self.latitude = latitude
        self.longitude = longitude
        self.timestamp = timestamp
    }
}

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