From b1c493714ae251697645e0dcff62a3e8eb238685 Mon Sep 17 00:00:00 2001
From: RohitM-IN <rohit.mahajan.in@gmail.com>
Date: Sun, 31 Dec 2023 13:02:18 +0530
Subject: [PATCH] Code optimizate

---
 src/QueryGenerator.cs | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/QueryGenerator.cs b/src/QueryGenerator.cs
index ebfa8c5..fe384c8 100644
--- a/src/QueryGenerator.cs
+++ b/src/QueryGenerator.cs
@@ -187,8 +187,7 @@ public void Dispose()
         public List<string> GetCondition<T>(T entity, List<string> keyColumns) where T : IDataContractComparer
         {
             Type entityType = typeof(T);
-            PropertyInfo[] keyProperties = entityType.GetProperties().
-                Where(p => keyColumns.Contains(p.Name)).ToArray();
+            PropertyInfo[] keyProperties = GetKeyProperties<T>();
 
             return keyProperties.Select(p => $"{EscapeColumn(p.Name)} = '{EscapeValue(p.GetValue(entity))}'").ToList();
         }
@@ -213,9 +212,10 @@ public List<string> GetCondition<T>(T entity, List<string> keyColumns) where T :
         /// <returns>The escaped column name enclosed in double quotes.</returns>
         public string EscapeColumn(string? input)
         {
-            if(input.Contains(" "))
+            if(input is string && input.Contains(" "))
                 return $"\"{input}\"";
-            return input;
+
+            return input.ToString();
         }
 
         /// <summary>