NSArray的Category

NSArray的Category

前言

项目中自己通过各种渠道及结合项目的经验整理了一套自己的工具包，里面有各种Category，及封装的方法，方便项目使用，今天先分享一下NSarray的Category。

NSArray+LSCore.h

#import <Foundation/Foundation.h>

@interface NSArray (LSCore)

#pragma mark - 取值
- (id)firstObject;
- (id)randomObject;

#pragma mark - 排序
- (NSArray *)shuffledArray;
- (NSArray *)reversedArray;
- (NSArray *)uniqueArray;
- (NSArray *)arraySorting:(NSString *)parameters  ascending:(BOOL)ascending;

#pragma mark - 安全操作
-(id)objectWithIndex:(NSUInteger)index;

- (NSString*)stringWithIndex:(NSUInteger)index;
- (NSNumber*)numberWithIndex:(NSUInteger)index;
- (NSArray*)arrayWithIndex:(NSUInteger)index;
- (NSDictionary*)dictionaryWithIndex:(NSUInteger)index;
- (NSInteger)integerWithIndex:(NSUInteger)index;
- (NSUInteger)unsignedIntegerWithIndex:(NSUInteger)index;
- (BOOL)boolWithIndex:(NSUInteger)index;

- (int16_t)int16WithIndex:(NSUInteger)index;
- (int32_t)int32WithIndex:(NSUInteger)index;
- (int64_t)int64WithIndex:(NSUInteger)index;

- (char)charWithIndex:(NSUInteger)index;
- (short)shortWithIndex:(NSUInteger)index;
- (float)floatWithIndex:(NSUInteger)index;
- (double)doubleWithIndex:(NSUInteger)index;

- (CGFloat)CGFloatWithIndex:(NSUInteger)index;
- (CGPoint)pointWithIndex:(NSUInteger)index;
- (CGSize)sizeWithIndex:(NSUInteger)index;
- (CGRect)rectWithIndex:(NSUInteger)index;

@end

NSArray+LSCore.m

#import "NSArray+LSCore.h"

@implementation NSArray (LSCore)
#pragma mark - 取值
/**
 *  获取数组的第一个元素
 *
 *  @return 取得的元素
 */
- (id)firstObject {
    if ([self count] > 0) {
        return [self objectAtIndex:0];
    } else {
        return nil;
    }
}

/**
 *  随机取数组里的一个元素
 *
 *  @return 随机取得的元素
 */
- (id)randomObject {
    if ([self count] > 0) {
        int index = arc4random() % [self count];
        id object = [self objectAtIndex:index];
        return object;
    } else {
        return nil;
    }
}

#pragma mark - 排序
/**
 *  重组数组(打乱顺序)
 *
 *  @return 重组之后的数组
 */
- (NSArray *)shuffledArray {
    NSMutableArray *array = [NSMutableArray arrayWithCapacity:[self count]];
    NSMutableArray *copy = [self mutableCopy];
    while ([copy count] > 0)
    {
        int index = arc4random() % [copy count];
        id objectToMove = [copy objectAtIndex:index];
        [array addObject:objectToMove];
        [copy removeObjectAtIndex:index];
    }
    return array;
}

/**
 *  数组倒序
 *
 *  @return 将数组倒序后得到的数组
 */
- (NSArray *)reversedArray {
    NSArray *reversedArray = [[self reverseObjectEnumerator] allObjects];
    return reversedArray;
}

/**
 *  数组去除相同的元素，并获得新的array
 *
 *  @return 去除相同元素后的array
 */
- (NSArray *)uniqueArray {
    NSSet *set = [NSSet setWithArray:self];
    NSArray *array = [[NSArray alloc] initWithArray:[set allObjects]];
    return array;
}

/**
 *  根据关键词 对array的内容进行排序，并返回排序后的array
 *
 *  @param parameters 关键词
 *  @param ascending  YES 升序 NO 降序
 *
 *  @return 经过排序后的数组
 */
