Type erasure concept in Swift

Introduction

Type erasure helps hide complex or different types behind a simple, common type. This makes code easier to use and understand.

When you want to store different types that follow the same protocol in one collection.

When you want to return a protocol type from a function but the protocol has associated types or self requirements.

When you want to simplify complex generic types for easier use in your code.

When you want to hide implementation details from users of your API.

Syntax

Swift

struct AnyExample: ExampleProtocol {
    private let _doSomething: () -> Void

    init<T: ExampleProtocol>(_ example: T) {
        _doSomething = example.doSomething
    }

    func doSomething() {
        _doSomething()
    }
}

Type erasure is usually done by creating a wrapper struct or class.

The wrapper stores the original type's behavior using closures or stored properties.

Examples

Two shapes with different types but same protocol.

Swift

protocol Shape {
    func area() -> Double
}

struct Circle: Shape {
    var radius: Double
    func area() -> Double {
        return Double.pi * radius * radius
    }
}

struct Square: Shape {
    var side: Double
    func area() -> Double {
        return side * side
    }
}

Type erasure wrapper hides the specific shape type.

Swift

struct AnyShape: Shape {
    private let _area: () -> Double

    init<T: Shape>(_ shape: T) {
        _area = shape.area
    }

    func area() -> Double {
        return _area()
    }
}

Store different shapes in one array using type erasure.

Swift

let shapes: [AnyShape] = [AnyShape(Circle(radius: 3)), AnyShape(Square(side: 4))]
for shape in shapes {
    print(shape.area())
}

Sample Program

This program shows how to use type erasure to store different shapes in one array and call their area method.

Swift

import Foundation

protocol Shape {
    func area() -> Double
}

struct Circle: Shape {
    var radius: Double
    func area() -> Double {
        return Double.pi * radius * radius
    }
}

struct Square: Shape {
    var side: Double
    func area() -> Double {
        return side * side
    }
}

struct AnyShape: Shape {
    private let _area: () -> Double

    init<T: Shape>(_ shape: T) {
        _area = shape.area
    }

    func area() -> Double {
        return _area()
    }
}

let shapes: [AnyShape] = [AnyShape(Circle(radius: 3)), AnyShape(Square(side: 4))]

for shape in shapes {
    print(String(format: "%.2f", shape.area()))
}

Important Notes

Type erasure helps when protocols have associated types or self requirements that prevent direct use.

It adds a small performance cost due to indirection but improves code flexibility.

Summary

Type erasure hides specific types behind a common interface.

It allows storing and using different types uniformly.

It is useful for protocols with associated types or complex generics.

Practice

1. What is the main purpose of type erasure in Swift?

easy

A. To convert classes into structs automatically

B. To create new types from existing ones

C. To improve runtime performance by avoiding protocols

D. To hide specific types behind a common interface

Solution

1. Step 1: Understand type erasure concept

   Type erasure is used to hide the concrete underlying type and expose a common interface.

2. Step 2: Identify the main purpose

   This allows different types to be used uniformly without exposing their specific details.

3. Final Answer:

   To hide specific types behind a common interface -> Option D

4. Quick Check:

   Type erasure = hiding types behind interface [OK]

Hint: Type erasure hides types to unify usage [OK]

Common Mistakes:

• Confusing type erasure with type conversion
• Thinking it improves performance directly
• Assuming it creates new types

2. Which of the following is the correct syntax to create a type-erased wrapper for a protocol MyProtocol?

easy

A. struct AnyMyProtocol: MyProtocol { private let _base: any MyProtocol }

B. struct AnyMyProtocol<T>: MyProtocol { private let _base: T }

C. struct AnyMyProtocol: MyProtocol { private let _base: MyProtocol }

D. struct AnyMyProtocol: MyProtocol { private let _base: some MyProtocol }

Solution

1. Step 1: Recall type erasure syntax

   Type erasure uses a struct that conforms to the protocol and stores an instance of any Protocol to hide the concrete type.

2. Step 2: Identify correct syntax

   struct AnyMyProtocol: MyProtocol { private let _base: any MyProtocol } uses any MyProtocol which is the correct way to store an existential type in Swift 5.7+.

3. Final Answer:

   struct AnyMyProtocol: MyProtocol { private let _base: any MyProtocol } -> Option A

4. Quick Check:

   Use any Protocol for type erasure storage [OK]

Hint: Use 'any Protocol' to store erased type [OK]

