如何添加其他用户建议的代码段？含代码优化相关疑问

嘿，我来帮你把这些优化建议一步步加到代码里，分两部分来拆解，保证清晰易懂：

要让指定的文本框成为第一响应者（自动获取焦点或弹出键盘），你只需要在合适的时机调用 becomeFirstResponder() 方法就行，比如按钮点击事件、视图加载完成后：

场景1：按钮点击时激活文本框 #

@IBAction func triggerTextField(_ sender: UIButton) {
    // 确保你已经把文本框通过IBOutlet连接到代码里，比如命名为targetTextField
    targetTextField.becomeFirstResponder()
    // 这里可以接着写其他逻辑，比如弹出选择器等
}

场景2：视图加载完成后自动激活文本框 #

override func viewDidAppear(_ animated: Bool) {
    super.viewDidAppear(animated)
    targetTextField.becomeFirstResponder()
}

针对你那些长长的 if/else if 链，有两种优雅的重构方式：

方式一：用Switch语句替换 #

如果原来的逻辑是通过按钮的tag、标题或者其他标识来分支，直接换成switch会更清晰：

原代码（模拟你的冗长结构） #

@IBAction func buttonTapped(_ sender: UIButton) {
    if sender.tag == 1 {
        // 按钮1对应的数据源加载、选择器显示逻辑
    } else if sender.tag == 2 {
        // 按钮2对应的逻辑
    } else if sender.tag == 3 {
        // 按钮3对应的逻辑
    }
    // 更多else if...
}

重构后的Switch版本 #

@IBAction func buttonTapped(_ sender: UIButton) {
    switch sender.tag {
    case 1:
        // 处理按钮1的逻辑，比如绑定数据源到选择器
        pickerView1.dataSource = self
        pickerView1.reloadAllComponents()
    case 2:
        // 处理按钮2的逻辑
        pickerView2.dataSource = self
        pickerView2.reloadAllComponents()
    case 3:
        // 处理按钮3的逻辑
        pickerView3.dataSource = self
        pickerView3.reloadAllComponents()
    default:
        break // 处理未匹配到的按钮，可选
    }
}

方式二：用字典绑定按钮与配置（更灵活） #

这种方式更适合按钮和对应配置（数据源、选择器）一一对应的场景，先定义一个结构体封装配置，再用字典关联按钮和配置：

第一步：定义配置结构体 #

// 封装每个按钮对应的数据源和选择器
struct ButtonRelatedConfig {
    let dataSource: [String] // 选择器需要的数据源数组
    let targetPicker: UIPickerView // 对应的选择器视图
}

第二步：初始化字典关联按钮与配置 #

在你的视图控制器里，先声明字典，然后在viewDidLoad里完成绑定：

// 声明字典，键是按钮，值是对应的配置
var buttonConfigMap: [UIButton: ButtonRelatedConfig] = [:]

override func viewDidLoad() {
    super.viewDidLoad()
    
    // 准备各个按钮对应的数据源
    let fruitData = ["苹果", "香蕉", "橙子"]
    let colorData = ["红色", "蓝色", "绿色"]
    let animalData = ["猫", "狗", "兔子"]
    
    // 把按钮和对应的配置绑定到字典里（假设你已经有button1、button2、button3等IBOutlet）
    buttonConfigMap[button1] = ButtonRelatedConfig(dataSource: fruitData, targetPicker: fruitPicker)
    buttonConfigMap[button2] = ButtonRelatedConfig(dataSource: colorData, targetPicker: colorPicker)
    buttonConfigMap[button3] = ButtonRelatedConfig(dataSource: animalData, targetPicker: animalPicker)
}

第三步：简化按钮点击事件 #

现在按钮点击逻辑就变得超级简洁了：

@IBAction func buttonTapped(_ sender: UIButton) {
    // 从字典里取出当前按钮对应的配置
    guard let config = buttonConfigMap[sender] else { return }
    
    // 直接使用配置里的数据源和选择器
    config.targetPicker.dataSource = self // 记得你的类要遵守UIPickerViewDataSource协议
    config.targetPicker.reloadAllComponents()
    
    // 如果你需要激活文本框，这里也可以加上
    targetTextField.becomeFirstResponder()
}

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