ios - Swift 中 UITextView 和 UITextField 的单一扩展

Question

我想为 UITextField 和 UITextView 创建一个extension 并向其添加以下方法:

    func addDoneButtonOnKeyboardWith(selector : Selector)
    {
        let keyBoardToolBar = UIToolbar(frame: CGRect(x: 0, y: 0, width: 320, height: 30))
        let barButtonItem = UIBarButtonItem(barButtonSystemItem: .done, target: nil, action: selector)
        let flexibleSpace = UIBarButtonItem(barButtonSystemItem: .flexibleSpace, target: nil, action: nil)
        keyBoardToolBar.items = [flexibleSpace, barButtonItem]
        keyBoardToolBar.barStyle = .default
        self.inputAccessoryView = keyBoardToolBar
        self.reloadInputViews()
    }

我希望扩展方法仅适用于 UITextField 和 UITextView。

Answer 1

也许这行得通，因为 UIView 是它们的父类，缺点是这可能会出现在各种 View 上，不确定是否有其他方法可以实现您的需要:

extension UIView  {
    func addDoneButtonOnKeyboardWith(selector : Selector)
    {
        if self is UITextField || self is UITextView {
            //do something
        }
    }
}