子类化UIButton后修改标题文本导致框架重置为原始尺寸的解决办法

嘿，这个问题我之前踩过坑！大概率是你的UIButton子类没处理好自定义尺寸的持久化，或者修改标题的函数不小心触发了系统默认的布局逻辑，导致尺寸和标题都打回原形了。下面给你几个针对性的解决思路和代码示例：

UIButton本身自带一套基于内容的布局逻辑，当你调用setTitle(_:for:)这类系统方法时，如果子类没有重写布局相关的方法，系统会自动根据新标题重新计算intrinsicContentSize，直接覆盖你之前设置的自定义宽高。另外，如果没把自定义尺寸存在实例属性里，布局刷新时就会丢失自定义状态。

1. 给子类加自定义尺寸的持久化存储 #

在你的UIButton子类里添加一个私有属性，专门用来保存用户设置的自定义宽高，这样不管怎么刷新布局，都能优先用这个值而不是系统计算的：

class CustomButton: UIButton {
    // 存储自定义尺寸，初始为nil时用系统默认值
    private var storedCustomSize: CGSize?
    
    // 自定义修改宽高的方法
    func applyCustomSize(width: CGFloat, height: CGFloat) {
        storedCustomSize = CGSize(width: width, height: height)
        // 通知系统重新计算 intrinsicContentSize
        invalidateIntrinsicContentSize()
        // 触发布局更新
        setNeedsLayout()
    }
    
    // 重写 intrinsicContentSize，优先返回自定义尺寸
    override var intrinsicContentSize: CGSize {
        return storedCustomSize ?? super.intrinsicContentSize
    }
}

2. 精简标题修改函数，只做标题更新 #

你的标题修改函数要避免触发任何会重置尺寸的逻辑，比如别调用sizeToFit()或者其他会让系统重新计算布局的方法，只专注于设置标题：

extension CustomButton {
    func updateButtonTitle(to newTitle: String, for state: UIControl.State = .normal) {
        // 仅设置标题，不触碰尺寸相关代码
        setTitle(newTitle, for: state)
        // 只刷新标题显示，不影响布局
        setNeedsDisplay()
    }
}

3. 排查AutoLayout或Frame布局的冲突 #

• 如果用AutoLayout：确保给按钮加的是固定宽高约束，或者约束是基于你重写后的intrinsicContentSize，这样系统就不会因为标题变化而修改尺寸。
• 如果用Frame布局：修改标题后绝对不要调用sizeToFit()，这个方法会根据标题内容重新计算Frame，直接覆盖你自定义的尺寸。

4. 检查是否不小心重置了原始状态 #

看看你的标题修改函数里有没有调用父类的某些方法，或者不小心重置了按钮的其他状态（比如把标题又设回了原始值），这种低级错误也容易导致问题。

可以在重写的intrinsicContentSize里加个打印，看看每次布局时返回的是什么值，这样就能快速排查自定义尺寸有没有被正确读取：

override var intrinsicContentSize: CGSize {
    let size = storedCustomSize ?? super.intrinsicContentSize
    print("当前返回的尺寸：\(size)")
    return size
}

这样调整后，再调用标题修改函数应该就不会出现尺寸和标题回滚的问题了！

内容的提问来源于stack exchange，提问作者PruitIgoe