|
|
|
//
|
|
|
|
// NSURLConnection+AutoTrack.m
|
|
|
|
// YHEventReport
|
|
|
|
//
|
|
|
|
// Created by 孙凯 on 2017/9/30.
|
|
|
|
// Copyright © 2017年 YOHO. All rights reserved.
|
|
|
|
//
|
|
|
|
|
|
|
|
#import "NSURLConnection+AutoTrack.h"
|
|
|
|
#import "YHSwizzle.h"
|
|
|
|
#import <objc/runtime.h>
|
|
|
|
#import <objc/message.h>
|
|
|
|
#import "YHEventReport.h"
|
|
|
|
#import "YHLog.h"
|
|
|
|
#import "YHSwizzle.h"
|
|
|
|
|
|
|
|
typedef void(^ActionHandlerBlock)(NSURLResponse* _Nullable response, NSData* _Nullable data, NSError* _Nullable connectionError);
|
|
|
|
|
|
|
|
static const void *yher_actionHandlerBlock = &yher_actionHandlerBlock;
|
|
|
|
|
|
|
|
@interface NSURLConnection()
|
|
|
|
|
|
|
|
@property(nullable,nonatomic,weak) id my_delegate;
|
|
|
|
|
|
|
|
@property (nonatomic, copy) ActionHandlerBlock actionHandlerBlock;
|
|
|
|
|
|
|
|
@end
|
|
|
|
|
|
|
|
@implementation NSURLConnection (AutoTrack)
|
|
|
|
|
|
|
|
#pragma mark - 属性
|
|
|
|
|
|
|
|
- (ActionHandlerBlock)actionHandlerBlock{
|
|
|
|
return objc_getAssociatedObject(self, @selector(actionHandlerBlock));
|
|
|
|
}
|
|
|
|
|
|
|
|
- (void)setActionHandlerBlock:(ActionHandlerBlock)actionHandlerBlock{
|
|
|
|
objc_setAssociatedObject(self, @selector(actionHandlerBlock), actionHandlerBlock, OBJC_ASSOCIATION_COPY_NONATOMIC);
|
|
|
|
}
|
|
|
|
|
|
|
|
- (void)setYh_viewId:(NSString *)yh_viewId{
|
|
|
|
objc_setAssociatedObject(self,@selector(yh_viewId),yh_viewId,OBJC_ASSOCIATION_RETAIN);
|
|
|
|
}
|
|
|
|
|
|
|
|
- (NSString *)yh_viewId{
|
|
|
|
return objc_getAssociatedObject(self, @selector(yh_viewId));
|
|
|
|
}
|
|
|
|
|
|
|
|
- (id)my_delegate
|
|
|
|
{
|
|
|
|
return objc_getAssociatedObject(self, _cmd);
|
|
|
|
}
|
|
|
|
|
|
|
|
- (void)setMy_delegate:(id)my_delegate
|
|
|
|
{
|
|
|
|
objc_setAssociatedObject(self, @selector(my_delegate), my_delegate, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
|
|
|
|
}
|
|
|
|
|
|
|
|
+(void)startTrack{
|
|
|
|
static dispatch_once_t onceToken;
|
|
|
|
dispatch_once(&onceToken, ^{
|
|
|
|
@try {
|
|
|
|
|
|
|
|
SEL selectors[] = {
|
|
|
|
@selector(initWithRequest:delegate:),
|
|
|
|
@selector(initWithRequest:delegate:startImmediately:),
|
|
|
|
@selector(start),
|
|
|
|
};
|
|
|
|
|
|
|
|
for (NSUInteger index = 0; index < sizeof(selectors) / sizeof(SEL); ++index) {
|
|
|
|
SEL originalSelector = selectors[index];
|
|
|
|
SEL swizzledSelector = NSSelectorFromString([@"yher_" stringByAppendingString:NSStringFromSelector(originalSelector)]);
|
|
|
|
NSError *error = NULL;
|
|
|
|
[[self class] yh_swizzleMethod:originalSelector
|
|
|
|
withMethod:swizzledSelector
|
|
|
|
error:&error];
|
|
|
|
if (error) {
|
|
|
|
YHLog(@"Failed to swizzle: on UIControl. Details: %@", error);
|
|
|
|
error = NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
SEL class_selectors[] = {
|
|
|
|
@selector(sendSynchronousRequest:returningResponse:error:),
|
|
|
|
@selector(sendAsynchronousRequest:queue:completionHandler:),
|
|
|
|
};
|
|
|
|
|
|
|
|
for (NSUInteger index = 0; index < sizeof(class_selectors) / sizeof(SEL); ++index) {
|
|
|
|
Class selfClass = object_getClass([self class]);
|
|
|
|
SEL oriSEL = class_selectors[index];
|
|
|
|
Method oriMethod = class_getClassMethod(selfClass, oriSEL);
|
|
|
|
SEL cusSEL = NSSelectorFromString([@"yher_" stringByAppendingString:NSStringFromSelector(oriSEL)]);
|
|
|
|
Method cusMethod = class_getClassMethod(selfClass, cusSEL);
|
|
|
|
|
|
|
|
BOOL addSucc = class_addMethod(selfClass, oriSEL, method_getImplementation(cusMethod), method_getTypeEncoding(cusMethod));
|
|
|
|
if (addSucc) {
|
|
|
|
class_replaceMethod(selfClass, cusSEL, method_getImplementation(oriMethod), method_getTypeEncoding(oriMethod));
|
|
|
|
}else {
|
|
|
|
method_exchangeImplementations(oriMethod, cusMethod);
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
} @catch (NSException *exception) {
|
|
|
|
YHLog(@"%@ error: %@", self, exception);
|
|
|
|
}
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
- (nullable instancetype)yher_initWithRequest:(NSURLRequest *)request delegate:(nullable id)delegate startImmediately:(BOOL)startImmediately
|
|
|
|
{
|
|
|
|
[self setMy_delegate:delegate];
|
|
|
|
return [self yher_initWithRequest:request delegate:self startImmediately:startImmediately];
|
|
|
|
}
|
|
|
|
|
|
|
|
- (nullable instancetype)yher_initWithRequest:(NSURLRequest *)request delegate:(nullable id)delegate
|
|
|
|
{
|
|
|
|
[self setMy_delegate:delegate];
|
|
|
|
return [self yher_initWithRequest:request delegate:self];
|
|
|
|
}
|
|
|
|
|
|
|
|
- (void)yher_start
|
|
|
|
{
|
|
|
|
[self yher_start];
|
|
|
|
}
|
|
|
|
|
|
|
|
+ (nullable NSData *)yher_sendSynchronousRequest:(NSURLRequest *)request returningResponse:(NSURLResponse * _Nullable * _Nullable)response error:(NSError **)error
|
|
|
|
{
|
|
|
|
NSData *data = [self yher_sendSynchronousRequest:request returningResponse:response error:error];
|
|
|
|
return data;
|
|
|
|
}
|
|
|
|
|
|
|
|
+ (void)yher_sendAsynchronousRequest:(NSURLRequest*) request
|
|
|
|
queue:(NSOperationQueue*) queue
|
|
|
|
completionHandler:(void (^)(NSURLResponse* _Nullable response, NSData* _Nullable data, NSError* _Nullable connectionError)) handler
|
|
|
|
{
|
|
|
|
objc_setAssociatedObject(self, yher_actionHandlerBlock, handler, OBJC_ASSOCIATION_COPY_NONATOMIC);
|
|
|
|
[self yher_sendAsynchronousRequest:request queue:queue completionHandler:^(NSURLResponse * _Nullable response, NSData * _Nullable data, NSError * _Nullable connectionError) {
|
|
|
|
ActionHandlerBlock actionHandlerBlock = objc_getAssociatedObject(self, yher_actionHandlerBlock);
|
|
|
|
if (actionHandlerBlock) {
|
|
|
|
actionHandlerBlock(response,data,connectionError);
|
|
|
|
}
|
|
|
|
}];
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
#pragma mark - NSURLConnectionDelegate
|
|
|
|
- (void) connection:(NSURLConnection *)connection didReceiveResponse:(NSURLResponse *)response {
|
|
|
|
if (self.my_delegate && [self.my_delegate respondsToSelector:@selector(connection:didReceiveResponse:)]) {
|
|
|
|
[self.my_delegate connection:connection didReceiveResponse:response];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
- (void) connection:(NSURLConnection *)connection didReceiveData:(NSData *)data {
|
|
|
|
if (self.my_delegate && [self.my_delegate respondsToSelector:@selector(connection:didReceiveData:)]) {
|
|
|
|
[self.my_delegate connection:connection didReceiveData:data];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
- (void) connectionDidFinishLoading:(NSURLConnection *)connection {
|
|
|
|
if (self.my_delegate && [self.my_delegate respondsToSelector:@selector(connectionDidFinishLoading:)]) {
|
|
|
|
[self.my_delegate connectionDidFinishLoading:connection];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
- (void)connection:(NSURLConnection *)connection didFailWithError:(NSError *)error {
|
|
|
|
if (self.my_delegate && [self.my_delegate respondsToSelector:@selector(connection:didFailWithError:)]) {
|
|
|
|
[self.my_delegate connection:connection didFailWithError:error];
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
@end |
...
|
...
|
|
