封装一个自己的网络请求库（一）

现今一个app中网络库必不可少，而且大部分一般都会选择稳定第三方网络库，比如大名顶顶的AFNetworking性能好，稳定，大家用了都说好。之前公司用的基于libcurl写的网络库，因为libcurl是c++编写，所以可以实现ios与andorid跨平台使用，现在公司基于AF。所以一直很少关注ios提供的网络api。所以今天就聊聊封装一个基于NSURLSession的轻量级网络库。

封装一个post请求

定义回调

// 网络请求数据返回回调
typedef void (^WBHttpRequestSuccess)(id response);
typedef void (^WBHttpRequestFailure)(NSString *error);

// 调用时，自定义参数参入，避免出现大量重载函数，WBHttpRequestOrder是一个model，包涵请求需要的url，parameters等参数，可扩展
typedef void (^WBHttpRequest)(WBHttpRequestOrder *request);

定义请求方法

//  WBNetWork.h

@interface WBNetWork : NSObject
+ (void)post:(WBHttpRequest)request success:(WBHttpRequestSuccess)success failure:(WBHttpRequestFailure)failure;
@end

//  WBNetWork.m

@implementation WBNetWork

// 从block中获取model
+ (id)setupRequest:(id)instance block:(WBHttpRequest)request {
    if (request) {
        request(instance);
    }
    return instance;
}

+ (void)post:(WBHttpRequest)request success:(WBHttpRequestSuccess)success failure:(WBHttpRequestFailure)failure {
    
    WBHttpRequestOrder *order = [[self class] setupRequest:[WBHttpRequestOrder new] block:request];
    NSURL *url = [NSURL URLWithString:order.url];
    id parameters = order.parameters;
    
    NSMutableURLRequest *mutableRequest = [NSMutableURLRequest requestWithURL:url];
    mutableRequest.HTTPMethod = @"POST";
    mutableRequest.HTTPBody = [NSJSONSerialization dataWithJSONObject:parameters options:NSJSONWritingPrettyPrinted error:NULL];
    [mutableRequest addValue:@"application/json" forHTTPHeaderField:@"Content-Type"];

    NSURLSession *session = [NSURLSession sharedSession];
    NSURLSessionTask *task = [session dataTaskWithRequest:mutableRequest
                                        completionHandler:^(NSData * _Nullable data, NSURLResponse * _Nullable response, NSError * _Nullable error) {
                                            if ( error ) {
                                                if ( failure ) failure([error description]);
                                                return ;
                                            }
                                            
                                            if ( success ) {
                                                NSString *reslut = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
                                                NSLog(@"reslut  >> %@", reslut);
                                                success(reslut);
                                            }
                                            
                                        }];
    [task resume];
}

@end

测试调用

[WBNetWork post:^(WBHttpRequestOrder *request) {
     request.url = @"http://www.jianshu.com/login";
     request.parameters = @{@"username":@"wans", @"password":@"123"};
 } success:^(id response) {
     NSLog(@" seccess >> %@", response);
 } failure:^(NSString *error) {
     NSLog(@" failure >> %@", error);
 }];

封装一个get请求

get请求与post请求类似，不同之处在于请求方法名不同和get请求没有body，parameters通过“&”拼接在url后面。

// 请求
url：http://www.jianshu.com/p/7c3256e26808
parameters：@{@"username":@"wans", @"password":@"123",@"info":@[@1,@"w",@"b"]}

// 转换结果
http://www.jianshu.com/p/7c3256e26808?username=wans&password=123&info[]=1&info[]=w&info[]=b

parameters转换成拼接字符串算法copy自AFNetWorking。

+ (void)get:(WBHttpRequest)request success:(WBHttpRequestSuccess)success failure:(WBHttpRequestFailure)failure {
    
    WBHttpRequestOrder *order = [self setupRequest:[WBHttpRequestOrder new] block:request];
    NSURL *url = [NSURL URLWithString:order.url];
    NSDictionary *parameters = order.parameters;
    
    NSMutableURLRequest *mutableRequest = [NSMutableURLRequest requestWithURL:url];
    mutableRequest.HTTPMethod = @"GET";
    NSString *queryString = AFQueryStringFromParameters(parameters);
    if (queryString && queryString.length > 0) {
        mutableRequest.URL = [NSURL URLWithString:[[mutableRequest.URL absoluteString] stringByAppendingFormat:mutableRequest.URL.query ? @"&%@" : @"?%@", queryString]];
        
        if ( parameters.count  > 0 ) {
            [mutableRequest addValue:@"application/x-www-form-urlencoded" forHTTPHeaderField:@"Content-Type"];
        }
    }
    NSURLSession *session = [NSURLSession sharedSession];
    NSURLSessionTask *task = [session dataTaskWithRequest:mutableRequest
                                        completionHandler:^(NSData * _Nullable data, NSURLResponse * _Nullable response, NSError * _Nullable error) {
                                            if ( error ) {
                                                if ( failure ) failure([error description]);
                                                return ;
                                            }
                                            
                                            if ( success ) {
                                                NSString *reslut = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
                                                NSLog(@"reslut  >> %@", reslut);
                                                success(reslut);
                                            }
                                            
                                        }];
    [task resume];
}

