接口部分

@interface GradientButton : UIButton

@property (nonatomic,retain) CAGradientLayer *gradientLayer;

@end

實(shí)現(xiàn)部分

@interface GradientButton ()

@end

@implementation GradientButton

-(void)setFrame:(CGRect)frame
{
    [super setFrame:frame];
    self.gradientLayer.frame = self.bounds;
}

- (instancetype)initWithFrame:(CGRect)frame
{
    self = [super initWithFrame:frame];
    if (self) {
        self.gradientLayer.frame = self.bounds;
        self.layer.masksToBounds = YES;
        [self.layer addSublayer:self.gradientLayer];
    }
    return self;
}

-(CAGradientLayer *)gradientLayer
{
    if (_gradientLayer == nil) {
        _gradientLayer = [[CAGradientLayer alloc] init];
    }
    return _gradientLayer;
}


@end