Flutter网络请求框架Dio源码分析以及封装--Cookie管理分析
- 前言
- 问题
- 如何使用
- CookieJar
- CookieManager
- PersistCookieJar
- 总结
前言
上一篇文章我们简单分析了一下Dio发出请求时的大致工作流程,这个只是Dio最基本的功能,而且我们还没有分析走到httpClientAdapter之后的内容。不过不用着急,这次我们先接着上一次的内容,看一下Dio当中Cookie管理的问题,因为之前在项目中碰到了这个问题,回过头来再从源码的角度去分析一下,算是复盘。
问题
之前碰到的问题是这样的:登录接口在登陆成功后会返回一个Cookie给客户端,包含在Response的
“set-cookie”Headers属性里。之后在客户端这边调用某些接口时后台需要对Cookie进行验证,所以需要把Cookie存到内存甚至硬盘里,在请求头里面带上传给后台,否则会后台会报请求失败。
由于以前的项目基本上是采取token的方式去做接口鉴权,所以这种cookie持久化的方法对我来说有点陌生,不过最后还是能想到利用拦截器去做这个事情。Dio允许我们自定义拦截器,对请求与返回的参数进行调整,我们在onResponse回调方法里,从Response里面的headers拿到Cookie,再在onRequest里面加到RequestOptions里。原理很简单,但是之前自己实现的时候很多细节没有处理好,造成了代码的冗余以及不够健壮。
实际上,Dio提供了配套的Plugin:dio_cookie_manager 来帮助我们进行Cookie的管理,使用起来也非常的方便。那么这次我们就一起来看一下它是怎么处理这些问题的。
如何使用
首先我们需要导入cookie_manager的库,基于项目中用到的2.0.0版本
它又间接依赖cookie_jar3.0.0这个版本
dependencies:dio_cookie_manager: ^2.0.0cookie_jar: ^3.0.0
final dio = Dio();final cookieJar = CookieJar();dio.interceptors.add(CookieManager(cookieJar));
如示例中所示,我们需要构造一个CookieJar的实例,然后将它传入CookieManager的构造方法中,最后添加到Dio的拦截器列表中。
CookieJar
我们先来看看CookieJar这个类:
/// CookieJar is a cookie manager for http requests。
abstract class CookieJar {factory CookieJar({bool ignoreExpires = false}) {return DefaultCookieJar(ignoreExpires: ignoreExpires);}/// Save the cookies for specified uri.Future<void> saveFromResponse(Uri uri, List<Cookie> cookies);/// Load the cookies for specified uri.Future<List<Cookie>> loadForRequest(Uri uri);Future<void> deleteAll();Future<void> delete(Uri uri, [bool withDomainSharedCookie = false]);final bool ignoreExpires = false;
}
CookieJar是抽象类,最终还是调用了DefaultCookieJar这个类的构造方法
/// [DefaultCookieJar] is a default cookie manager which implements the standard
/// cookie policy declared in RFC. [DefaultCookieJar] saves the cookies in RAM, so if the application
/// exit, all cookies will be cleared.
class DefaultCookieJar implements CookieJar {/// [ignoreExpires]: save/load even cookies that have expired.DefaultCookieJar({this.ignoreExpires = false});...}
DefaultCookieJar是CookieJar的一种默认实现,按照http的格式将cookie从Request与Response中解析出来,并保存在Ram中。
CookieManager
接着来看看CookieManager这个类:
/// Don't use this class in Browser environment
class CookieManager extends Interceptor {final CookieJar cookieJar;CookieManager(this.cookieJar);
```、
```dartvoid onRequest(RequestOptions options, RequestInterceptorHandler handler) {cookieJar.loadForRequest(options.uri).then((cookies) {var cookie = getCookies(cookies);if (cookie.isNotEmpty) {options.headers[HttpHeaders.cookieHeader] = cookie;}handler.next(options);}).catchError((e, stackTrace) {var err = DioError(requestOptions: options, error: e);err.stackTrace = stackTrace;handler.reject(err, true);});}
static String getCookies(List<Cookie> cookies) {return cookies.map((cookie) => '${cookie.name}=${cookie.value}').join('; ');}
构造的cookieJar传进来后,在onRequest方法里面调用loadForRequest方法获取格式化后的Cookie,将他们转化为请求需要的格式然后设置在请求头里,之后继续执行下面的拦截器的逻辑。
void onResponse(Response response, ResponseInterceptorHandler handler) {_saveCookies(response).then((_) => handler.next(response)).catchError((e, stackTrace) {var err = DioError(requestOptions: response.requestOptions, error: e);err.stackTrace = stackTrace;handler.reject(err, true);});}
Future<void> _saveCookies(Response response) async {var cookies = response.headers[HttpHeaders.setCookieHeader];if (cookies != null) {await cookieJar.saveFromResponse(response.requestOptions.uri,cookies.map((str) => Cookie.fromSetCookieValue(str)).toList(),);}}
onResponse也类似,拿到Cookie之后,调用saveFromResponse格式化保存起来,之后继续执行下面的拦截器的逻辑。
当然,正常的顺序应该是先在onResponse拿到Cookie之后再在onResponse使用。
PersistCookieJar
如果想对Cookie进行持久化处理,可以考虑使用PersistCookieJar:
final Directory appDocDir = await getApplicationDocumentsDirectory();final String appDocPath = appDocDir.path;final jar = PersistCookieJar(ignoreExpires: true,storage: FileStorage(appDocPath + "/.cookies/"),);dio!.interceptors.add(CookieManager(jar));
/// [PersistCookieJar] is a cookie manager which implements the standard
/// cookie policy declared in RFC. [PersistCookieJar] persists the cookies in files,
/// so if the application exit, the cookies always exist unless call [delete] explicitly.
class PersistCookieJar extends DefaultCookieJar {////// [persistSession]: Whether persisting the cookies that without/// "expires" or "max-age" attribute;/// If false, the session cookies will be discarded;/// otherwise, the session cookies will be persisted.////// [ignoreExpires]: save/load even cookies that have expired.////// [storage]: Defaults to FileStoragePersistCookieJar({this.persistSession = true,bool ignoreExpires = false,Storage? storage}): super(ignoreExpires: ignoreExpires) {this.storage = storage ?? FileStorage();}
为了实现持久化,引入了一个Storage类,默认实现是FileStorage,文件存储,也可以自己实现其他方式。
Future<List<Cookie>> loadForRequest(Uri uri) async {await _checkInitialized();await _load(uri);return super.loadForRequest(uri);}
PersistCookieJar最终也调用了DefaultCookieJar的loadForRequest方法,但是在那之前还执行了两个方法,我们一个一个看:
Future<void> _checkInitialized({bool force = false}) async {if (force || !_initialized) {await storage.init(persistSession, ignoreExpires);// Load domain cookiesvar str = await storage.read(DomainsKey);...}
执行了storage的init与read方法
Future<void> init(bool persistSession, bool ignoreExpires) async {_curDir = dir ?? './.cookies/';if (!_curDir!.endsWith('/')) {_curDir = _curDir! + '/';}_curDir = _curDir! + 'ie${ignoreExpires ? 1 : 0}_ps${persistSession ? 1 : 0}/';await _makeCookieDir();}
Future<void> _makeCookieDir() async {final directory = Directory(_curDir!);if (!directory.existsSync()) {await directory.create(recursive: true);}}
init方法主要是检查传入的保存目录是否存在,不存在就创建。
read方法与load方法分别将文件中数据读取并存入内存中,domainCookies与hostCookies
Future<void> _load(Uri uri) async {final host = uri.host;if (_hostSet.contains(host) && hostCookies[host] == null) {var str = await storage.read(host);
Future<void> saveFromResponse(Uri uri, List<Cookie> cookies) async {await _checkInitialized();if (cookies.isNotEmpty) {await super.saveFromResponse(uri, cookies);if (cookies.every((Cookie e) => e.domain == null)) {await _save(uri);} else {await _save(uri, true);}}}
先检查有无保存目录,再看是否传入cookies,若有,直接调用DefaultCookieJar的saveFromResponse,然后调用_save方法:
Future<void> _save(Uri uri, [bool withDomainSharedCookie = false]) async {final host = uri.host;if (!_hostSet.contains(host)) {_hostSet.add(host);await storage.write(IndexKey, json.encode(_hostSet.toList()));}final cookies = hostCookies[host];if (cookies != null) {await storage.write(host, json.encode(_filter(cookies)));}if (withDomainSharedCookie) {var filterDomainCookies =domainCookies.map((key, value) => MapEntry(key, _filter(value)));await storage.write(DomainsKey, json.encode(filterDomainCookies));}}
根据Cookie层级,依次写入到文件中。
总结
CookieManager与CookieJar的整个工作流程就基本分析完了,可以看到,整个流程非常清晰,代码也很简洁,健壮性也很好,这次源码学习让我对代码编写的理解有深入了一些。那么,下次我们一起来看看如何封装Dio,让我们平时使用起来更便捷。