如何自定义Swift ForEach结构体并保留原有语法实现索引功能？

问题背景 #

自定义了一个带索引功能的ForEach结构体，语法和SwiftUI原生ForEach接近，实现代码如下：

struct ForEach<Data: RandomAccessCollection,Content: View>: View where Data.Element: Identifiable, Data.Element: Hashable {
    let data: Data
    @ViewBuilder let content: (Data.Index, Data.Element) -> Content
    
    init(_ data: Data, content: @escaping (Data.Index, Data.Element) -> Content) {
        self.data = data
        self.content = content
    }
    
    var body: some View {
        SwiftUI.ForEach(Array(zip(data.indices, data)), id: \.1) { index, element in
            content(index, element)
        }
    }
}

该结构体支持带索引的调用语法：

ForEach(array) { index, item in
    // 处理逻辑
}

但原有SwiftUI风格的仅传入元素的语法ForEach(array) { item in }无法正常工作，且因代码复用需求不能用_忽略索引，需要同时保留两种调用语法。

修复方案 #

方案1：给自定义结构体添加双初始化器 #

通过新增一个仅接收元素闭包的初始化器，让Swift根据闭包参数自动匹配对应的初始化逻辑，同时复用原生ForEach的实现来兼容原有语法。修改后的完整代码：

struct ForEach<Data: RandomAccessCollection, Content: View>: View where Data.Element: Identifiable {
    private let data: Data
    private let contentView: AnyView
    
    // 带索引的初始化器
    init(_ data: Data, content: @escaping (Data.Index, Data.Element) -> Content) {
        self.data = data
        self.contentView = AnyView(
            SwiftUI.ForEach(Array(zip(data.indices, data)), id: \.1) { index, element in
                content(index, element)
            }
        )
    }
    
    // 兼容原有语法的初始化器
    init(_ data: Data, content: @escaping (Data.Element) -> Content) {
        self.data = data
        self.contentView = AnyView(
            SwiftUI.ForEach(data) { element in
                content(element)
            }
        )
    }
    
    var body: some View {
        contentView
    }
}

说明：

• 移除了多余的Data.Element: Hashable约束，Identifiable已满足原生ForEach的id需求
• 通过AnyView统一两种初始化器的视图输出类型
• Swift会根据闭包的参数个数自动选择对应的初始化器，两种调用语法均可正常使用

方案2：扩展原生SwiftUI.ForEach #

如果不想自定义新结构体，直接扩展原生ForEach添加带索引的初始化器，这样无需修改原有代码，同时新增带索引功能：

extension SwiftUI.ForEach where Data: RandomAccessCollection, Data.Element: Identifiable {
    init(_ data: Data, content: @escaping (Data.Index, Data.Element) -> Content) {
        self.init(Array(zip(data.indices, data)), id: \.1) { index, element in
            content(index, element)
        }
    }
}

说明：

• 直接增强原生ForEach的功能，原有仅传元素的语法保持不变
• 当闭包传入两个参数时，自动调用新增的带索引初始化器

使用示例 #

两种方案均支持以下两种调用方式：

// 带索引的新语法
ForEach(items) { index, item in
    Text("第\(index)项：\(item.title)")
}

// 原有仅传元素的语法
ForEach(items) { item in
    Text(item.title)
}

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