局部初始化交叉引用对象时，如何保留weak引用的懒加载实例？

首先看你的代码结构：

class Generator {
    weak var machine: Machine!
    
    public init(for machine: Machine) {
        self.machine = machine
    }
    
    func start() {
        // ... 启动生成器的代码
    }
    
    func generate() {
        machine.didGenerate(value: 1)
    }
}


class Machine {
    public lazy var generator: Generator = Generator(for: self)
    
    func start() {
        generator.start()
    }
    
    func didGenerate(value: Float) {
        // 处理生成的值
    }
}

问题现象 #

• 当machine是全局/类成员变量时，代码能正常运行：外部的强引用会持续持有machine实例，Generator的weak引用始终有效。
• 但把machine放在viewDidLoad内部作为局部变量时，Generator里的machine引用会被提前释放，调用generate()时会出现空引用问题；换成unowned也会直接崩溃，因为unowned要求对象必须始终存在，而局部变量在方法执行完毕后就会被ARC销毁。

核心原因 #

局部变量的生命周期仅限于所在方法的执行期间，viewDidLoad执行完成后，machine没有任何强引用持有，会被立即释放。而Generator使用weak/unowned引用machine，无法阻止这个销毁过程。

解决方案 #

解决的核心逻辑是：在需要使用machine的整个生命周期内，确保至少有一个强引用持有它，以下是几种可行的实现方式：

方式1：用类成员变量长期持有 #

将局部的machine赋值给当前类的成员变量，通过成员变量的强引用确保machine不会被提前释放：

class ViewController: UIViewController {
    // 成员变量作为强引用容器
    private var retainedMachine: Machine?
    
    override func viewDidLoad() {
        super.viewDidLoad()
        let machine = Machine()
        retainedMachine = machine
        machine.start()
    }
    
    // 不需要时手动释放，避免内存泄漏
    func cleanup() {
        retainedMachine = nil
    }
}

方式2：临时强引用（适用于异步操作场景） #

如果start()或generate()包含异步逻辑（比如后台线程任务、定时器），可以在Generator内部添加临时强引用，在异步操作期间持有machine，操作完成后再释放，既避免循环引用，又能保证对象存活：

class Generator {
    weak var machine: Machine!
    // 临时强引用，仅在需要时持有
    private var temporaryMachineHolder: Machine?
    
    public init(for machine: Machine) {
        self.machine = machine
    }
    
    func start() {
        // 启动异步操作前，临时持有强引用
        temporaryMachineHolder = machine
        DispatchQueue.global().async {
            // 操作完成后释放临时引用，打破强引用链
            defer { self.temporaryMachineHolder = nil }
            self.generate()
        }
    }
    
    func generate() {
        // 用可选链避免空引用崩溃
        machine?.didGenerate(value: 1)
    }
}

方式3：同步操作场景的优化 #

如果start()和generate()都是同步执行，其实Swift的ARC会自动在方法执行期间保留局部变量machine，但如果代码中存在隐式的异步调用（比如UIKit的延迟回调），还是需要手动捕获强引用：

func viewDidLoad() {
    super.viewDidLoad()
    let machine = Machine()
    machine.start()
    
    // 如果有异步回调，闭包捕获machine的强引用
    DispatchQueue.main.asyncAfter(deadline: .now() + 2) {
        machine.generator.generate()
    }
}

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