Use StreamedRequest with onSendProgress in adapter

cfug · Jul 23, 2023 · d79873b · d79873b
1 parent ef1ed39
commit d79873b
Showing 1 changed file with 36 additions and 19 deletions.
diff --git a/plugins/native_dio_adapter/lib/src/conversion_layer_adapter.dart b/plugins/native_dio_adapter/lib/src/conversion_layer_adapter.dart
@@ -34,34 +34,51 @@ class ConversionLayerAdapter implements HttpClientAdapter {
     RequestOptions options,
     Stream<Uint8List>? requestStream,
   ) async {
-    final request = Request(
-      options.method,
-      options.uri,
-    );
+    final BaseRequest request;
+    if (options.onSendProgress != null && requestStream != null) {
+      final streamedRequest = StreamedRequest(
+        options.method,
+        options.uri,
+      );
+      requestStream.listen(
+        streamedRequest.sink.add,
+        onDone: streamedRequest.sink.close,
+        onError: streamedRequest.sink.addError,
+        cancelOnError: true,
+      );
+      request = streamedRequest;
+    } else {
+      final simpleRequest = Request(
+        options.method,
+        options.uri,
+      );
+      if (requestStream != null) {
+        final completer = Completer<Uint8List>();
+        final sink = ByteConversionSink.withCallback(
+          (bytes) => completer.complete(Uint8List.fromList(bytes)),
+        );
+        requestStream.listen(
+          sink.add,
+          onError: completer.completeError,
+          onDone: sink.close,
+          cancelOnError: true,
+        );
+        final bytes = await completer.future;
+        simpleRequest.bodyBytes = bytes;
+        // simpleRequest.bodyBytes = await ByteStream(requestStream).toBytes();
+      }
+      request = simpleRequest;
+    }
 
     request.headers.addAll(
       Map.fromEntries(
         options.headers.entries.map((e) => MapEntry(e.key, e.value.toString())),
       ),
     );
-
     request.followRedirects = options.followRedirects;
     request.maxRedirects = options.maxRedirects;
+    request.persistentConnection = options.persistentConnection;
 
-    if (requestStream != null) {
-      final completer = Completer<Uint8List>();
-      final sink = ByteConversionSink.withCallback(
-        (bytes) => completer.complete(Uint8List.fromList(bytes)),
-      );
-      requestStream.listen(
-        sink.add,
-        onError: completer.completeError,
-        onDone: sink.close,
-        cancelOnError: true,
-      );
-      final bytes = await completer.future;
-      request.bodyBytes = bytes;
-    }
     return request;
   }
 }