SwiftUI新版@Environment无法使用协议注入服务的问题求助

问题核心 #

WWDC 2023推出的Observation框架搭配新版@Environment时，确实不直接支持协议类型的依赖注入——它是基于具体类型作为键来存储/读取环境值的，和旧版@EnvironmentObject依赖协议的逻辑完全不同。你遇到的编译器错误正是这个限制导致的。

可行解决方案 #

方案1：类型擦除包装器

通过一个@Observable的类型擦除类，将协议实现包装成具体类型，适配新版@Environment的要求。

// 调整协议为类协议
protocol UserServiceRepresentable: AnyObject {
    var user: User { get }
    func fetchUser() async
}

// 类型擦除包装类，符合@Observable和协议
@Observable
final class AnyUserService: UserServiceRepresentable {
    private let base: UserServiceRepresentable
    
    var user: User { base.user }
    
    init(_ base: UserServiceRepresentable) {
        self.base = base
    }
    
    func fetchUser() async {
        await base.fetchUser()
    }
}

// 原服务实现
@Observable
final class UserService: UserServiceRepresentable {
    private(set) var user: User = User() // 示例初始化
    
    func fetchUser() async {
        // 真实业务逻辑
        user = User(name: "Production User")
    }
}

// Mock服务
@Observable
final class UserServiceMock: UserServiceRepresentable {
    let user: User = User(name: "Mock User")
    
    func fetchUser() async {
        // Mock逻辑
    }
}

使用方式：

struct UserView: View {
    @Environment(AnyUserService.self) private var userService
    
    var body: some View {
        Text(userService.user.name)
            .task {
                await userService.fetchUser()
            }
    }
}

#Preview {
    UserView()
        .environment(AnyUserService(UserServiceMock()))
}

方案2：自定义环境键（EnvironmentKey）

通过自定义环境键，直接将协议类型作为环境值的类型，绕开@Environment对具体类型的限制。

// 协议需继承Observable（@Observable类会自动符合该协议）
protocol UserServiceRepresentable: AnyObject, Observable {
    var user: User { get }
    func fetchUser() async
}

// 自定义环境键
struct UserServiceKey: EnvironmentKey {
    // 设置默认值为真实服务
    static var defaultValue: UserServiceRepresentable = UserService()
}

// 扩展EnvironmentValues添加协议属性
extension EnvironmentValues {
    var userService: UserServiceRepresentable {
        get { self[UserServiceKey.self] }
        set { self[UserServiceKey.self] = newValue }
    }
}

// 原服务和Mock保持@Observable修饰
@Observable
final class UserService: UserServiceRepresentable {
    private(set) var user: User = User()
    
    func fetchUser() async {
        user = User(name: "Production User")
    }
}

@Observable
final class UserServiceMock: UserServiceRepresentable {
    let user: User = User(name: "Mock User")
    
    func fetchUser() async {}
}

使用方式：

struct UserView: View {
    @Environment(\.userService) private var userService
    
    var body: some View {
        Text(userService.user.name)
            .task {
                await userService.fetchUser()
            }
    }
}

#Preview {
    UserView()
        .environment(\.userService, UserServiceMock())
}

方案对比 #

• 类型擦除：贴合@Observable原生用法，无需自定义环境键，但需要额外编写包装类，适合协议方法较少的场景。
• 自定义环境键：直接使用协议类型，无需额外包装，但要求协议继承Observable，且需合理设置默认值。

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