added count feature

michael-shattuck · Feb 2, 2015 · 58f533c · 58f533c
1 parent 446ea36
commit 58f533c
Show file tree

Hide file tree

Showing 5 changed files with 65 additions and 7 deletions.
diff --git a/SalesforceMagic/Abstract/ISalesforceClient.cs b/SalesforceMagic/Abstract/ISalesforceClient.cs
@@ -19,6 +19,20 @@ public interface ISalesforceClient : IDisposable
 
         #endregion
 
+        #region Count Methods
+
+        /// <summary>
+        ///     Retrieve an object total count.
+        ///      - Allows for filtering count totals 
+        ///        using the conditional lambda.
+        /// </summary>
+        /// <typeparam name="T"></typeparam>
+        /// <param name="predicate"></param>
+        /// <returns></returns>
+        int Count<T>(Expression<Func<T, bool>> predicate = null) where T : SObject;
+
+        #endregion
+
         #region Query Methods
 
         /// <summary>

diff --git a/SalesforceMagic/ORM/QueryBuilder.cs b/SalesforceMagic/ORM/QueryBuilder.cs
@@ -8,7 +8,7 @@ namespace SalesforceMagic.ORM
 {
     public static class QueryBuilder
     {
-        public static string GenerateQuery<T>(int limit = default(int))
+        internal static string GenerateQuery<T>(int limit = default(int))
             where T : SObject
         {
             Type type = typeof(T);
@@ -18,7 +18,7 @@ public static class QueryBuilder
             return query;
         }
 
-        public static string GenerateQuery<T>(Expression<Func<T, bool>> predicate, int limit = default(int))
+        internal static string GenerateQuery<T>(Expression<Func<T, bool>> predicate, int limit = default(int))
             where T : SObject
         {
             Type type = typeof(T);
@@ -29,6 +29,15 @@ public static class QueryBuilder
             return query;
         }
 
+        internal static string GenerateCountyQuery<T>(Expression<Func<T, bool>> predicate)
+        {
+            Type type = typeof(T);
+            string query = string.Format("SELECT COUNT() FROM {0}", type.GetName());
+            if (predicate != null) AddConditionsSet(ref query, predicate);
+
+            return query;
+        }
+
         private static string CompileSelectStatements(Type type)
         {
             return string.Format("SELECT {0} FROM {1}", string.Join(", ", type.GetPropertyNames(true)), type.GetName());

diff --git a/SalesforceMagic/ORM/ResponseReader.cs b/SalesforceMagic/ORM/ResponseReader.cs
@@ -26,6 +26,12 @@ internal static bool ReadBoolResponse(string name, XmlDocument document)
             return node != null ? Convert.ToBoolean(node.InnerText) : default(bool);
         }
 
+        internal static int ReadCountResponse(string name, XmlDocument document)
+        {
+            XmlNode node = GetSingleNamedNodes(document, name);
+            return node != null ? Convert.ToInt32(node.InnerText) : default(int);
+        }
+
         internal static SalesforceResponse ReadSimpleResponse(XmlDocument document)
         {
             SalesforceResponse response = new SalesforceResponse();

diff --git a/SalesforceMagic/SalesforceClient.cs b/SalesforceMagic/SalesforceClient.cs
@@ -95,6 +95,20 @@ public SalesforceSession Login()
             }
         }
 
+        /// <summary>
+        ///     Retrieve an object total count.
+        ///      - Allows for filtering count totals 
+        ///        using the conditional lambda.
+        /// </summary>
+        /// <typeparam name="T"></typeparam>
+        /// <param name="predicate"></param>
+        /// <returns></returns>
+        public int Count<T>(Expression<Func<T, bool>> predicate = null) where T : SObject
+        {
+            var request = SoapRequestManager.GetCountRequest(Login(), predicate);
+            return PerformCountRequest(request);
+        }
+
         /// <summary>
         ///     Simple Query
         ///      - Query items based on generic object
@@ -320,6 +334,15 @@ private T PerformGenericRequest<T>(HttpRequest request, string rootName = "resul
             }
         }
 
+        private int PerformCountRequest(HttpRequest request)
+        {
+            using (HttpClient httpClient = new HttpClient())
+            {
+                XmlDocument response = httpClient.PerformRequest(request);
+                return ResponseReader.ReadCountResponse("size", response);
+            }
+        }
+
         private SalesforceResponse PerformSimpleRequest(HttpRequest request)
         {
             using (HttpClient httpClient = new HttpClient())

diff --git a/SalesforceMagic/SoapApi/SoapRequestManager.cs b/SalesforceMagic/SoapApi/SoapRequestManager.cs
@@ -26,19 +26,19 @@ internal static HttpRequest GetLoginRequest(SalesforceConfig config)
             return request;
         }
 
-        public static HttpRequest GetQueryRequest<T>(Expression<Func<T, bool>> predicate, int limit, SalesforceSession session) where T : SObject
+        internal static HttpRequest GetQueryRequest<T>(Expression<Func<T, bool>> predicate, int limit, SalesforceSession session) where T : SObject
         {
             string query = QueryBuilder.GenerateQuery(predicate, limit);
             return GetQueryRequest(query, session);
         }
 
-        public static HttpRequest GetQueryAllRequest<T>(int limit, SalesforceSession session) where T : SObject
+        internal static HttpRequest GetQueryAllRequest<T>(int limit, SalesforceSession session) where T : SObject
         {
             string query = QueryBuilder.GenerateQuery<T>(limit);
             return GetQueryRequest(query, session);
         }
 
-        public static HttpRequest GetQueryRequest(string query, SalesforceSession session)
+        internal static HttpRequest GetQueryRequest(string query, SalesforceSession session)
         {
             HttpRequest request = new HttpRequest
             {
@@ -51,7 +51,7 @@ public static HttpRequest GetQueryRequest(string query, SalesforceSession sessio
             return request;
         }
 
-        public static HttpRequest GetQueryMoreRequest(string queryLocator, SalesforceSession session)
+        internal static HttpRequest GetQueryMoreRequest(string queryLocator, SalesforceSession session)
         {
             HttpRequest request = new HttpRequest
             {
@@ -64,7 +64,7 @@ public static HttpRequest GetQueryMoreRequest(string queryLocator, SalesforceSes
             return request;
         }
 
-        public static HttpRequest GetCrudRequest<T>(CrudOperation<T> operation, SalesforceSession session) where T : SObject
+        internal static HttpRequest GetCrudRequest<T>(CrudOperation<T> operation, SalesforceSession session) where T : SObject
         {
             string body = SoapCommands.CrudOperation(operation, session.SessionId);
             HttpRequest request = new HttpRequest
@@ -77,5 +77,11 @@ public static HttpRequest GetCrudRequest<T>(CrudOperation<T> operation, Salesfor
 
             return request;
         }
+
+        internal static HttpRequest GetCountRequest<T>(SalesforceSession session, Expression<Func<T, bool>> predicate = null)
+        {
+            string query = QueryBuilder.GenerateCountyQuery(predicate);
+            return GetQueryRequest(query, session);
+        }
     }
 }