flutter_package / auto_track_plugin

Go to a project
Toggle navigation Toggle navigation pinning
Dubhe
e95a049a 1 parent add2ed95

feat(request): add request listener

Inline Side-by-side
Showing 2 changed files with 421 additions and 0 deletions
import 'dart:async';
import 'dart:convert';
import 'dart:io';
import 'package:auto_track/auto_track/track/track.dart';
import '../../utils/request_model.dart';
import '../page_view/page_stack.dart';
class HttpClientRequestWithChecker implements HttpClientRequest {
final HttpClientRequest _realRequest;
final Stopwatch _stopwatch;
final Page? pageInfoData;
HttpClientRequestWithChecker(
this._realRequest, this._stopwatch, this.pageInfoData);
@override
bool get bufferOutput => _realRequest.bufferOutput;
@override
int get contentLength => _realRequest.contentLength;
@override
Encoding get encoding => _realRequest.encoding;
@override
bool get followRedirects => _realRequest.followRedirects;
@override
int get maxRedirects => _realRequest.maxRedirects;
@override
bool get persistentConnection => _realRequest.persistentConnection;
@override
void add(List<int> data) {
_realRequest.add(data);
}
@override
void addError(Object error, [StackTrace? stackTrace]) {
_realRequest.addError(error, stackTrace);
}
@override
Future addStream(Stream<List<int>> stream) {
return _realRequest.addStream(stream);
}
@override
Future<HttpClientResponse> close() async {
return _realRequest.close().then((HttpClientResponse response) {
_checkResponse(_realRequest, response);
return response;
}).catchError((dynamic error, dynamic stackTrace) {}, test: (error) {
_stopwatch.stop();
String message;
if (error is HttpException) {
message = error.message;
} else {
message = error.toString();
}
Track.instance.reportHttpRequest(RequestModel(
uri: _realRequest.uri,
method: method,
pageId: pageInfoData?.pageInfo?.pageKey ?? "",
requestHeaders: _realRequest.headers,
message: message,
status: -1,
spent: _stopwatch.elapsedMilliseconds));
return false;
});
}
@override
HttpConnectionInfo? get connectionInfo => _realRequest.connectionInfo;
@override
List<Cookie> get cookies => _realRequest.cookies;
@override
Future<HttpClientResponse> get done async {
return close();
}
@override
Future flush() {
return _realRequest.flush();
}
@override
HttpHeaders get headers => _realRequest.headers;
@override
String get method => _realRequest.method;
@override
Uri get uri => _realRequest.uri;
@override
void write(Object? obj) {
_realRequest.write(obj);
}
@override
void writeAll(Iterable objects, [String separator = '']) {
_realRequest.writeAll(objects, separator);
}
@override
void writeCharCode(int charCode) {
_realRequest.writeCharCode(charCode);
}
@override
void writeln([Object? obj = '']) {
_realRequest.writeln(obj);
}
void _checkResponse(HttpClientRequest request, HttpClientResponse response) {
String message = 'status ${response.statusCode}';
message = '$message: ${response.reasonPhrase}';
_stopwatch.stop();
Track.instance.reportHttpRequest(RequestModel(
uri: _realRequest.uri,
method: method,
pageId: pageInfoData?.pageInfo?.pageKey ?? "",
requestHeaders: request.headers,
responseHeaders: response.headers,
message: message,
status: response.statusCode,
spent: _stopwatch.elapsedMilliseconds));
}
@override
set bufferOutput(bool bufferOutput) {
_realRequest.bufferOutput = bufferOutput;
}
@override
set contentLength(int contentLength) {
_realRequest.contentLength = contentLength;
}
@override
set encoding(Encoding encoding) {
_realRequest.encoding = encoding;
}
@override
set followRedirects(bool followRedirects) {
_realRequest.followRedirects = followRedirects;
}
@override
set maxRedirects(int maxRedirects) {
_realRequest.maxRedirects = maxRedirects;
}
@override
set persistentConnection(bool persistentConnection) {
_realRequest.persistentConnection = persistentConnection;
}
@override
void abort([Object? exception, StackTrace? stackTrace]) {
_realRequest.abort(exception, stackTrace);
}
}
class HttpClientWithChecker implements HttpClient {
final HttpClient _realClient;
Uri? url;
String? method;
HttpClientWithChecker(this._realClient);
@override
set connectionFactory(
Future<ConnectionTask<Socket>> Function(
Uri url, String? proxyHost, int? proxyPort)?
f) {
// TODO: add impl here
assert(false);
}
@override
set keyLog(Function(String line)? callback) {
// TODO: add impl here
assert(false);
}
@override
bool get autoUncompress => _realClient.autoUncompress;
@override
set autoUncompress(bool value) => _realClient.autoUncompress = value;
@override
Duration? get connectionTimeout => _realClient.connectionTimeout;
@override
set connectionTimeout(Duration? value) =>
_realClient.connectionTimeout = value;
@override
Duration get idleTimeout => _realClient.idleTimeout;
@override
set idleTimeout(Duration value) => _realClient.idleTimeout = value;
@override
int? get maxConnectionsPerHost => _realClient.maxConnectionsPerHost;
@override
set maxConnectionsPerHost(int? value) =>
_realClient.maxConnectionsPerHost = value;
@override
String? get userAgent => _realClient.userAgent;
@override
set userAgent(String? value) => _realClient.userAgent = value;
@override
void addCredentials(
Uri url, String realm, HttpClientCredentials credentials) =>
_realClient.addCredentials(url, realm, credentials);
@override
void addProxyCredentials(String host, int port, String realm,
HttpClientCredentials credentials) =>
_realClient.addProxyCredentials(host, port, realm, credentials);
@override
set authenticate(
Future<bool> Function(Uri url, String scheme, String? realm)? f) =>
_realClient.authenticate = f;
@override
set authenticateProxy(
Future<bool> Function(
String host, int port, String scheme, String? realm)?
f) =>
_realClient.authenticateProxy = f;
@override
set badCertificateCallback(
bool Function(X509Certificate cert, String host, int port)?
callback) =>
_realClient.badCertificateCallback = callback;
@override
void close({bool force = false}) => _realClient.close(force: force);
@override
Future<HttpClientRequest> delete(String host, int port, String path) =>
_realClient.delete(host, port, path);
@override
Future<HttpClientRequest> deleteUrl(Uri url) => _realClient.deleteUrl(url);
@override
set findProxy(String Function(Uri url)? f) => _realClient.findProxy = f;
@override
Future<HttpClientRequest> get(String host, int port, String path) =>
_realClient.get(host, port, path);
@override
Future<HttpClientRequest> getUrl(Uri url) =>
_addCheck(_realClient.getUrl(url), 'get', url);
@override
Future<HttpClientRequest> head(String host, int port, String path) =>
_realClient.head(host, port, path);
@override
Future<HttpClientRequest> headUrl(Uri url) =>
_addCheck(_realClient.headUrl(url), 'head', url);
@override
Future<HttpClientRequest> patch(String host, int port, String path) =>
_realClient.patch(host, port, path);
@override
Future<HttpClientRequest> patchUrl(Uri url) =>
_addCheck(_realClient.patchUrl(url), 'patch', url);
@override
Future<HttpClientRequest> post(String host, int port, String path) =>
_realClient.post(host, port, path);
@override
Future<HttpClientRequest> postUrl(Uri url) =>
_addCheck(_realClient.postUrl(url), 'post', url);
@override
Future<HttpClientRequest> put(String host, int port, String path) =>
_realClient.put(host, port, path);
@override
Future<HttpClientRequest> putUrl(Uri url) =>
_addCheck(_realClient.putUrl(url), 'put', url);
@override
Future<HttpClientRequest> open(
String method, String host, int port, String path) {
const int hashMark = 0x23;
const int questionMark = 0x3f;
int fragmentStart = path.length;
int queryStart = path.length;
for (int i = path.length - 1; i >= 0; i--) {
final char = path.codeUnitAt(i);
if (char == hashMark) {
fragmentStart = i;
queryStart = i;
} else if (char == questionMark) {
queryStart = i;
}
}
String? query;
if (queryStart < fragmentStart) {
query = path.substring(queryStart + 1, fragmentStart);
path = path.substring(0, queryStart);
}
final Uri uri =
Uri(scheme: 'http', host: host, port: port, path: path, query: query);
return _addCheck(_realClient.open(method, host, port, path), method, uri);
}
@override
Future<HttpClientRequest> openUrl(String method, Uri url) =>
_addCheck(_realClient.openUrl(method, url), method, url);
Future<HttpClientRequest> _addCheck(
Future<HttpClientRequest> request, String method, Uri url) {
final Stopwatch stopwatch = Stopwatch()..start();
final Page? pageInfoData = PageStack.instance.getCurrentPage();
return request
.then((HttpClientRequest request) =>
HttpClientRequestWithChecker(request, stopwatch, pageInfoData))
.catchError((dynamic error, dynamic stackTrace) {}, test: (error) {
String message = error.toString();
if (error is SocketException) {
message = error.message;
}
Track.instance.reportHttpRequest(RequestModel(
uri: url,
method: method,
pageId: pageInfoData?.pageInfo?.pageKey ?? "",
requestHeaders: null,
message: message,
status: -1,
spent: stopwatch.elapsedMilliseconds));
return false;
});
}
}
class _DefaultHttpOverrides extends HttpOverrides {}
class AutoTrackHttpOverrides extends HttpOverrides {
HttpOverrides? _currentOverrides;
AutoTrackHttpOverrides(this._currentOverrides) : super() {
_currentOverrides ??= _DefaultHttpOverrides();
}
@override
HttpClient createHttpClient(SecurityContext? context) {
return HttpClientWithChecker(_currentOverrides!.createHttpClient(context));
}
@override
String findProxyFromEnvironment(Uri url, Map<String, String>? environment) {
return _currentOverrides!.findProxyFromEnvironment(url, environment);
}
}
... ...
class RequestModel {
RequestModel({
required this.uri,
required this.method,
required this.message,
required this.status,
required this.spent,
required this.pageId,
this.requestBody,
this.requestHeaders,
this.responseHeaders,
});
Uri uri;
String method;
String message;
String pageId;
int status;
int spent;
dynamic requestBody;
dynamic requestHeaders;
dynamic responseHeaders;
Map<String, dynamic> toMap() {
return {
'uri': uri,
'method': method,
'message': message,
'pageId': pageId,
'status': status,
'spent': spent,
'requestBody': requestBody,
'requestHeaders': requestHeaders,
'responseHeaders': responseHeaders,
};
}
}
\ No newline at end of file
... ...