- (NSArray *)arraySorting:(NSString *)parameters  ascending:(BOOL)ascending {
    NSSortDescriptor*sorter=[[NSSortDescriptor alloc]initWithKey:parameters ascending:ascending];
    NSMutableArray *sortDescriptors=[[NSMutableArray alloc]initWithObjects:&sorter count:1];
    NSArray *sortArray=[self sortedArrayUsingDescriptors:sortDescriptors];
    return sortArray;
}

#pragma mark - 安全操作
/**
 *  获取下标为index的对象
 *
 *  @param index 下标
 *
 *  @return 获得的对象
 */
-(id)objectWithIndex:(NSUInteger)index {
    if (index <self.count) {
        return self[index];
    }else{
        return nil;
    }
}

/**
 *  获取下标为index的字符串，不存在返回空，能转为string的转，不能的返回nil
 *
 *  @param index 下标
 *
 *  @return 获得的字符串
 */
- (NSString*)stringWithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];
    if (value == nil || value == [NSNull null])
    {
        return @"";
    }
    if ([value isKindOfClass:[NSString class]]) {
        return (NSString*)value;
    }
    if ([value isKindOfClass:[NSNumber class]]) {
        return [value stringValue];
    }

    return nil;
}

/**
 *  获取下标为index的number，不存在返回空，能转为number的转，不能的返回nil
 *
 *  @param index 下标
 *
 *  @return 获得的number
 */
- (NSNumber*)numberWithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];
    if ([value isKindOfClass:[NSNumber class]]) {
        return (NSNumber*)value;
    }
    if ([value isKindOfClass:[NSString class]]) {
        NSNumberFormatter * f = [[NSNumberFormatter alloc] init];
        [f setNumberStyle:NSNumberFormatterDecimalStyle];
        return [f numberFromString:(NSString*)value];
    }
    return nil;
}

/**
 *  获取下标为index的array，不存返回nil
 *
 *  @param index 下标
 *
 *  @return 获得的array
 */
- (NSArray*)arrayWithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];
    if (value == nil || value == [NSNull null])
    {
        return nil;
    }
    if ([value isKindOfClass:[NSArray class]])
    {
        return value;
    }
    return nil;
}

/**
 *  获取下标为index的Dictionary，不存返回nil
 *
 *  @param index 下标
 *
 *  @return 获得的Dictionary
 */
- (NSDictionary*)dictionaryWithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];
    if (value == nil || value == [NSNull null])
    {
        return nil;
    }
    if ([value isKindOfClass:[NSDictionary class]])
    {
        return value;
    }
    return nil;
}

/**
 *  获取下标为index的integer，不存返回0
 *
 *  @param index 下标
 *
 *  @return 获得的integer
 */
- (NSInteger)integerWithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];
    if (value == nil || value == [NSNull null])
    {
        return 0;
    }
    if ([value isKindOfClass:[NSString class]] || [value isKindOfClass:[NSNumber class]])
    {
        return [value integerValue];
    }
    return 0;
}

/**
 *  获取下标为index的unsignedInteger，不存返回0
 *
 *  @param index 下标
 *
 *  @return 获得的unsignedInteger
 */
- (NSUInteger)unsignedIntegerWithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];
    if (value == nil || value == [NSNull null])
    {
        return 0;
    }
    if ([value isKindOfClass:[NSString class]] || [value isKindOfClass:[NSNumber class]])
    {
        return [value unsignedIntegerValue];
    }
    return 0;
}

/**
 *  获取下标为index的bool，不存返回NO
 *
 *  @param index 下标
 *
 *  @return 获得的bool
 */
- (BOOL)boolWithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];

    if (value == nil || value == [NSNull null])
    {
        return NO;
    }
    if ([value isKindOfClass:[NSNumber class]])
    {
        return [value boolValue];
    }
    if ([value isKindOfClass:[NSString class]])
    {
        return [value boolValue];
    }
    return NO;
}

