如何在Swift中实现不同区块的objc_setAssociatedObject？

import ObjectiveC

// 定义一个闭包类型，用于存储关联对象的值
typealias AssociatedClosure = () -> Void

// 定义一个关联的键
private var associatedClosureKey: UInt8 = 0

extension NSObject {
    // 定义一个方法，用于设置关联对象
    func setAssociatedClosure(_ closure: AssociatedClosure, forKey key: UnsafeRawPointer) {
        // 使用objc_setAssociatedObject函数将闭包与对象关联起来
        objc_setAssociatedObject(self, key, closure, .OBJC_ASSOCIATION_COPY_NONATOMIC)
    }
    
    // 定义一个方法，用于获取关联对象
    func associatedClosure(forKey key: UnsafeRawPointer) -> AssociatedClosure? {
        // 使用objc_getAssociatedObject函数获取关联对象
        return objc_getAssociatedObject(self, key) as? AssociatedClosure
    }
}

// 示例用法
class MyClass: NSObject {
    func doSomething() {
        // 定义一个闭包
        let closure: AssociatedClosure = {
            print("Associated closure called")
        }
        
        // 设置关联对象
        setAssociatedClosure(closure, forKey: &associatedClosureKey)
        
        // 获取关联对象并调用闭包
        if let associatedClosure = associatedClosure(forKey: &associatedClosureKey) {
            associatedClosure()
        }
    }
}

// 创建一个实例并调用方法
let myObject = MyClass()
myObject.doSomething()