如何根据参数属性约束Swift泛型函数的类型推断？

这个需求的核心是要让ItemType的每个case和特定的模型类型（比如UIImage、视频类）建立强绑定，让编译器能根据传入的枚举值自动推导泛型闭包的类型。下面是两种可行的实现方式，优先推荐类型安全的协议方案：

方案一：使用关联类型协议（类型安全，推荐） #

我们可以定义一个带关联类型的协议，把媒体类型的UTType字符串和对应的模型类型绑定起来，再为图片、视频分别实现这个协议的结构体，最后让泛型函数基于这个协议做约束，编译器就能自动推断闭包类型了。

步骤1：定义协议与对应类型 #

import MobileCoreServices
import UIKit
import AVFoundation

// 定义协议，关联UTType字符串和对应的模型类型
protocol ItemRepresentable {
    associatedtype Item: NSSecureCoding
    var utType: String { get }
}

// 图片类型实现：绑定UIImage
struct ImageItem: ItemRepresentable {
    typealias Item = UIImage
    var utType: String { String(kUTTypeImage) }
}

// 视频类型实现：这里以AVAsset为例，你可以换成自己需要的视频类（比如PHAsset）
struct MovieItem: ItemRepresentable {
    typealias Item = AVAsset
    var utType: String { String(kUTTypeMovie) }
}

步骤2：修改泛型函数 #

把函数的参数类型改为ItemRepresentable，利用协议的关联类型自动推导闭包的参数类型：

func items<MediaType: ItemRepresentable>(for type: MediaType, completion: ([MediaType.Item]) -> Void) {
    // 这里可以通过type.utType获取对应的UTType字符串，执行你的资源获取逻辑
    // 示例：模拟获取到的数据，替换成你的真实业务逻辑
    let mockItems: [MediaType.Item] = []
    completion(mockItems)
}

调用示例（自动推断类型） #

调用时不需要手动指定泛型类型，编译器会根据传入的ImageItem/MovieItem自动推断闭包的参数类型：

// 图片场景：completion自动推断为([UIImage]) -> Void
items(for: ImageItem()) { images in
    // images的类型自动是[UIImage]，可以直接调用UIImage的方法
    images.forEach { image in
        print(image.size)
    }
}

// 视频场景：completion自动推断为([AVAsset]) -> Void
items(for: MovieItem()) { movies in
    // movies的类型自动是[AVAsset]
    movies.forEach { asset in
        print(asset.duration)
    }
}

方案二：基于原枚举的改进（兼容原有枚举结构） #

如果你想保留原来的ItemType枚举结构，可以给枚举添加一个泛型方法，通过switch强制绑定类型（注意：这种方式需要确保调用时类型匹配，否则会有运行时风险）：

修改枚举与泛型函数 #

import MobileCoreServices
import UIKit
import AVFoundation

enum ItemType: String {
    case image
    case movie
    
    var utType: String {
        switch self {
        case .image: return String(kUTTypeImage)
        case .movie: return String(kUTTypeMovie)
        }
    }
    
    // 泛型方法：根据case绑定对应类型
    func fetchItems<T: NSSecureCoding>(completion: ([T]) -> Void) {
        // 这里执行你的资源获取逻辑，根据self返回对应类型的数据
        switch self {
        case .image:
            let fetchedImages: [UIImage] = [] // 替换为真实的图片数据
            completion(fetchedImages as! [T])
        case .movie:
            let fetchedMovies: [AVAsset] = [] // 替换为真实的视频数据
            completion(fetchedMovies as! [T])
        }
    }
}

调用示例 #

调用时只需指定一次闭包参数的类型，编译器会自动匹配：

// 图片场景：闭包参数指定[UIImage]，自动匹配
ItemType.image.fetchItems { (images: [UIImage]) in
    print(images.count)
}

// 视频场景：闭包参数指定[AVAsset]，自动匹配
ItemType.movie.fetchItems { (movies: [AVAsset]) in
    print(movies.count)
}

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