如何避免在Swift中的UIViewController中再次编写相同的代码？

Question

我在每个UIViewController中都有以下代码段：

var navBar = self.navigationController?.navigationBar
    navBar?.titleTextAttributes = [NSAttributedString.Key.foregroundColor: UIColor.white]

我不想在我创建的每个UIViewController中编写这些代码。我很懒，我不想重复自己。我无法将其添加为扩展名。那么创建UIViewController时不必写这段代码怎么办？

Answer 1

在这种情况下，您可以使用默认的实现协议。

创建默认的实现协议：

import Foundation
import UIKit

protocol TitleSetupable: AnyObject {
    func setupTitle()
}

extension TitleSetupable where Self: UIViewController {
    func setupTitle() {
        var navBar = self.navigationController?.navigationBar
        navBar?.titleTextAttributes = [NSAttributedString.Key.foregroundColor: UIColor.white]
    }
}

在ViewController中使用：

class YourViewController: UIViewController, TitleSetupable {
    override func viewDidLoad() {
        super.viewDidLoad()
        setupTitle()
    }
}

Answer 2

除协议之外的另一种方法是创建一个从UIViewController继承的类，在其上以及在所有其他viewControllers上实现setUpTitle，而不是从UIViewController继承，而是使用您创建的该超类：

class TitleSetupController: UIViewController {
    func setupTitle() {
        var navBar = self.navigationController?.navigationBar
        navBar?.titleTextAttributes = [NSAttributedString.Key.foregroundColor: UIColor.white]
    }
}

class ViewController1: TitleSetupController {
    override func viewDidLoad() {
        setUpTitle()
    }
}

class ViewController2: TitleSetupController {
    override func viewDidLoad() {
        setUpTitle()
    }
}

Answer 3

1。使用基本的UIViewController（例如BaseViewController）并将此代码放入viewDidLoad中，并用BaseViewController替换UIViewController。

2。您可能不想对UIViewController做任何事情。然后，您可能会遇到术语AOP。例如Java经常使用AOP。

您可以使用像Aspects这样的AOP框架。