黑科技 扩展 YTKNetWork 对text/html的支持

项目中用了 YTKNetWork,服务器返回的json 数据,但是content-Type 是text/html (吐槽就不说了,不能惯着他们),解决问题才是关键;

首先查找YTK接口:

YTKBaseRequest 类中提供:

-(YTKResponseSerializerType)responseSerializerType;

来看枚举: YTKResponseSerializerType 并没有html,怎么办? 找 AFHTTPSessionManager 在哪呗;

///  Response serializer type, which determines response serialization process and
///  the type of `responseObject`.
typedef NS_ENUM(NSInteger, YTKResponseSerializerType) {
    /// NSData type
    YTKResponseSerializerTypeHTTP,
    /// JSON object type
    YTKResponseSerializerTypeJSON,
    /// NSXMLParser type
    YTKResponseSerializerTypeXMLParser,
};

YTKNetworkAgent 类

在YTKNetworkAgent 发现有如下片段代码

@implementation YTKNetworkAgent {
    AFHTTPSessionManager *_manager;
    ...
    ...
}
- (instancetype)init {
    self = [super init];
    if (self) {
        _config = [YTKNetworkConfig sharedConfig];
        _manager = [AFHTTPSessionManager manager];
        _requestsRecord = [NSMutableDictionary dictionary];
        _processingQueue = dispatch_queue_create("com.yuantiku.networkagent.processing", DISPATCH_QUEUE_CONCURRENT);
        _allStatusCodes = [NSIndexSet indexSetWithIndexesInRange:NSMakeRange(100, 500)];
        pthread_mutex_init(&_lock, NULL);

        _manager.securityPolicy = _config.securityPolicy;
        _manager.responseSerializer = [AFHTTPResponseSerializer serializer];
        // Take over the status code validation
        _manager.responseSerializer.acceptableStatusCodes = _allStatusCodes;
        _manager.completionQueue = _processingQueue;
    }
    return self;
}

runtime 动态修改属性

找到了AFHTTPSessionManager 就好办了,自然会想到拦截 YTKNetworkAgent 的init 方法,动态修改content-type,实现如下:

//
//  YTKNetworkAgent+dt_Agent.m
//  grid
//
//  Created by detao on 16/12/2.
//  Copyright © 2016年 德稻全球创新网络(北京)有限公司. All rights reserved.
//

//==============================================================================
// function:扩展 YTKNetWork acceptableContentTypes,只是提供类型支持
//==============================================================================
#import "objc/runtime.h"
@implementation YTKNetworkAgent (dt_Agent)

+(void)load{
// 实现init 与 dt_init方法的交换
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        SEL org_Selector = @selector(init);
        SEL dt_Selector  = @selector(dt_init);
        
        Method org_method = class_getInstanceMethod([self class], org_Selector);
        Method dt_method  = class_getInstanceMethod([self class], dt_Selector);
        
        BOOL isAdd = class_addMethod(self, org_Selector, method_getImplementation(dt_method), method_getTypeEncoding(dt_method));
        if (isAdd) {
            class_replaceMethod(self, dt_Selector, method_getImplementation(org_method), method_getTypeEncoding(org_method));
        }else{
            method_exchangeImplementations(org_method, dt_method);
        }
        
    });
}

-(instancetype)dt_init{
    [self dt_init];
    [self dt_class_copyIvarList:[self class]];
    return self;
}


- (void)dt_class_copyIvarList:(Class)class {
    unsigned int count;
    Ivar *ivarList = class_copyIvarList(class, &count);
    for (int i = 0; i < count; i++) {
        Ivar ivar = ivarList[i];
        // 获取成员属性名
        NSString *name = [NSString stringWithUTF8String:ivar_getName(ivar)];
        //        NSString *value = object_getIvar(obj, ivar);
        if ([name isEqualToString:@"_manager"]) {
            AFHTTPSessionManager *dt_manager =  [AFHTTPSessionManager manager];
            dt_manager.responseSerializer.acceptableContentTypes = [NSSet setWithObjects:@"text/html",@"text/json",@"application/json",nil];
           //  修改成员变量的值
            object_setIvar(self, ivar,dt_manager);
        }
        
    }
    // 记得释放哦
    free(ivarList);
}
@end

有没有留意到,实现这个分类根本不需要 .h(有了强大的load方法) ,写一个 .m 文件直接拖进工程就OK了;

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 135,292评论 19 139
  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 174,914评论 25 709
  • 小时候 我们拉拉手 就是好朋友 一起去上学,一起走一走 岁月挥挥手 来到十字路口 挥一挥衣袖 再也不见你的忧愁 别...
    简儿的十年阅读 3,188评论 16 20
  • 有趣的灵魂经历万水千山终究还是会相遇。---题记 三个多月前我们如此陌生,你也许在山之角,...
    杏子心语阅读 5,241评论 0 1
  • 昨儿夜里又多梦了,梦里的游戏规则是这样的,一个种族会突变的世界,每当夜幕降临,人类就基因突变,变得凶残无比...
    zcmorlen阅读 3,180评论 0 2