From 576b67ed1e87abbc4c1b450749d651d9995636d3 Mon Sep 17 00:00:00 2001
From: Ali Moazenzadeh <aliveravb@gmail.com>
Date: Mon, 10 Jan 2022 13:50:54 +0330
Subject: [PATCH] set isFinished true for gRPC requests (#10)

---
 Sources/NetShears/Models/RequestModel.swift | 5 +++--
 Sources/NetShears/NetShears.swift           | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/Sources/NetShears/Models/RequestModel.swift b/Sources/NetShears/Models/RequestModel.swift
index 0ba0fa2..1b03fee 100644
--- 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
index 313e8a0..5bd2f6f 100644
--- 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)
         }