TEAMModelOS/TEAMModelBI/Tool/CosmosBank/JointlySingleQuery1.cs

using Microsoft.Azure.Cosmos;
using System.Collections.Generic;
using System.Threading.Tasks;
using TEAMModelOS.SDK.DI;

namespace TEAMModelBI.Tool.CosmosBank
{
    public class JointlySingleQuery1
    {
        /// <summary>
        /// 多个容器返回整数 使用
        /// </summary>
        /// <param name="cosmosClient"></param>
        /// <param name="containerId"></param>
        /// <param name="sqlTxt"></param>
        /// <param name="code"></param>
        /// <returns></returns>
        public static async Task<int> GetValueInt(CosmosClient cosmosClient, List<string> containerId, string sqlTxt = null, string code = null) 
        {
            int totals = 0;
            string sql = $"select value(count(c.id)) from c";
            if (!string.IsNullOrEmpty(sqlTxt)) 
            {
                sql = sqlTxt;
            }

            foreach (string conId in containerId) 
            {
                await foreach (var itemInt in cosmosClient.GetContainer("TEAMModelOS", conId).GetItemQueryIteratorSql<int>(queryText: sql, requestOptions: string.IsNullOrEmpty($"{code}") ? new QueryRequestOptions() { } : new QueryRequestOptions() { PartitionKey = new PartitionKey($"{code}") }))
                {
                    totals += itemInt;
                }
            }

            return totals;            
        }

        /// <summary>
        /// 单个容器返回整数 统计使用
        /// </summary>
        /// <param name="cosmosClient"></param>
        /// <param name="container"></param>
        /// <param name="sqlTxt"></param>
        /// <param name="code"></param>
        /// <returns></returns>
        public static async Task<int> GetValueInt(CosmosClient cosmosClient, string container, string sqlTxt = null, string code = null)
        {
            int totals = 0;
            string sql = $"select value(count(c.id)) from c";
            if (!string.IsNullOrEmpty(sqlTxt))
            {
                sql = sqlTxt;
            }

            await foreach (var itemInt in cosmosClient.GetContainer("TEAMModelOS", container).GetItemQueryIteratorSql<int>(queryText: sql, requestOptions: string.IsNullOrEmpty($"{code}") ? new QueryRequestOptions() { } : new QueryRequestOptions() { PartitionKey = new PartitionKey($"{code}") }))
            {
                totals += itemInt;
            }

            return totals;
        }

        /// <summary>   
        /// 返回字符集合
        /// </summary>
        /// <param name="cosmosClient"></param>
        /// <param name="containerId"></param>
        /// <param name="sqlTxt"></param>
        /// <param name="code"></param>
        /// <returns></returns>
        public static async Task<List<string>> GetListString(CosmosClient cosmosClient, string containerId, string sqlTxt, string code = null) 
        {
            List<string> strList = new();

            await foreach (var itemInt in cosmosClient.GetContainer("TEAMModelOS", containerId).GetItemQueryIteratorSql<string>(queryText: sqlTxt, requestOptions: string.IsNullOrEmpty($"{code}") ? new QueryRequestOptions() { } : new QueryRequestOptions() { PartitionKey = new PartitionKey($"{code}") }))
            {
                strList.Add(itemInt);
            }

            return strList;
        }
    }
}