Merge pull request #390 from sqlkata/feature/add-exists

add Exists and NotExists

Author: ahmad-moussawi

Program/Program.cs

@@ -34,24 +34,18 @@ private class Installment
 
         static void Main(string[] args)
         {
-
-            var query = new Query("accounts").AsInsert(new
+            using (var db = SqlLiteQueryFactory())
             {
-                name = "new Account",
-                currency_id = "USD",
-                created_at = DateTime.UtcNow,
-                Value = SqlKata.Expressions.UnsafeLiteral("nextval('hello')", replaceQuotes: false)
-            });
-
-            var compiler = new SqlServerCompiler();
-            var sql = compiler.Compile(query).Sql;
-            Console.WriteLine(sql);
+                var query = db.Query("accounts")
+                    .Where("balance", ">", 0)
+                    .GroupBy("balance")
+                .Limit(10);
 
+                var accounts = query.Clone().Get();
+                Console.WriteLine(JsonConvert.SerializeObject(accounts, Formatting.Indented));
 
-            using (var db = SqlLiteQueryFactory())
-            {
-                var accounts = db.Query("accounts").Get();
-                Console.WriteLine(accounts.Count());
+                var exists = query.Clone().Exists();
+                Console.WriteLine(exists);
             }
         }
 
@@ -81,7 +75,17 @@ private static QueryFactory SqlLiteQueryFactory()
 
                 SQLiteConnection.CreateFile("Demo.db");
 
-                db.Statement("create table accounts(id integer primary key autoincrement, name varchar, currency_id varchar, created_at datetime);");
+                db.Statement("create table accounts(id integer primary key autoincrement, name varchar, currency_id varchar, balance decimal, created_at datetime);");
+                for (var i = 0; i < 10; i++)
+                {
+                    db.Statement("insert into accounts(name, currency_id, balance, created_at) values(@name, @currency, @balance, @date)", new
+                    {
+                        name = $"Account {i}",
+                        currency = "USD",
+                        balance = 100 * i * 1.1,
+                        date = DateTime.UtcNow,
+                    });
+                }
 
             }
 

SqlKata.Execution/Query.Extensions.cs

@@ -7,6 +7,26 @@ namespace SqlKata.Execution
 {
     public static class QueryExtensions
     {
+        public static bool Exists(this Query query, IDbTransaction transaction = null, int? timeout = null)
+        {
+            return CreateQueryFactory(query).Exists(query, transaction, timeout);
+        }
+
+        public async static Task<bool> ExistsAsync(this Query query, IDbTransaction transaction = null, int? timeout = null)
+        {
+            return await CreateQueryFactory(query).ExistsAsync(query, transaction, timeout);
+        }
+
+        public static bool NotExist(this Query query, IDbTransaction transaction = null, int? timeout = null)
+        {
+            return !CreateQueryFactory(query).Exists(query, transaction, timeout);
+        }
+
+        public async static Task<bool> NotExistAsync(this Query query, IDbTransaction transaction = null, int? timeout = null)
+        {
+            return !(await CreateQueryFactory(query).ExistsAsync(query, transaction, timeout));
+        }
+
         public static IEnumerable<T> Get<T>(this Query query, IDbTransaction transaction = null, int? timeout = null)
         {
             return CreateQueryFactory(query).Get<T>(query, transaction, timeout);

SqlKata.Execution/QueryFactory.cs

@@ -342,6 +342,30 @@ public async Task<IEnumerable<IEnumerable<T>>> GetAsync<T>(
 
         }
 
+        public bool Exists(Query query, IDbTransaction transaction = null, int? timeout = null)
+        {
+            var clone = query.Clone()
+                .ClearComponent("select")
+                .SelectRaw("1 as [Exists]")
+                .Limit(1);
+
+            var rows = Get(clone, transaction, timeout);
+
+            return rows.Any();
+        }
+
+        public async Task<bool> ExistsAsync(Query query, IDbTransaction transaction = null, int? timeout = null)
+        {
+            var clone = query.Clone()
+                .ClearComponent("select")
+                .SelectRaw("1 as [Exists]")
+                .Limit(1);
+
+            var rows = await GetAsync(clone, transaction, timeout);
+
+            return rows.Any();
+        }
+
         public T Aggregate<T>(
             Query query,
             string aggregateOperation,