set isFinished true for gRPC requests (#10)

divar-ir · Jan 10, 2022 · 576b67e · 576b67e
1 parent 0de3746
commit 576b67e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/Sources/NetShears/Models/RequestModel.swift b/Sources/NetShears/Models/RequestModel.swift
@@ -99,7 +99,8 @@ public final class NetShearsRequestModel: Codable {
          statusCode: Int,
          duration: Double?,
          HPACKHeadersRequest: [String: String]?,
-         HPACKHeadersResponse: [String: String]?) {
+         HPACKHeadersResponse: [String: String]?,
+         isFinished: Bool = true) {
         self.id = UUID().uuidString
         self.method = "gRPC"
         self.scheme = "gRPC"
@@ -114,7 +115,7 @@ public final class NetShearsRequestModel: Codable {
         self.port = nil
         self.duration = duration
         self.credentials = [:]
-        self.isFinished = false
+        self.isFinished = isFinished
     }
 
     var curlRequest: String {

diff --git a/Sources/NetShears/NetShears.swift b/Sources/NetShears/NetShears.swift
@@ -87,7 +87,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)
+        let request = NetShearsRequestModel(url: url, host: host, requestObject: requestObject, responseObject: responseObject, success: success, statusCode: statusCode, duration: duration, HPACKHeadersRequest: HPACKHeadersRequest, HPACKHeadersResponse: HPACKHeadersResponse, isFinished: true)
         if loggerEnable {
             RequestStorage.shared.newRequestArrived(request)
         }