Protocol Extension trong Swift: nó là gì và cách sử dụng

Protocol Extension là gì?

Protocol Extension là một cách để thêm triển khai của các phương thức và thuộc tính tính toán vào một protocol hiện có. Không có extension, protocol chỉ định nghĩa các yêu cầu và mỗi kiểu tự triển khai chúng.

Protocol Extension giải quyết vấn đề trùng lặp mã: nếu năm struct chấp nhận cùng một protocol và triển khai cùng một phương thức, extension cung cấp triển khai mặc định một lần.

Theo Swift Evolution proposal SE-0186, protocol extensions là một trong những tính năng quan trọng đã định trước thành công của POP. Chúng cho phép thêm hành vi chung mà không cần tạo lớp cơ sở và không vi phạm nguyên tắc đơn trách nhiệm.

Cú pháp protocol extension

Protocol Extension được khai báo như một extension thông thường nhưng với tên protocol thay vì một kiểu.

protocol Greetable {
    var name: String { get }
    func greet() -> String
}

extension Greetable {
    func greet() -> String {
        return "Hello, \(name)!"
    }
}

Bây giờ bất kỳ kiểu nào chấp nhận Greetable đều tự động nhận được triển khai greet:

struct Person: Greetable {
    let name: String
}
// Person tự động có greet()

let user = Person(name: "Alice")
print(user.greet()) // "Xin chào, Alice!"

Protocol Extension có thể chứa thuộc tính tính toán nhưng không phải thuộc tính lưu trữ (protocol không thể định nghĩa bộ nhớ). Bạn cũng có thể thêm subscripts và kiểu lồng nhau thông qua extension.

Triển khai phương thức mặc định

Triển khai mặc định là cách sử dụng chính của protocol extension. Một kiểu có thể ghi đè phương thức bằng cách cung cấp phiên bản riêng của nó.

protocol Loggable {
    func log(message: String)
}

extension Loggable {
    func log(message: String) {
        print("[Default] \(message)")
    }
}

struct ConsoleLogger: Loggable {}
// Sử dụng triển khai mặc định

struct FileLogger: Loggable {
    func log(message: String) {
        // Triển khai tùy chỉnh ghi đè mặc định
        writeToFile(message)
    }
}

Một điểm khác biệt quan trọng so với kế thừa lớp: nếu kiểu tự triển khai phương thức protocol, triển khai của nó được gọi. Nếu không, triển khai mặc định từ extension được sử dụng. Đây là dispatch tĩnh — quyết định được đưa ra tại thời điểm biên dịch.

Ràng buộc where trong protocol extension

Mệnh đề where cho phép giới hạn protocol extension chỉ cho những kiểu đáp ứng các điều kiện bổ sung. Đây là một cơ chế mạnh mẽ để thêm hành vi chuyên biệt.

protocol Printable {
    var content: String { get }
}

extension Printable where Self: CustomStringConvertible {
    func debugPrint() -> String {
        return "[Printable] \(content)"
    }
}

Ở đây debugPrint chỉ khả dụng cho các kiểu đồng thời triển khai Printable và CustomStringConvertible. Thư viện chuẩn của Swift sử dụng rộng rãi mẫu này — ví dụ: extensions cho Collection where Element.

Các mệnh đề where với ràng buộc đẳng thức kiểu đặc biệt hữu ích:

extension Collection where Element == String {
    func commaJoined() -> String {
        return self.joined(separator: ", ")
    }
}

let words = ["Swift", "Kotlin", "Java"]
print(words.commaJoined()) // "Swift, Kotlin, Java"

Cơ chế này làm cho protocol extension có tính chọn lọc: phương thức commaJoined chỉ khả dụng cho tập hợp chuỗi, không phải tập hợp số. Trình biên dịch kiểm tra các ràng buộc một cách tĩnh.

Các ràng buộc where có thể kiểm tra:

• Tuân thủ protocol: where Self: Equatable
• Ràng buộc kiểu: where Element == String
• Kết hợp: where Element: Numeric, Element: Comparable

Điều này làm cho protocol extension trở thành một cơ chế mạnh mẽ để thêm hành vi chuyên biệt mà không làm ô nhiễm triển khai protocol chung.

Protocol Extensions vs kế thừa

Nhiều nhà phát triển thắc mắc: khi nào sử dụng protocol extensions và khi nào sử dụng kế thừa lớp? Câu trả lời phụ thuộc vào mô hình kiến trúc.

Apple khuyến nghị bắt đầu với protocol + extension và chỉ chuyển sang lớp khi cần trạng thái chia sẻ hoặc định danh (ngữ nghĩa tham chiếu). Protocol Extensions cung cấp sự kết hợp thay vì kế thừa — một cách tiếp cận linh hoạt và dễ kiểm tra hơn.

Trong thực tế, protocol extensions thường được sử dụng để thêm các phương thức bao bọc tiện lợi trên các yêu cầu của protocol. Ví dụ: nếu một protocol yêu cầu phương thức validate với báo cáo chi tiết, extension có thể thêm phương thức isValid trả về giá trị boolean dựa trên phiên bản đầy đủ. Điều này đơn giản hóa mã client mà không thay đổi hợp đồng protocol. Mẫu này được gọi là “triển khai mặc định với API dẫn xuất” và được sử dụng rộng rãi trong thư viện chuẩn Swift và các framework bên thứ ba phổ biến. Đây là một trong những kỹ thuật chính của lập trình hướng protocol trong thực tế và là nền tảng của kiến trúc linh hoạt.