Swift：如何实现结构体接收多种ShapeStyle兼容类型作为参数并返回对应类型

你遇到的问题核心在于ShapeStyle协议带有Self关联类型要求，所以不能直接把它用作普通的参数或返回类型。要实现你想要的“传入什么类型的ShapeStyle，就返回什么类型”的效果，泛型是Swift里的最佳解决方案，和TypeScript的类型推断异曲同工。

下面是改造后的完整实现：

import SwiftUI
import PlaygroundSupport

// 把RectangleStyle改成泛型结构体，关联具体的ShapeStyle类型
struct RectangleStyle<Style: ShapeStyle> {
    private let style: Style
    
    // 泛型构造器，接收任意符合ShapeStyle的类型
    init(style: Style) {
        self.style = style
    }
    
    // 返回具体的Style类型，和构造时传入的类型完全一致
    func getStyle() -> Style {
        style
    }
}

struct DemoView: View {
    var body: some View {
        // 使用渐变的例子
        Rectangle()
            .fill(
                RectangleStyle(style: LinearGradient(
                    gradient: Gradient(colors: [.red, .blue]),
                    startPoint: .top,
                    endPoint: .bottom
                )).getStyle()
            )
            .frame(width: 200, height: 200)
            .padding()
        
        // 使用纯色的例子，类型自动推断为Color
        Rectangle()
            .fill(RectangleStyle(style: Color.green).getStyle())
            .frame(width: 200, height: 200)
    }
}

PlaygroundPage.current.setLiveView(DemoView())

关键说明： #

• 泛型绑定：通过RectangleStyle<Style: ShapeStyle>，我们让结构体和具体的ShapeStyle实现类型绑定，这样Swift可以在编译期就确定传入和返回的具体类型，避免了类型擦除带来的强制解包问题。
• 类型一致性：构造器接收Style类型，getStyle()也返回Style类型，完全满足你“传入A类型就返回A类型”的需求，Swift会自动推断具体类型（比如传入LinearGradient就推断Style为LinearGradient，传入Color就推断为Color）。
• 消除繁琐代码：对比你原来的写法，不需要再手动解包gradient!，代码更安全也更简洁。

如果你需要支持多种可选的样式（比如同时支持color和gradient的可选存储），也可以结合泛型和枚举来实现，但如果你的核心需求是“统一接收任意ShapeStyle并原样返回”，上面的泛型方案是最直接的。

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