Common Mistakes:

• Using generic parameter instead of existential
• Using 'some Protocol' which is opaque, not erased
• Trying to store protocol directly without 'any'

3. Consider this Swift code using type erasure:

protocol Shape {
    func area() -> Double
}

struct Circle: Shape {
    var radius: Double
    func area() -> Double { return .pi * radius * radius }
}

struct Square: Shape {
    var side: Double
    func area() -> Double { return side * side }
}

struct AnyShape: Shape {
    private let _area: () -> Double
    init(_ shape: S) {
        _area = shape.area
    }
    func area() -> Double { _area() }
}

let shapes: [AnyShape] = [AnyShape(Circle(radius: 2)), AnyShape(Square(side: 4))]
let totalArea = shapes.reduce(0) { $0 + $1.area() }
print(totalArea)

What is the output?

medium

A. 28.566370614359172

B. 13.141592653589793

C. Error: Cannot convert Circle to AnyShape

D. 0

Solution

1. Step 1: Calculate area of Circle and Square

   Circle area = π * 2 * 2 = 12.566370614359172, Square area = 4 * 4 = 16

2. Step 2: Sum areas in array

   Total area = 12.566370614359172 + 16 = 28.566370614359172

3. Step 3: Check code output

   Code prints totalArea, which is 28.566370614359172

4. Final Answer:

   28.566370614359172 -> Option A

5. Quick Check:

   Sum of areas = 28.566370614359172 [OK]

Hint: Add areas of shapes wrapped in AnyShape [OK]

Common Mistakes:

• Calculating area incorrectly
• Confusing type erasure with type casting
• Expecting compile error due to protocol

4. This Swift code attempts type erasure but causes a compile error:

protocol Printer {
    associatedtype Item
    func printItem(_ item: Item)
}

struct AnyPrinter: Printer {
    private let _print: (Any) -> Void
    init(_ printer: P) {
        _print = printer.printItem
    }
    func printItem(_ item: Any) {
        _print(item)
    }
}

What is the main cause of the error?

medium

A. Missing conformance to Printer protocol

B. Cannot assign generic method to closure with Any parameter

C. Cannot use associatedtype in protocol

D. Function printItem has wrong return type

Solution

1. Step 1: Analyze closure assignment

   The closure _print expects (Any) -> Void but printer.printItem expects (P.Item) -> Void, which is a mismatch.

2. Step 2: Understand type erasure limitation

   Associated types prevent direct assignment because the types differ; you cannot assign a generic method to a closure with Any parameter.

3. Final Answer:

   Cannot assign generic method to closure with Any parameter -> Option B

4. Quick Check:

   Generic method ≠ closure with Any parameter [OK]

Hint: Generic methods can't be assigned to Any closures directly [OK]

Common Mistakes:

• Thinking protocol conformance is missing
• Confusing associatedtype usage
• Assuming return type mismatch

5. You want to create a type-erased wrapper for a protocol DataSource with an associated type Item and a method func fetch() -> [Item]. Which approach correctly applies type erasure to allow storing different DataSource implementations in a single array?

hard

A. Create a struct AnyDataSource with a private closure _fetch: () -> [Any] and cast results inside

B. Use a generic struct AnyDataSource<T> storing DataSource directly

C. Create a struct AnyDataSource with a private closure _fetch: () -> [Item] and use type erasure with a generic initializer

D. Use protocol inheritance to remove associated type

Solution

1. Step 1: Understand associated type challenge

   Protocols with associated types cannot be used directly in collections because the type varies.

2. Step 2: Apply type erasure pattern

   Create a struct with a private closure returning [Item], and initialize it generically to wrap any DataSource, hiding the concrete type.

3. Step 3: Evaluate options

   Create a struct AnyDataSource with a private closure _fetch: () -> [Item] and use type erasure with a generic initializer correctly uses a generic initializer and closure with [Item], preserving type safety and enabling storage in arrays.

4. Final Answer:

   Create a struct AnyDataSource with a private closure _fetch: () -> [Item] and use type erasure with a generic initializer -> Option C

5. Quick Check:

   Type erasure with generic init and closure = Create a struct AnyDataSource with a private closure _fetch: () -> [Item] and use type erasure with a generic initializer [OK]

Hint: Use generic init and closure returning [Item] for erasure [OK]

Common Mistakes:

• Casting to [Any] loses type safety
• Using generic struct prevents uniform storage
• Trying to remove associated type via inheritance