Improved primary key column name referencing in the SqlKataRepository.cs

Author: topnax

Repository/SqlKataRepository.cs

@@ -1,4 +1,3 @@
-using System;
 using System.Collections.Generic;
 using System.Linq;
 using Database;
@@ -44,21 +43,26 @@ public SqlKataRepository(SqlKataDatabase db, string tableName)
         /// <returns>ID of the given object</returns>
         protected abstract int GetEntryId(T entry);
 
+        /// <summary>
+        /// Returns the name of the primary key column
+        /// </summary>
+        protected string GetPrimaryKeyColumnName => SqlSchema.TableIdPrimaryKey;
+
         // use the InsertGetId method of the SqlKata library in order to insert a new row and get it's ID
         public int Add(T entry) => Db.Get().Query(TableName).InsertGetId<int>(ToRow(entry));
 
         // updates the given object of type T using the WHERE and UPDATE method calls, returns a boolean flag indicating
         // whether at least one table row was updated
         public bool Update(T entry) =>
-            Db.Get().Query(TableName).Where(SqlSchema.TableIdPrimaryKey, GetEntryId(entry)).Update(ToRow(entry)) > 0;
+            Db.Get().Query(TableName).Where(GetPrimaryKeyColumnName, GetEntryId(entry)).Update(ToRow(entry)) > 0;
 
         // deletes the given object of type T and returns a boolean flag indicating whether at least one table row was deleted
-        public bool Delete(T entry) => Db.Get().Query(TableName).Where(SqlSchema.TableIdPrimaryKey, GetEntryId(entry)).Delete() > 0;
+        public bool Delete(T entry) => Db.Get().Query(TableName).Where(GetPrimaryKeyColumnName, GetEntryId(entry)).Delete() > 0;
 
         public T Get(int id)
         {
             // find a table rows based on the given ID
-            var result = Db.Get().Query(TableName).Where(SqlSchema.TableIdPrimaryKey, id).FirstOrDefault();
+            var result = Db.Get().Query(TableName).Where(GetPrimaryKeyColumnName, id).FirstOrDefault();
 
             // convert the given table row a ta an object of type T
             return result == null ? default : FromRow(result);