如何在Swift中绘制一条从一点到下一点的线

import UIKit

class LineView: UIView {
    override func draw(_ rect: CGRect) {
        guard let context = UIGraphicsGetCurrentContext() else { return }
        
        context.setStrokeColor(UIColor.black.cgColor)
        context.setLineWidth(2)
        
        let startPoint = CGPoint(x: 50, y: 50)
        let endPoint = CGPoint(x: 200, y: 200)
        
        context.move(to: startPoint)
        context.addLine(to: endPoint)
        
        context.strokePath()
    }
}

// 在ViewController中使用LineView来绘制线
class ViewController: UIViewController {
    override func viewDidLoad() {
        super.viewDidLoad()
        
        let lineView = LineView(frame: CGRect(x: 0, y: 0, width: 300, height: 300))
        lineView.backgroundColor = UIColor.white
        view.addSubview(lineView)
    }
}