Extension trong Swift: khái niệm, cú pháp và cách thêm phương thức

Extension trong Swift là gì?

Extension trong Swift là một cấu trúc ngôn ngữ cho phép mở rộng một kiểu hiện có (lớp, struct, enum, giao thức) với các khả năng mới. Nó tương tự như category trong Objective-C nhưng mạnh mẽ hơn.

Extension có thể thêm:

• Thuộc tính tính toán — instance và kiểu
• Phương thức — instance và kiểu
• Bộ khởi tạo (convenience init)
• Subscript
• Kiểu lồng nhau
• Tuân thủ giao thức

Theo tài liệu Apple, extension không thể ghi đè phương thức hiện có (dynamic dispatch) và không thể thêm thuộc tính lưu trữ. Những giới hạn này bảo toàn tính toàn vẹn của kiểu gốc.

Cú pháp mở rộng

Extension được khai báo bằng từ khóa extension với kiểu được mở rộng.

extension Double {
    var km: Double { return self * 1000.0 }
    var cm: Double { return self / 100.0 }
    var mm: Double { return self / 1000.0 }
}

let distance = 5.0.km
print("5 km = \(distance) meters") // 5000.0

Có thể mở rộng không chỉ các kiểu riêng mà còn các kiểu thư viện chuẩn của Swift, Foundation, UIKit và các framework khác. Điều này làm cho extension trở thành công cụ mạnh mẽ để thêm chức năng dành riêng cho lĩnh vực.

Thêm phương thức qua extension

Extension cho phép thêm phương thức instance và kiểu vào bất kỳ kiểu hiện có nào.

extension String {
    func isValidEmail() -> Bool {
        let emailRegex = "[A-Z0-9a-z._%+-]+@[A-Za-z0-9.-]+\\.[A-Za-z]{2,}"
        return self.range(of: emailRegex, options: .regularExpression) != nil
    }

    static func random(length: Int) -> String {
        let letters = "abcdefghijklmnopqrstuvwxyz0123456789"
        return String((0..<length).map { _ in letters.randomElement()! })
    }
}

let email = "user@example.com"
print(email.isValidEmail()) // true
let code = String.random(length: 8) // "a3f8k2m1"

Phương thức mutating mở rộng các kiểu giá trị (struct, enum) và có thể sửa đổi self. Đối với lớp, không cần mutating. Extension có thể thêm phương thức mutating ngay cả vào các kiểu thư viện chuẩn.

Thuộc tính tính toán trong extension

Extension chỉ có thể thêm thuộc tính tính toán — thuộc tính lưu trữ không được phép. Điều này có lý vì extension không quản lý bộ nhớ của kiểu gốc.

extension Date {
    var isPast: Bool {
        return self < Date()
    }

    var isToday: Bool {
        return Calendar.current.isDateInToday(self)
    }
}

let someDate = Date().addingTimeInterval(3600)
print(someDate.isPast) // false (one hour from now is future)

Thuộc tính tính toán trong extension tuân theo cùng quy tắc như trong triển khai chính: có thể tạo thuộc tính { get } và { get set }. Swift không đặt giới hạn về số lượng extension cho một kiểu. Bạn có thể khai báo bao nhiêu extension tùy ý cho một kiểu trong các tệp khác nhau — đây là thực hành tiêu chuẩn trong các dự án lớn.

Tuân thủ giao thức qua extension

Extension là cách ưu tiên để triển khai các giao thức. Thay vì làm lộn xộn khai báo kiểu chính, tất cả các phương thức giao thức được đặt trong một extension riêng.

struct User {
    let id: Int
    let name: String
    let email: String
}

// Extension per protocol — clean separation
extension User: Equatable {
    static func == (lhs: User, rhs: User) -> Bool {
        return lhs.id == rhs.id
    }
}

extension User: Codable {
    enum CodingKeys: String, CodingKey {
        case id, name, email
    }
}

Cách tiếp cận này phân tách trách nhiệm: cấu trúc chính chỉ chứa thuộc tính lưu trữ, các extension chứa triển khai giao thức. Swift thậm chí cho phép triển khai giao thức qua extension cho các kiểu mà chúng ta không kiểm soát.

Extension và tổ chức mã

Extension không chỉ là một tính năng ngôn ngữ mà còn là công cụ tổ chức kiến trúc. Sử dụng extension đúng cách làm cho mã dễ đọc và dễ bảo trì.

• Nhóm theo giao thức — mỗi giao thức trong một extension riêng
• Đánh dấu (MARK) — // MARK: - Equatable để điều hướng
• Phân tách tệp — extension cho các giao thức khác nhau trong các tệp khác nhau
• Helper riêng — phương thức riêng trong extension được đánh dấu private

// MARK: - User + Displayable
extension User: Displayable {
    var displayName: String {
        return "\(name) (\(email))"
    }
}

Theo Swift API Design Guidelines, extensions cải thiện khả năng đọc: nhà phát triển thấy cấu trúc dữ liệu trong khai báo chính và logic trong các extension. Điều này phù hợp với nguyên tắc phân tách giao diện.

Extension với ràng buộc where thêm phương thức chỉ cho các kiểu cụ thể: extension Array where Element: Equatable { func isUnique() -> Bool }. Mã như vậy ngắn gọn và an toàn kiểu — phương thức isUnique chỉ xuất hiện trên các mảng có phần tử Equatable, loại bỏ lỗi khi biên dịch và làm cho API sạch hơn, an toàn hơn và rõ ràng hơn cho các nhà phát triển khác.