Add method for gRPC

divar-ir · Jan 16, 2022 · 8bcba66 · 8bcba66
1 parent 90c57f5
commit 8bcba66
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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)
         }