c# - 使用 linq 从列表中的 2 个可能字段中选择不同的 int

Question

我一直在寻找解决方案，但我不太明白。令我震惊的是 linq 应该能够巧妙地处理这个问题，但我并不是最擅长制定 linq 查询。基本上我有课；

public class Message
{
    public int CreatedByPersonID { get; set; }
    public int? PostedToPersonID { get; set; }
}

我正在寻找的是 linq 等价物；

List<int> personIDs = new List<int>();
foreach (Message message in messages)
{
    if (!personIDs.Contains(message.CreatedByPersonID))
    {
        personIDs.Add(message.CreatedByPersonID);
    }
    if (message.PostedToPersonID != null && !personIDs.Contains(message.PostedToPersonID.Value))
    {
        personIDs.Add(message.PostedToPersonID.Value);
    }
}

有这个的linq版本吗？还是我对技术的期望太高了……？我知道我正在寻找某个版本，Select()但我不知道如何distinct用多个字段来制定

Answer 1

List<int> personIDs = 
     messages.SelectMany(m => m.PostedToPersonID.HasValue ?
                  new int[] { m.PostedToPersonID.Value, m.CreatedByPersonID } :
                  new int[] { m.CreatedByPersonID })
             .Distinct()
             .ToList();

另一种选择是枚举两次：

List<int> personIDs =
    messages.Where(m => m.PostedToPersonID.HasValue)
            .Select(m => m.PostedToPersonID.Value)
            .Concat(messages.Select(m => m.CreatedByPersonID))
            .Distinct()
            .ToList();

但可能没有 Linq，您的代码可读性更高、速度更快

public IEnumerable<int> GetAllPersonIdsFrom(IEnumerable<Message> messages)
{
     foreach(var message in messages)
     {
         yield return message.CreatedByPersonID;

         if (message.PostedToPersonID.HasValue)
             yield return message.PostedToPersonID.Value;
     }
}

---

另一种解决方案 - 具有不同值的单个枚举

public static List<int> GetDistinctPersonIdsFrom(IEnumerable<Message> messages)
{
    HashSet<int> ids = new HashSet<int>();

    foreach (var message in messages)
    {
        ids.Add(message.CreatedByPersonID);
        if (message.PostedToPersonID.HasValue)
            ids.Add(message.PostedToPersonID.Value);
    }

    return ids.ToList();
}