如何在swift中制作带有角点和渐变颜色的按钮？

import UIKit

class GradientButton: UIButton {
    override init(frame: CGRect) {
        super.init(frame: frame)
        setupButton()
    }
    
    required init?(coder aDecoder: NSCoder) {
        super.init(coder: aDecoder)
        setupButton()
    }
    
    private func setupButton() {
        // 设置按钮标题等属性
        setTitle("按钮", for: .normal)
        setTitleColor(.white, for: .normal)
        titleLabel?.font = UIFont.systemFont(ofSize: 16)
        
        // 设置按钮圆角效果
        layer.cornerRadius = 10
        layer.masksToBounds = true
        
        // 创建渐变层
        let gradientLayer = CAGradientLayer()
        gradientLayer.frame = bounds
        gradientLayer.colors = [UIColor.red.cgColor, UIColor.blue.cgColor] // 设置渐变颜色数组
        gradientLayer.startPoint = CGPoint(x: 0, y: 0)
        gradientLayer.endPoint = CGPoint(x: 1, y: 1)
        
        // 将渐变层添加到按钮的layer中
        layer.insertSublayer(gradientLayer, at: 0)
    }
}

// 在视图控制器中使用GradientButton
class ViewController: UIViewController {
    override func viewDidLoad() {
        super.viewDidLoad()
        
        let button = GradientButton(frame: CGRect(x: 100, y: 100, width: 200, height: 50))
        view.addSubview(button)
    }
}