fix(iOS): concurrent inputStream callback falied (#1105)

Author: yFeii

iOS/DoraemonKit/Src/Core/Plugin/Performance/NetFlow/Function/DoraemonNetFlowManager.m

@@ -12,53 +12,39 @@
 #import "DoraemonNetworkInterceptor.h"
 #import "UIViewController+Doraemon.h"
 #import "DoraemonHealthManager.h"
+#import <objc/runtime.h>
 
-@interface DoraemonNetFlowManager() <DoraemonNetworkInterceptorDelegate, NSStreamDelegate>
-
-@property (nonatomic, copy) HttpBodyCallBack bodyCallBack;
-@property (nonatomic, strong) NSMutableData *bodyData;
+@interface NSInputStream (DoraemonHttpBodyCallBack)
 
+@property (nonatomic, strong) id<NSStreamDelegate> dkStrongDelegate;
 @end
 
-@implementation DoraemonNetFlowManager
+@implementation NSInputStream (DoraemonHttpBodyCallBack)
 
-+ (DoraemonNetFlowManager *)shareInstance{
-    static dispatch_once_t once;
-    static DoraemonNetFlowManager *instance;
-    dispatch_once(&once, ^{
-        instance = [[DoraemonNetFlowManager alloc] init];
-    });
-    return instance;
+- (void)setDkStrongDelegate:(id<NSStreamDelegate>)dkStrongDelegate {
+    objc_setAssociatedObject(self, @selector(dkStrongDelegate), dkStrongDelegate, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
 }
 
-- (void)canInterceptNetFlow:(BOOL)enable{
-    _canIntercept = enable;
-    if (enable) {
-        [[DoraemonNetworkInterceptor shareInstance] addDelegate:self];
-        _startInterceptDate = [NSDate date];
-    }else{
-        [DoraemonNetworkInterceptor.shareInstance removeDelegate:self];
-        _startInterceptDate = nil;
-        [[DoraemonNetFlowDataSource shareInstance] clear];
-    }
+- (id<NSStreamDelegate>)dkStrongDelegate {
+    return objc_getAssociatedObject(self, _cmd);
 }
+@end
 
-- (void)httpBodyFromRequest:(NSURLRequest *)request bodyCallBack:(HttpBodyCallBack)complete {
-    NSData *httpBody = nil;
-    if (request.HTTPBody) {
-        httpBody = request.HTTPBody;
-        complete(httpBody);
-        return;
-    }
-    if ([request.HTTPMethod isEqualToString:@"POST"]) {
-        NSInputStream *stream = request.HTTPBodyStream;
-        [stream setDelegate:self];
-        self.bodyCallBack = complete;
-        [stream scheduleInRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
-        [stream open];
-    } else {
-        complete(httpBody);
+@interface DoraemonInputStreamDelegate : NSObject<NSStreamDelegate>
+
+@property (nonatomic, strong) NSMutableData *bodyData;
+@property (nonatomic, copy) HttpBodyCallBack bodyCallBack;
+@end
+
+@implementation DoraemonInputStreamDelegate
+- (instancetype)initWithCallback:(HttpBodyCallBack)callback inputStream:(NSInputStream *)inputStream{
+    self = [super init];
+    if(self){
+        _bodyCallBack = callback;
+        inputStream.dkStrongDelegate = self;//keep alive
+        inputStream.delegate = self;
     }
+    return self;
 }
 
 #pragma mark -- NSStreamDelegate
@@ -89,7 +75,6 @@ - (void)stream:(NSStream *)aStream handleEvent:(NSStreamEvent)eventCode {
         {
             [aStream close];
             [aStream removeFromRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
-            aStream = nil;
             if (self.bodyCallBack) {
                 self.bodyCallBack([self.bodyData copy]);
             }
@@ -101,6 +86,60 @@ - (void)stream:(NSStream *)aStream handleEvent:(NSStreamEvent)eventCode {
     }
 }
 
+
+@end
+
+
+
+
+@interface DoraemonNetFlowManager() <DoraemonNetworkInterceptorDelegate>
+
+
+@property (nonatomic, strong) NSMapTable *bodyCallBackMap;
+
+@end
+
+@implementation DoraemonNetFlowManager
+
++ (DoraemonNetFlowManager *)shareInstance{
+    static dispatch_once_t once;
+    static DoraemonNetFlowManager *instance;
+    dispatch_once(&once, ^{
+        instance = [[DoraemonNetFlowManager alloc] init];
+        instance.bodyCallBackMap = [NSMapTable mapTableWithKeyOptions:NSPointerFunctionsWeakMemory valueOptions:NSPointerFunctionsStrongMemory];
+    });
+    return instance;
+}
+
+- (void)canInterceptNetFlow:(BOOL)enable{
+    _canIntercept = enable;
+    if (enable) {
+        [[DoraemonNetworkInterceptor shareInstance] addDelegate:self];
+        _startInterceptDate = [NSDate date];
+    }else{
+        [DoraemonNetworkInterceptor.shareInstance removeDelegate:self];
+        _startInterceptDate = nil;
+        [[DoraemonNetFlowDataSource shareInstance] clear];
+    }
+}
+
+- (void)httpBodyFromRequest:(NSURLRequest *)request bodyCallBack:(HttpBodyCallBack)complete {
+    NSData *httpBody = nil;
+    if (request.HTTPBody) {
+        httpBody = request.HTTPBody;
+        complete(httpBody);
+        return;
+    }
+    if ([request.HTTPMethod isEqualToString:@"POST"]) {
+        NSInputStream *stream = request.HTTPBodyStream;
+        DoraemonInputStreamDelegate *delegate = [[DoraemonInputStreamDelegate alloc] initWithCallback:complete inputStream:stream];
+        [stream scheduleInRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
+        [stream open];
+    } else {
+        complete(httpBody);
+    }
+}
+
 #pragma mark -- DoraemonNetworkInterceptorDelegate
 - (void)doraemonNetworkInterceptorDidReceiveData:(NSData *)data response:(NSURLResponse *)response request:(NSURLRequest *)request error:(NSError *)error startTime:(NSTimeInterval)startTime {
     [DoraemonNetFlowHttpModel dealWithResponseData:data response:response request:request complete:^(DoraemonNetFlowHttpModel *httpModel) {