默认配置

定义WBNetWorkConfig勇于配置默认网络请求的默认信息，包括服务器地址，默认请求头，超时时常等。

//  WBNetWork.m

static WBNetWorkConfig *httpConfig;

+ (WBNetWorkConfig *)httpConfig {
    if ( !httpConfig ) {
        httpConfig = [WBNetWorkConfig new];
    }
    return httpConfig;
}

+ (void)setHttpConfig:(WBNetWorkConfig *)config {
    httpConfig = config;
}

程序启动时，初始化默认信息

//  AppDelegate.m

- (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions {
   
   [WBNetWork httpConfig].baseUrl = @"http://www.jianshu.com";
   [WBNetWork httpConfig].timeoutInterval = 10;
   [WBNetWork httpConfig].httpHeaderFields = [@{@"Accept":@"*/*;q=0.8",
                                               @"Accept-Encoding":@"gzip, deflate, sdch",
                                               @"Accept-Language":@"keep-alive",
                                               @"User-Agent":@"WBNetworking/1.0 (iPhone; iOS 10.3; Scale/2.00)",
                                               @"appChannel":@"App Store",
                                               } mutableCopy];
}

优化&&解耦

将get与post方法合并，然后再做解耦处理。定义WBHTTPRequestManager来完成请求封装与执行操作。WBNetWork只负责调用。

//  WBHTTPRequestManager.h

@interface WBHTTPRequestManager : NSObject

- (instancetype)initWithRequest:(WBHttpRequest)request method:(NSString *)method;

- (void)excuteTaskSuccess:(WBHttpRequestSuccess)success failure:(WBHttpRequestFailure)failure;

@end

//  WBHTTPRequestManager.m

- (instancetype)initWithRequest:(WBHttpRequest)request method:(NSString *)method {

    self = [super init];
    if ( self ) {
        
        WBHttpRequestOrder *order = [self setupRequest:[WBHttpRequestOrder new] block:request];
        self.URL = [NSURL URLWithString:order.url];
        self.parameters = order.parameters;
        
        self.method = method;
    }
    
    return self;
}

- (void)setRequest {

    NSMutableURLRequest *mutableRequest = [NSMutableURLRequest requestWithURL:self.URL];
    NSString *queryString = nil;
    if ( [self.method isEqualToString:kWBHttpRequestMethodGet] ) {
        queryString = AFQueryStringFromParameters(self.parameters);
        if (queryString && queryString.length > 0) {
            mutableRequest.URL = [NSURL URLWithString:[[mutableRequest.URL absoluteString] stringByAppendingFormat:mutableRequest.URL.query ? @"&%@" : @"?%@", queryString]];
            
            if ( self.parameters.count  > 0 ) {
                [mutableRequest addValue:@"application/x-www-form-urlencoded" forHTTPHeaderField:@"Content-Type"];
            }
        }
    }
    
    mutableRequest.HTTPMethod = self.method;
    
    // 设置http header
    NSMutableDictionary *fields = [WBNetWork httpConfig].httpHeaderFields;
    for (NSString *field in fields.allKeys) {
        [mutableRequest setValue:fields[field] forHTTPHeaderField:field];
    }
    
    // 设置超时时间
    mutableRequest.timeoutInterval = [WBNetWork httpConfig].timeoutInterval;
    
    self.mutableRequest = mutableRequest;
    
}

- (void)setBody {

    if ( [self.method isEqualToString:kWBHttpRequestMethodPost] ) {
        
        if (self.parameters) {
            if (![self.mutableRequest valueForHTTPHeaderField:@"Content-Type"]) {
                [self.mutableRequest addValue:@"application/json" forHTTPHeaderField:@"Content-Type"];
            }
            
            NSError *error;
            [self.mutableRequest setHTTPBody:[NSJSONSerialization dataWithJSONObject:self.parameters options:NSJSONWritingPrettyPrinted error:&error]];
            NSLog(@"json error: %@",[error description]);
        }
    }
    
}

- (void)setTask:(WBHttpRequestSuccess)success failure:(WBHttpRequestFailure)failure {

    NSURLSessionDataTask *task = [[NSURLSession sharedSession] dataTaskWithRequest:self.mutableRequest completionHandler:^(NSData * _Nullable data, NSURLResponse * _Nullable response, NSError * _Nullable error) {
        
        if ( error ) {
            if ( failure ) failure([error description]);
            return ;
        }
        
        if ( success ) {
            NSString *reslut = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
            success(reslut);
        }
        
    }];
    [task resume];
    
}
- (void)excuteTaskSuccess:(WBHttpRequestSuccess)success failure:(WBHttpRequestFailure)failure {

    [self setRequest];
    [self setBody];
    [self setTask:success failure:failure];
}

@end

//  WBNetWork.h

@interface WBNetWork : NSObject

+ (WBNetWorkConfig *)httpConfig;
+ (void)setHttpConfig:(WBNetWorkConfig *)config;

+ (void)post:(WBHttpRequest)request success:(WBHttpRequestSuccess)success failure:(WBHttpRequestFailure)failure;
+ (void)get:(WBHttpRequest)request success:(WBHttpRequestSuccess)success failure:(WBHttpRequestFailure)failure;

@end

//  WBNetWork.m

+ (void)post:(WBHttpRequest)request success:(WBHttpRequestSuccess)success failure:(WBHttpRequestFailure)failure {

    [[self self] request:request method:@"POST" success:success failure:failure];
}

+ (void)get:(WBHttpRequest)request success:(WBHttpRequestSuccess)success failure:(WBHttpRequestFailure)failure {
    
    [[self self] request:request method:@"GET" success:success failure:failure];
}

+ (void)request:(WBHttpRequest)request method:(NSString *)method success:(WBHttpRequestSuccess)success failure:(WBHttpRequestFailure)failure {
    
    WBHTTPRequestManager *requestManager = [[WBHTTPRequestManager alloc] initWithRequest:request method:method];
    [requestManager excuteTaskSuccess:success failure:failure];
}

以上，初步完成一个网络请求库的封装，demo，欢迎评论批评指正。

最后编辑于：2017.12.07 03:42:50

人面猴
序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
沈念sama阅读 212,657评论 6赞 492
死咒
序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
沈念sama阅读 90,662评论 3赞 385
救了他两次的神仙让他今天三更去死
文/潘晓璐我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
开封第一讲书人阅读 158,143评论 0赞 348
道士缉凶录：失踪的卖姜人
文/不坏的土叔我叫张陵，是天一观的道长。经常有香客问我，道长，这世上最难降的妖魔是什么？我笑而不...
开封第一讲书人阅读 56,732评论 1赞 284
港岛之恋（遗憾婚礼）
正文为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
茶点故事阅读 65,837评论 6赞 386
恶毒庶女顶嫁案：这布局不是一般人想出来的
文/花漫我一把揭开白布。她就那样静静地躺着，像睡着了一般。火红的嫁衣衬着肌肤如雪。梳的纹丝不乱的头发上，一...
开封第一讲书人阅读 50,036评论 1赞 291
城市分裂传说
那天，我揣着相机与录音，去河边找鬼。笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。我是一名探鬼主播，决...
沈念sama阅读 39,126评论 3赞 410
双鸳鸯连环套：你想象不到人心有多黑
文/苍兰香墨我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
开封第一讲书人阅读 37,868评论 0赞 268
万荣杀人案实录
序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
沈念sama阅读 44,315评论 1赞 303
护林员之死
正文独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋以下内容为张勋视角年9月15日...
茶点故事阅读 36,641评论 2赞 327
白月光启示录
正文我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
茶点故事阅读 38,773评论 1赞 341
活死人
序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
沈念sama阅读 34,470评论 4赞 333
日本核电站爆炸内幕
正文年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
茶点故事阅读 40,126评论 3赞 317
男人毒药：我在死后第九天来索命
文/蒙蒙一、第九天我趴在偏房一处隐蔽的房顶上张望。院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
开封第一讲书人阅读 30,859评论 0赞 21
一桩弑父案，背后竟有这般阴谋
文/苍兰香墨我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。一阵脚步声响...
开封第一讲书人阅读 32,095评论 1赞 267
情欲美人皮
我被黑心中介骗来泰国打工，没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。一个月前我还...
沈念sama阅读 46,584评论 2赞 362
代替公主和亲
正文我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
茶点故事阅读 43,676评论 2赞 351

封装一个自己的网络请求库（一）

封装一个post请求

封装一个get请求

默认配置

优化&&解耦

推荐阅读更多精彩内容