如何根据条件为UIButton添加额外图片（显示第二个ImageView）？

嘿，我来给你分享几个iOS开发里实现这个需求的靠谱思路，都是实际项目中验证过的：

方案一：巧用UIButton内置属性快速实现 #

UIButton本身就支持左侧放主图，右侧挂载自定义视图的特性，我们可以把第二张图放在titleView里，通过条件控制它的显示隐藏，不用自定义控件就能搞定：

// 初始化Button和副图ImageView
let mainButton = UIButton(type: .custom)
let secondaryImageView = UIImageView(image: UIImage(named: "secondary_icon"))

// 把副图设为Button的titleView
mainButton.titleView = secondaryImageView
// 默认隐藏副图
secondaryImageView.isHidden = true

// 设置主图片
mainButton.setImage(UIImage(named: "main_icon"), for: .normal)

// 封装更新逻辑的方法
func updateButtonDisplay(showSecondaryImage: Bool) {
    secondaryImageView.isHidden = !showSecondaryImage
    // 调整主图和副图的间距，让布局更美观
    mainButton.imageEdgeInsets = showSecondaryImage ? UIEdgeInsets(top: 0, left: 0, bottom: 0, right: 8) : .zero
}

// 使用时直接调用方法传入条件
updateButtonDisplay(showSecondaryImage: yourCondition)

这个方案的优势是代码量少、上手快，适合需求简单的场景。

方案二：自定义UIButton子类，封装性更强 #

如果后续需要调整两张图的位置、大小，或者给副图加交互，自定义子类会更灵活，把所有逻辑封装在内部，外部调用更简洁：

class DualImageButton: UIButton {
    // 内部管理主、副两个ImageView
    private let mainImageView = UIImageView()
    private let secondaryImageView = UIImageView()
    
    override init(frame: CGRect) {
        super.init(frame: frame)
        setupSubviews()
    }
    
    required init?(coder: NSCoder) {
        super.init(coder: coder)
        setupSubviews()
    }
    
    private func setupSubviews() {
        // 添加子视图
        addSubview(mainImageView)
        addSubview(secondaryImageView)
        
        // 这里用SnapKit做自动布局，你也可以用系统原生AutoLayout
        mainImageView.snp.makeConstraints { make in
            make.centerY.equalToSuperview()
            make.left.equalToSuperview()
            make.size.equalTo(CGSize(width: 24, height: 24))
        }
        
        secondaryImageView.snp.makeConstraints { make in
            make.centerY.equalToSuperview()
            make.left.equalTo(mainImageView.snp.right).offset(8)
            make.size.equalTo(CGSize(width: 24, height: 24))
            make.right.equalToSuperview()
        }
        
        // 默认隐藏副图
        secondaryImageView.isHidden = true
    }
    
    // 对外暴露设置主图的方法
    func setMainImage(_ image: UIImage?, for state: UIControl.State = .normal) {
        mainImageView.image = image
    }
    
    // 对外暴露设置副图的方法
    func setSecondaryImage(_ image: UIImage?, for state: UIControl.State = .normal) {
        secondaryImageView.image = image
    }
    
    // 更新显示状态的方法
    func toggleSecondaryImage(_ show: Bool) {
        secondaryImageView.isHidden = !show
        // 触发自动布局更新，让Button自适应大小
        invalidateIntrinsicContentSize()
    }
}

// 使用示例
let dualButton = DualImageButton()
dualButton.setMainImage(UIImage(named: "main_icon"))
dualButton.setSecondaryImage(UIImage(named: "secondary_icon"))
dualButton.toggleSecondaryImage(yourCondition)

自定义子类的好处是解耦性好，后续需求变动时，只需要修改子类内部代码，不会影响外部调用逻辑。

方案三：用UIStackView组合Button和ImageView #

如果不想修改Button本身的结构，也可以把Button和副图ImageView放在UIStackView里，通过控制ImageView的显示隐藏来实现需求：

// 创建水平方向的StackView，设置间距
let stackView = UIStackView()
stackView.axis = .horizontal
stackView.spacing = 8

// 初始化Button和副图
let mainButton = UIButton(type: .custom)
mainButton.setImage(UIImage(named: "main_icon"), for: .normal)

let secondaryImageView = UIImageView(image: UIImage(named: "secondary_icon"))
secondaryImageView.isHidden = true

// 添加到StackView中
stackView.addArrangedSubview(mainButton)
stackView.addArrangedSubview(secondaryImageView)

// 更新显示的方法
func updateStackDisplay(showSecondary: Bool) {
    secondaryImageView.isHidden = !showSecondary
}

这个方案适合Button需要保留原有所有交互逻辑，副图只是纯展示的场景，StackView会自动帮你处理布局适配。

几个注意点 #

• 不管用哪种方案，都要确保自适应布局正常，避免在不同尺寸屏幕上出现布局错乱。
• 如果需要支持Button的不同状态（比如高亮、选中），记得同步更新对应状态下的图片。
• 尽量避免直接修改frame，优先用AutoLayout来保证布局的稳定性。

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