如何在swift中将UIViews的背景色设置为渐变色？

import UIKit

extension UIView {
    func setGradientBackground(colors: [UIColor]) {
        let gradientLayer = CAGradientLayer()
        gradientLayer.frame = bounds
        gradientLayer.colors = colors.map { $0.cgColor }
        gradientLayer.startPoint = CGPoint(x: 0.0, y: 0.0)
        gradientLayer.endPoint = CGPoint(x: 1.0, y: 1.0)
        
        layer.insertSublayer(gradientLayer, at: 0)
    }
}

// 使用示例
let view = UIView(frame: CGRect(x: 0, y: 0, width: 200, height: 200))
view.setGradientBackground(colors: [UIColor.red, UIColor.blue])