Added identityInsert option in GenerateInsertWithIDAttribute #1

Author: RohitM-IN

Attributes/GenerateInsertWithIDAttribute.cs

@@ -6,7 +6,9 @@
     /// <remarks>
     /// This attribute can be applied to classes to indicate whether the ID property's value should be included
     /// in the insert query generation. For example, it can be used to determine whether the ID property should be
-    /// part of the generated insert query or excluded.
+    /// part of the generated insert query or excluded. The IncludeIdentityInsert property can contain database-specific
+    /// statements that affect identity insert behavior (e.g., SET IDENTITY_INSERT for MSSQL), or similar statements
+    /// applicable to other databases like MySQL or PostgreSQL.
     /// </remarks>
     [AttributeUsage(AttributeTargets.Class)]
     public class GenerateInsertWithIDAttribute : Attribute
@@ -17,16 +19,24 @@ public class GenerateInsertWithIDAttribute : Attribute
         /// <value><c>true</c> if the insert query should include the ID property; otherwise, <c>false</c>.</value>
         public bool GenerateWithID { get; }
 
+        /// <summary>
+        /// Gets a value indicating whether to include database-specific SQL statements during insert query generation
+        /// that affect identity insert behavior (e.g., SET IDENTITY_INSERT for MSSQL).
+        /// </summary>
+        /// <value><c>true</c> to include identity insert statements; otherwise, <c>false</c>.</value>
+        public bool IncludeIdentityInsert { get; }
+
         /// <summary>
         /// Initializes a new instance of the <see cref="GenerateInsertWithIDAttribute"/> class.
         /// </summary>
         /// <param name="generateWithID">Indicates whether the insert query generation should include the ID property.
         /// Default is <c>true</c>.</param>
-        public GenerateInsertWithIDAttribute(bool generateWithID = true)
+        /// <param name="includeIdentityInsert">Indicates whether to include identity insert statements in the
+        /// insert query generation. Default is <c>true</c>.</param>
+        public GenerateInsertWithIDAttribute(bool generateWithID = true, bool includeIdentityInsert = true)
         {
             GenerateWithID = generateWithID;
+            IncludeIdentityInsert = includeIdentityInsert;
         }
-
-
     }
-}
+}

Helper/QueryHelper.cs

@@ -69,6 +69,23 @@ public bool GetInsertWithID<T>() where T : IDataContractComparer
             return false;
         }
 
+        /// <summary>
+        /// Gets whether the type specifies to include database-specific SQL statements for identity insert behavior
+        /// during insert query generation, considering the GenerateInsertWithIDAttribute if present.
+        /// </summary>
+        /// <typeparam name="T">The type for which to determine the inclusion of identity insert statements.
+        /// Must implement <see cref="IDataContractComparer"/>.</typeparam>
+        /// <returns><c>true</c> if identity insert statements should be included; otherwise, <c>false</c>.</returns>
+        public bool GetIncludeIdentityInsert<T>() where T : IDataContractComparer
+        {
+            GenerateInsertWithIDAttribute? attribute = (GenerateInsertWithIDAttribute?)Attribute.GetCustomAttribute(typeof(T), typeof(GenerateInsertWithIDAttribute));
+
+            if (attribute != null)
+                return attribute.IncludeIdentityInsert;
+
+            return false;
+        }
+
         /// <summary>
         /// Gets the names of properties marked as key columns for a specified type.
         /// </summary>