iOS Swift屏幕旋转问题：切换控制器后内容未重绘的标准解法

1. 先正确限制目标控制器的屏幕方向 #

首先要确保第二个控制器仅支持竖屏，iOS 16前后的API有差异，需要分别处理：

class PortraitOnlyVC: UIViewController {
    override func viewDidLoad() {
        super.viewDidLoad()
        // iOS 16+ 强制系统更新方向限制
        if #available(iOS 16.0, *) {
            setNeedsUpdateOfSupportedInterfaceOrientations()
        } else {
            // iOS 15及以下，强制切换到竖屏（避免初始横屏状态）
            UIDevice.current.setValue(UIInterfaceOrientation.portrait.rawValue, forKey: "orientation")
        }
    }

    // 全局限制仅支持竖屏
    override var supportedInterfaceOrientations: UIInterfaceOrientationMask {
        .portrait
    }

    override var shouldAutorotate: Bool {
        false
    }

    // iOS 16+ 方向更新回调，可在这里触发UI调整
    @available(iOS 16.0, *)
    override func setNeedsUpdateOfSupportedInterfaceOrientations() {
        super.setNeedsUpdateOfSupportedInterfaceOrientations()
        adjustUIForOrientation()
    }
}

2. 用可靠的时机触发UI重绘 #

viewWillTransition仅在设备主动旋转时触发，控制器切换时的被动方向切换不会调用它。**viewWillLayoutSubviews**是更可靠的选择——它会在视图首次布局、方向变化、父视图布局改变时自动调用，适合用来处理UI调整：

extension PortraitOnlyVC {
    private func adjustUIForOrientation() {
        // 触发布局更新，适配Auto Layout
        view.setNeedsLayout()
        view.layoutIfNeeded()
        
        // 如果是自定义绘制的视图（比如重写了draw(_:)），直接触发重绘
        // customDrawView.setNeedsDisplay()
    }

    override func viewWillLayoutSubviews() {
        super.viewWillLayoutSubviews()
        adjustUIForOrientation()
    }
}

3. 自定义视图的正确重绘姿势 #

如果问题出在自定义绘制的视图上，不要依赖外部调用setNeedsDisplay()，应该在视图内部的layoutSubviews里处理：

class CustomDrawView: UIView {
    override func layoutSubviews() {
        super.layoutSubviews()
        // 根据当前bounds更新绘制所需的参数（比如路径、字体大小）
        updateDrawParams()
        // 触发重绘
        setNeedsDisplay()
    }

    private func updateDrawParams() {
        // 示例：根据视图宽高调整绘制内容
        // drawPath = UIBezierPath(rect: self.bounds)
    }

    override func draw(_ rect: CGRect) {
        // 使用更新后的参数执行绘制逻辑
    }
}

关键说明 #

• 避免依赖viewDidLoad里的一次性UI初始化，因为此时控制器可能还处于横屏状态，布局参数不准确；
• 优先用Auto Layout而非手动设置frame，这样layoutIfNeeded()就能自动适配方向变化；
• 不要滥用UIDevice.current.setValue(...)，iOS 16+完全可以通过setNeedsUpdateOfSupportedInterfaceOrientations()实现方向强制切换。

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