add increment execution

sqlkata · Jun 10, 2021 · a79c21f · a79c21f
1 parent 550122d
commit a79c21f
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/SqlKata.Execution/Query.Extensions.cs b/SqlKata.Execution/Query.Extensions.cs
@@ -245,6 +245,26 @@ public static async Task<int> UpdateAsync(this Query query, object data, IDbTran
             return await CreateQueryFactory(query).ExecuteAsync(query.AsUpdate(data), transaction, timeout, cancellationToken);
         }
 
+        public static int Increment(this Query query, string column, int value = 1, IDbTransaction transaction = null, int? timeout = null)
+        {
+            return CreateQueryFactory(query).Execute(query.AsIncrement(column, value), transaction, timeout);
+        }
+
+        public static async Task<int> IncrementAsync(this Query query, string column, int value = 1, IDbTransaction transaction = null, int? timeout = null, CancellationToken cancellationToken = default)
+        {
+            return await CreateQueryFactory(query).ExecuteAsync(query.AsIncrement(column, value), transaction, timeout, cancellationToken);
+        }
+
+        public static int Decrement(this Query query, string column, int value = 1, IDbTransaction transaction = null, int? timeout = null)
+        {
+            return CreateQueryFactory(query).Execute(query.AsDecrement(column, value), transaction, timeout);
+        }
+
+        public static async Task<int> DecrementAsync(this Query query, string column, int value = 1, IDbTransaction transaction = null, int? timeout = null, CancellationToken cancellationToken = default)
+        {
+            return await CreateQueryFactory(query).ExecuteAsync(query.AsDecrement(column, value), transaction, timeout, cancellationToken);
+        }
+
         public static int Delete(this Query query, IDbTransaction transaction = null, int? timeout = null)
         {
             return CreateQueryFactory(query).Execute(query.AsDelete(), transaction, timeout);