From 8bcba660da0dd1757b9cca14a99d2015bb05bf5c Mon Sep 17 00:00:00 2001 From: Ali Moazenzadeh Date: Sun, 16 Jan 2022 10:55:35 +0330 Subject: [PATCH] Add method for gRPC --- Sources/NetShears/Models/RequestModel.swift | 3 ++- Sources/NetShears/NetShears.swift | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/Sources/NetShears/Models/RequestModel.swift b/Sources/NetShears/Models/RequestModel.swift index 1b03fee..4b15dc5 100644 --- a/Sources/NetShears/Models/RequestModel.swift +++ b/Sources/NetShears/Models/RequestModel.swift @@ -93,6 +93,7 @@ public final class NetShearsRequestModel: Codable { init(url: String, host: String, + method: String, requestObject: Data?, responseObject: Data?, success: Bool, @@ -102,7 +103,7 @@ public final class NetShearsRequestModel: Codable { HPACKHeadersResponse: [String: String]?, isFinished: Bool = true) { self.id = UUID().uuidString - self.method = "gRPC" + self.method = method self.scheme = "gRPC" self.url = url self.host = host diff --git a/Sources/NetShears/NetShears.swift b/Sources/NetShears/NetShears.swift index 5bd2f6f..f38c500 100644 --- a/Sources/NetShears/NetShears.swift +++ b/Sources/NetShears/NetShears.swift @@ -79,6 +79,7 @@ public final class NetShears: NSObject { public func addGRPC(url: String, host: String, + method: String, requestObject: Data?, responseObject: Data?, success: Bool, @@ -87,7 +88,7 @@ public final class NetShears: NSObject { duration: Double?, HPACKHeadersRequest: [String: String]?, HPACKHeadersResponse: [String: String]?){ - let request = NetShearsRequestModel(url: url, host: host, requestObject: requestObject, responseObject: responseObject, success: success, statusCode: statusCode, duration: duration, HPACKHeadersRequest: HPACKHeadersRequest, HPACKHeadersResponse: HPACKHeadersResponse, isFinished: true) + let request = NetShearsRequestModel(url: url, host: host, method: method, requestObject: requestObject, responseObject: responseObject, success: success, statusCode: statusCode, duration: duration, HPACKHeadersRequest: HPACKHeadersRequest, HPACKHeadersResponse: HPACKHeadersResponse, isFinished: true) if loggerEnable { RequestStorage.shared.newRequestArrived(request) }