|
| 1 | +#import "SentryNetworkRequestData.h" |
| 2 | + |
| 3 | +NSString *const SentryReplayNetworkDetailsKey = @"_networkDetails"; |
| 4 | + |
| 5 | +@interface SentryNetworkRequestData () |
| 6 | +@property (nonatomic, copy, readwrite, nullable) NSString *method; |
| 7 | +@property (nonatomic, strong, readwrite, nullable) NSNumber *statusCode; |
| 8 | +@property (nonatomic, strong, readwrite, nullable) NSNumber *requestBodySize; |
| 9 | +@property (nonatomic, strong, readwrite, nullable) NSNumber *responseBodySize; |
| 10 | +@property (nonatomic, strong, readwrite, nullable) SentryReplayNetworkRequestOrResponse *request; |
| 11 | +@property (nonatomic, strong, readwrite, nullable) SentryReplayNetworkRequestOrResponse *response; |
| 12 | +@end |
| 13 | + |
| 14 | +@implementation SentryNetworkRequestData |
| 15 | + |
| 16 | +- (instancetype)initWithMethod:(nullable NSString *)method |
| 17 | +{ |
| 18 | + if (self = [super init]) { |
| 19 | + _method = [method copy]; |
| 20 | + } |
| 21 | + return self; |
| 22 | +} |
| 23 | + |
| 24 | +- (void)setRequestDetails:(SentryReplayNetworkRequestOrResponse *)requestData |
| 25 | +{ |
| 26 | + self.request = requestData; |
| 27 | + self.requestBodySize = requestData.size; |
| 28 | +} |
| 29 | + |
| 30 | +- (void)setResponseDetails:(NSInteger)statusCode |
| 31 | + responseData:(SentryReplayNetworkRequestOrResponse *)responseData |
| 32 | +{ |
| 33 | + self.statusCode = @(statusCode); |
| 34 | + self.response = responseData; |
| 35 | + self.responseBodySize = responseData.size; |
| 36 | +} |
| 37 | + |
| 38 | +- (NSDictionary *)serialize |
| 39 | +{ |
| 40 | + NSMutableDictionary *result = [NSMutableDictionary dictionary]; |
| 41 | + |
| 42 | + if (self.method) { |
| 43 | + result[@"method"] = self.method; |
| 44 | + } |
| 45 | + if (self.statusCode) { |
| 46 | + result[@"statusCode"] = self.statusCode; |
| 47 | + } |
| 48 | + if (self.requestBodySize) { |
| 49 | + result[@"requestBodySize"] = self.requestBodySize; |
| 50 | + } |
| 51 | + if (self.responseBodySize) { |
| 52 | + result[@"responseBodySize"] = self.responseBodySize; |
| 53 | + } |
| 54 | + if (self.request) { |
| 55 | + result[@"request"] = [self.request serialize]; |
| 56 | + } |
| 57 | + if (self.response) { |
| 58 | + result[@"response"] = [self.response serialize]; |
| 59 | + } |
| 60 | + |
| 61 | + return result; |
| 62 | +} |
| 63 | + |
| 64 | +- (NSString *)description |
| 65 | +{ |
| 66 | + NSDictionary *serialized = [self serialize]; |
| 67 | + return [NSString stringWithFormat:@"SentryNetworkRequestData: %@", serialized]; |
| 68 | +} |
| 69 | + |
| 70 | +@end |
0 commit comments