refactor: set textfield defalut clear button image

ls19491314 · May 21, 2019 · d8a518e · d8a518e
1 parent f46ab49
commit d8a518e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/QMUIKit/UIKitExtensions/UITextField+QMUI.h b/QMUIKit/UIKitExtensions/UITextField+QMUI.h
@@ -24,7 +24,7 @@
 /// 输入框右边的 clearButton，在 UITextField 初始化后就存在
 @property(nullable, nonatomic, weak, readonly) UIButton *qmui_clearButton;
 
-/// 自定义 clearButton 的图片，注意如果设置过则无法再恢复到系统默认的图片
-@property(nonnull, nonatomic, strong) UIImage *qmui_clearButtonImage UI_APPEARANCE_SELECTOR;
+/// 自定义 clearButton 的图片，设置成nil，恢复到系统默认的图片
+@property(nullable, nonatomic, strong) UIImage *qmui_clearButtonImage UI_APPEARANCE_SELECTOR;
 
 @end
diff --git a/QMUIKit/UIKitExtensions/UITextField+QMUI.m b/QMUIKit/UIKitExtensions/UITextField+QMUI.m
@@ -30,10 +30,8 @@ - (UIButton *)qmui_clearButton {
 
 static char kAssociatedObjectKey_clearButtonImage;
 - (void)setQmui_clearButtonImage:(UIImage *)qmui_clearButtonImage {
-    if (qmui_clearButtonImage) {
-        objc_setAssociatedObject(self, &kAssociatedObjectKey_clearButtonImage, qmui_clearButtonImage, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
-        [self.qmui_clearButton setImage:qmui_clearButtonImage forState:UIControlStateNormal];
-    }
+    objc_setAssociatedObject(self, &kAssociatedObjectKey_clearButtonImage, qmui_clearButtonImage, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
+    [self.qmui_clearButton setImage:qmui_clearButtonImage forState:UIControlStateNormal];
 }
 
 - (UIImage *)qmui_clearButtonImage {