using System.Collections.Concurrent;
using NewLife.Collections;
using NewLife.Reflection;
using NewLife.Serialization;
namespace System.Collections.Generic
{
/// <summary>集合扩展</summary>
public static class CollectionHelper
{
///// <summary>集合转为数组,加锁确保安全</summary>
///// <typeparam name="T"></typeparam>
///// <param name="collection"></param>
///// <param name="index">数组偏移量。大于0时,新数组将空出来前面一截,把数据拷贝到后面</param>
///// <returns></returns>
//[Obsolete("index参数晦涩难懂")]
//public static T[] ToArray<T>(this ICollection<T> collection, Int32 index)
//{
// if (collection == null) return null;
// lock (collection)
// {
// var count = collection.Count;
// if (count == 0) return Array.Empty<T>();
// var arr = new T[count + index];
// collection.CopyTo(arr, index);
// return arr;
// }
//}
/// <summary>集合转为数组,加锁确保安全</summary>
/// <typeparam name="T"></typeparam>
/// <param name="collection"></param>
/// <returns></returns>
public static T[] ToArray<T>(this ICollection<T> collection)
{
if (collection == null) return null;
lock (collection)
{
var count = collection.Count;
if (count == 0) return Array.Empty<T>();
var arr = new T[count];
collection.CopyTo(arr, 0);
return arr;
}
}
/// <summary>集合转为数组</summary>
/// <typeparam name="TKey"></typeparam>
/// <typeparam name="TValue"></typeparam>
/// <param name="collection"></param>
/// <param name="index"></param>
/// <returns></returns>
public static IList<TKey> ToKeyArray<TKey, TValue>(this IDictionary<TKey, TValue> collection, Int32 index = 0)
{
if (collection == null) return null;
if (collection is ConcurrentDictionary<TKey, TValue> cdiv) return cdiv.Keys as IList<TKey>;
if (collection.Count == 0) return Array.Empty<TKey>();
lock (collection)
{
var arr = new TKey[collection.Count - index];
collection.Keys.CopyTo(arr, index);
return arr;
}
}
/// <summary>集合转为数组</summary>
/// <typeparam name="TKey"></typeparam>
/// <typeparam name="TValue"></typeparam>
/// <param name="collection"></param>
/// <param name="index"></param>
/// <returns></returns>
public static IList<TValue> ToValueArray<TKey, TValue>(this IDictionary<TKey, TValue> collection, Int32 index = 0)
{
if (collection == null) return null;
if (collection is ConcurrentDictionary<TKey, TValue> cdiv) return cdiv.Values as IList<TValue>;
if (collection.Count == 0) return Array.Empty<TValue>();
lock (collection)
{
var arr = new TValue[collection.Count - index];
collection.Values.CopyTo(arr, index);
return arr;
}
}
/// <summary>目标匿名参数对象转为字典</summary>
/// <param name="target"></param>
/// <returns></returns>
public static IDictionary<String, Object> ToDictionary(this Object target)
{
//!! 即使传入为空,也返回字典,而不是null,避免业务层需要大量判空
//if (target == null) return null;
if (target is IDictionary<String, Object> dic) return dic;
dic = new NullableDictionary<String, Object>(StringComparer.OrdinalIgnoreCase);
if (target != null)
{
// 修正字符串字典的支持问题
if (target is IDictionary dic2)
{
foreach (DictionaryEntry item in dic2)
{
dic[item.Key + ""] = item.Value;
}
}
else
{
foreach (var pi in target.GetType().GetProperties(true))
{
var name = SerialHelper.GetName(pi);
dic[name] = target.GetValue(pi);
}
}
}
return dic;
}
/// <summary>合并字典参数</summary>
/// <param name="dic">字典</param>
/// <param name="target">目标对象</param>
/// <param name="overwrite">是否覆盖同名参数</param>
/// <param name="excludes">排除项</param>
/// <returns></returns>
public static IDictionary<String, Object> Merge(this IDictionary<String, Object> dic, Object target, Boolean overwrite = true, String[] excludes = null)
{
var exs = excludes != null ? new HashSet<String>(StringComparer.OrdinalIgnoreCase) : null;
foreach (var item in target.ToDictionary())
{
if (exs == null || !exs.Contains(item.Key))
{
if (overwrite || !dic.ContainsKey(item.Key)) dic[item.Key] = item.Value;
}
}
return dic;
}
/// <summary>转为可空字典</summary>
/// <typeparam name="TKey"></typeparam>
/// <typeparam name="TValue"></typeparam>
/// <param name="collection"></param>
/// <param name="comparer"></param>
/// <returns></returns>
public static IDictionary<TKey, TValue> ToNullable<TKey, TValue>(this IDictionary<TKey, TValue> collection, IEqualityComparer<TKey> comparer = null)
{
if (collection == null) return null;
if (collection is NullableDictionary<TKey, TValue> dic && (comparer == null || dic.Comparer == comparer)) return dic;
return new NullableDictionary<TKey, TValue>(collection, comparer);
}
/// <summary>从队列里面获取指定个数元素</summary>
/// <typeparam name="T"></typeparam>
/// <param name="collection">消费集合</param>
/// <param name="count">元素个数</param>
/// <returns></returns>
public static IEnumerable<T> Take<T>(this Queue<T> collection, Int32 count)
{
if (collection == null) yield break;
while (count-- > 0 && collection.Count > 0)
{
yield return collection.Dequeue();
}
}
/// <summary>从消费集合里面获取指定个数元素</summary>
/// <typeparam name="T"></typeparam>
/// <param name="collection">消费集合</param>
/// <param name="count">元素个数</param>
/// <returns></returns>
public static IEnumerable<T> Take<T>(this IProducerConsumerCollection<T> collection, Int32 count)
{
if (collection == null) yield break;
while (count-- > 0 && collection.TryTake(out var item))
{
yield return item;
}
}
}
}
|