项目需求
公司后端API调用，经历多次302重定向，初始请求需在请求头中携带Referer，调用成功后，可进行数据下载。
实际问题
使用NSURLSession进行网络下载请求，跳转两次后cookie丢失及请求头中Referer消失，导致请求失败返回认证失败网页。
解决思路
拦截跳转请求，获取下次跳转的URL，强制添加消失的cookie及请求头的Referer，再进行跳转。
实现方法
- 项目文档中注册代理类（我的项目为DownloadManager，请自行替换）
@interface DownloadManager()<NSCopying, NSURLSessionDelegate>
- 继承代理方法
- (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task
willPerformHTTPRedirection:(NSHTTPURLResponse *)response
newRequest:(NSURLRequest *)request
completionHandler:(void (^)(NSURLRequest * _Nullable))completionHandler {
//此函数为不进行任何操作直接用返回的request进行下一步跳转，
//若使用completionHandler(nil)，则不会进行跳转
completionHandler(request);
//返回的task具有唯一性，可用来识别是哪个发送的请求：task.taskIdentifier
//返回的response中可以获取到服务器返回的Set-Cookie
//返回的request中可以获取到即将跳转的链接地址
}
- 实际代码
以下为项目需要进行的操作，实际可以根据自己需求进行更改，定义了一个SessionModel来存储对应的task.taskIdentifier及相应的Referer等信息，我实际遇到的问题为跳转请求的cookie中LID的value丢失，所以代码内进行了LID的赋值
- (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task
willPerformHTTPRedirection:(NSHTTPURLResponse *)response
newRequest:(NSURLRequest *)request
completionHandler:(void (^)(NSURLRequest * _Nullable))completionHandler {
NSHTTPCookieStorage *cookieJar = [NSHTTPCookieStorage sharedHTTPCookieStorage];
for (NSHTTPCookie *cookie in [cookieJar cookies]) {
NSLog(@"cookie%@", cookie);
//手动设置LID (实际需求可以和后端开发协商
if([cookie.name isEqualToString:@"LID"]) {
NSMutableDictionary *cookieProperties = [NSMutableDictionary dictionary]; // 创建cookie属性字典
[cookieProperties setObject:cookie.name forKey:NSHTTPCookieName]; // 手动设置cookie的name
[cookieProperties setObject:@"LID" forKey:NSHTTPCookieValue];//手动设置cookie的value
[cookieProperties setObject:@"/" forKey:NSHTTPCookiePath];
[cookieProperties setObject:@"HOST" forKey:NSHTTPCookieDomain];//设置cookie的host地址
NSHTTPCookie *cookie = [NSHTTPCookie cookieWithProperties:cookieProperties];
[[NSHTTPCookieStorage sharedHTTPCookieStorage] setCookie:cookie];//保存cookie
}
}
//用返回的request.URL初始化请求，因为公司下载服务不稳定，故设置超时时间为100s
//NSURLRequestReloadRevalidatingCacheData：验证本地数据与远程数据是否相同，如果不同则下载远程数据，否则使用本地数据
NSMutableURLRequest *nextRequest = [NSMutableURLRequest requestWithURL:request.URL
cachePolicy:NSURLRequestReloadRevalidatingCacheData
timeoutInterval:100];
//getSessionModel:为自定义方法,并设置Referer
SessionModel *sessionModel = [self getSessionModel:task.taskIdentifier];
[nextRequest setValue:sessionModel.refer forHTTPHeaderField:@"Referer"];
//发送下个请求
completionHandler(nextRequest);
}