/**
 *  获取下标为index的int16，不存返回0
 *
 *  @param index 下标
 *
 *  @return 获得的int16
 */
- (int16_t)int16WithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];

    if (value == nil || value == [NSNull null])
    {
        return 0;
    }
    if ([value isKindOfClass:[NSNumber class]])
    {
        return [value shortValue];
    }
    if ([value isKindOfClass:[NSString class]])
    {
        return [value intValue];
    }
    return 0;
}

/**
 *  获取下标为index的int32，不存返回0
 *
 *  @param index 下标
 *
 *  @return 获得的int32
 */
- (int32_t)int32WithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];

    if (value == nil || value == [NSNull null])
    {
        return 0;
    }
    if ([value isKindOfClass:[NSNumber class]] || [value isKindOfClass:[NSString class]])
    {
        return [value intValue];
    }
    return 0;
}

/**
 *  获取下标为index的int64，不存返回0
 *
 *  @param index 下标
 *
 *  @return 获得的int64
 */
- (int64_t)int64WithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];

    if (value == nil || value == [NSNull null])
    {
        return 0;
    }
    if ([value isKindOfClass:[NSNumber class]] || [value isKindOfClass:[NSString class]])
    {
        return [value longLongValue];
    }
    return 0;
}

/**
 *  获取下标为index的charValue，不存返回0
 *
 *  @param index 下标
 *
 *  @return 获得的charValue
 */
- (char)charWithIndex:(NSUInteger)index {

    id value = [self objectWithIndex:index];

    if (value == nil || value == [NSNull null])
    {
        return 0;
    }
    if ([value isKindOfClass:[NSNumber class]] || [value isKindOfClass:[NSString class]])
    {
        return [value charValue];
    }
    return 0;
}

/**
 *  获取下标为index的intValue，不存返回0
 *
 *  @param index 下标
 *
 *  @return 获得的intValue
 */
- (short)shortWithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];

    if (value == nil || value == [NSNull null])
    {
        return 0;
    }
    if ([value isKindOfClass:[NSNumber class]])
    {
        return [value shortValue];
    }
    if ([value isKindOfClass:[NSString class]])
    {
        return [value intValue];
    }
    return 0;
}

/**
 *  获取下标为index的floatValue，不存返回0
 *
 *  @param index 下标
 *
 *  @return 获得的floatValue
 */
- (float)floatWithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];

    if (value == nil || value == [NSNull null])
    {
        return 0;
    }
    if ([value isKindOfClass:[NSNumber class]] || [value isKindOfClass:[NSString class]])
    {
        return [value floatValue];
    }
    return 0;
}

/**
 *  获取下标为index的doubleValue，不存返回0
 *
 *  @param index 下标
 *
 *  @return 获得的doubleValue
 */
- (double)doubleWithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];

    if (value == nil || value == [NSNull null])
    {
        return 0;
    }
    if ([value isKindOfClass:[NSNumber class]] || [value isKindOfClass:[NSString class]])
    {
        return [value doubleValue];
    }
    return 0;
}

/**
 *  获取下标为index的CGFloat
 *
 *  @param index 下标
 *
 *  @return 获得的CGFloat
 */
- (CGFloat)CGFloatWithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];

    CGFloat f = [value doubleValue];

    return f;
}

/**
 *  获取下标为index的point
 *
 *  @param index 下标
 *
 *  @return 获得的point
 */
- (CGPoint)pointWithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];

    CGPoint point = CGPointFromString(value);

    return point;
}

/**
 *  获取下标为index的size
 *
 *  @param index 下标
 *
 *  @return 获得的size
 */
- (CGSize)sizeWithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];

    CGSize size = CGSizeFromString(value);

    return size;
}

/**
 *  获取下标为index的rect
 *
 *  @param index 下标
 *
 *  @return 获得的rect
 */
- (CGRect)rectWithIndex:(NSUInteger)index {
    id value = [self objectWithIndex:index];

    CGRect rect = CGRectFromString(value);

    return rect;
}

@end