Entity Framework处理导航属性时遭遇ArgumentNullException问题求助

我来帮你梳理下这个问题的根源，以及对应的解决办法：

问题背景 #

你想要获取指定用户的所有聊天好友ID（即该用户作为发送方或接收方对应的所有用户ID），但当前代码抛出了System.ArgumentNullException: 'Value cannot be null.'异常，尽管数据库中该用户确实存在SentMessages和ReceivedMessages记录。

原代码问题点 #

你的实现代码如下：

public List<string> GetUserMessagesGrouped(string userId) { 
    var a = _context.Users.FirstOrDefault(x => x.Id == userId).SentMessages.Select(m => m.ReceiverID); 
    var b = _context.Users.FirstOrDefault(y => y.Id == userId).ReceivedMessages.Select(m => m.SenderID); 
    var ChatBuddies = a.Concat(b).Distinct(); 
    return ChatBuddies.ToList(); 
}

这里的核心问题有两个：

1. EF Core延迟加载导致导航属性为空：默认情况下EF Core不会自动加载关联的SentMessages和ReceivedMessages导航属性，直接访问时它们会是null，调用Select方法自然会触发空引用异常。
2. 重复数据库查询：你连续调用了两次FirstOrDefault查询同一个用户，造成不必要的性能损耗。

修复后的代码 #

下面是优化后的两种实现方案：

方案1：通过Include加载导航属性 #

public List<string> GetUserMessagesGrouped(string userId) 
{ 
    // 一次性加载用户及其关联的消息记录，避免导航属性为空
    var user = _context.Users
        .Include(u => u.ReceivedMessages)
        .Include(u => u.SentMessages)
        .FirstOrDefault(x => x.Id == userId); 

    var receivedFriendIds = user.ReceivedMessages.Select(r => r.SenderID).ToList(); 
    var sentFriendIds = user.SentMessages.Select(s => s.ReceiverID).ToList(); 

    // Union自动去重，替代Concat+Distinct，语义更清晰
    var chatBuddies = receivedFriendIds.Union(sentFriendIds).ToList(); 
    return chatBuddies; 
}

方案2：直接查询Messages表（性能更优） #

如果不需要用户实体的其他信息，直接查询消息表可以避免加载整个用户对象，效率更高：

public List<string> GetUserMessagesGrouped(string userId) 
{ 
    var receivedFriendIds = _context.Messages
        .Where(m => m.ReceiverID == userId)
        .Select(m => m.SenderID)
        .ToList(); 

    var sentFriendIds = _context.Messages
        .Where(m => m.SenderID == userId)
        .Select(m => m.ReceiverID)
        .ToList(); 

    var chatBuddies = receivedFriendIds.Union(sentFriendIds).ToList(); 
    return chatBuddies; 
}

关键修复说明 #

• 显式加载导航属性：使用强类型的Include(u => u.ReceivedMessages)（比字符串重载更安全，避免拼写错误）确保关联的消息集合被提前加载，不会出现null。
• 减少数据库交互：方案1只查询一次用户实体，方案2直接针对消息表查询，都避免了重复访问数据库。
• 优化去重逻辑：用Union替代Concat+Distinct，因为Union本身就会自动剔除重复项，代码更简洁高效。

内容的提问来源于stack exchange，提问作者nouko