如何在实体框架中获取新条目的主键并设置 FK 关系

Question

在我的 ASP.NET MVC 5 应用程序中，我需要获取为数据库中的特定条目生成的主键。此外，我需要获取该 ID 并在数据库中设置外键关系。我尝试了以下尝试，它似乎正在工作，但我不得不致电_context.SaveChanges() TWICE。

UI 行为：用户有一个下拉列表，其中包含推荐形式的公司列表。当未找到公司时，用户将在下拉列表中选择“其他”。它将显示一个文本框，用户将在其中输入公司名称。所以我需要将此公司添加到数据库中并将其链接到推荐行。

表Referral和Company之间的关系：

 dBModelBuilder.Entity<Company>()
            .HasMany(u => u.Referrals)
            .WithRequired(u => u.Company)
            .HasForeignKey(u => u.CompanyId);

模型类：

public class Company
{
    [Key]
    public int CompanyId { get; set; }
    public string CompanyName { get; set; }

    public ICollection<Referral> Referrals { get; set; }
    public ICollection<CoverLetter> CoverLetters { get; set; }
}

public class Referral
{
    [Key]
    public int ReferralId { get; set; }

    [Display(Name = "REFERRALS")]
    public string ReferralName { get; set; }

    public int? CoverLetterId { get; set; }
    public virtual CoverLetter CoverLetter { get; set; }

    public int CompanyId { get; set; }
    public virtual Company Company { get; set; }
}

视图模型：

public class Referral
{
    [Key]
    public int ReferralId { get; set; }

    [Display(Name = "REFERRALS")]
    public string ReferralName { get; set; }

    public int? CoverLetterId { get; set; }
    public virtual CoverLetter CoverLetter { get; set; }

    public int CompanyId { get; set; }
    public virtual Company Company { get; set; }
}

我在控制器操作中的尝试：

    [HttpPost]
    [ValidateAntiForgeryToken]
    public ActionResult Create(ReferralViewModel viewModel)
    {
        var candidateId = User.Identity.GetUserId();

        var referral = new Referral
        {
            ReferralName = viewModel.ReferralName,
        };

        if (viewModel.CompanyId.HasValue)
        // if it is option 4 then we need to add it to the company table
        {
            if (!string.IsNullOrEmpty(viewModel.TempCompany))
            {
                var f = new Company
                {
                    CompanyName = viewModel.TempCompany
                };

                _context.Companies.Add(f);
               // ********FIRST CALL
                 _context.SaveChanges(); 

                _context.Referrals.Add(referral);
                 referral.CompanyId = f.CompanyId;
               // **********SECOND CALL
                _context.SaveChanges(); // SECOND CALL ------
            }
            else
            {
                referral.CompanyId = viewModel.CompanyId.Value;
                _context.Referrals.Add(referral);
                _context.SaveChanges();
            }
        }

        return RedirectToAction("ReferralCenter");
    }

问题：我可以一次拨打_context.SaveChanges(); 完成这些步骤吗？

编辑

使用此代码，我得到一个NullReferenceException：

if (!string.IsNullOrEmpty(viewModel.TempCompany))
{
    var f = new Company
                {
                    CompanyName = viewModel.TempCompany
                };

    f.Referrals.Add(referral); // Referrals is NULL(EXCEPTION)
    _context.Companies.Add(f);
    // _context.SaveChanges();
    //_context.Referrals.Add(referral);
    // referral.CompanyId = f.CompanyId;
    _context.SaveChanges();
}

Answer 1

据我了解，FK 约束是在 EF 中管理的，只要它们在 DB 中进行配置。

如果在您的 EF 模型中，Referral 是 Company 的子代，则应该有一个可从 Company 对象实例访问的引荐集合。

更新 我刚刚构建了一个简单的示例 MVC Proj，2 个 SQL 表（CompanyId 上的推荐人和公司 FK）4 个公司，第 4 个是 OTHER，使用反向 POCO 构建了一个 EF 模型以提高速度，添加了一个带有脚手架和视图的控制器，并将我的控制器修改为下面。 （我只是插入日期时间而不是 tempCompanyName。这里生成的语法更像@ashin 的答案，但它对我有用。

    [HttpPost]
    [ValidateAntiForgeryToken]
    public ActionResult Create([Bind(Include = "ReferralId,CompanyId")] Referral referral)
    {
        if (ModelState.IsValid)
        {
            if (referral.CompanyId == 4)
            {
                var f = new Company() { Name = DateTime.Now.ToString() };
                referral.Company = f;
            }
            db.Referrals.Add(referral);
            db.SaveChanges();
            return RedirectToAction("Index");
        }

        ViewBag.CompanyId = new SelectList(db.Companies, "CompanyId", "Name", referral.CompanyId);
        return View(referral);
    }

我的公司模型确实实例化了一个新的推荐集合

    public Company()
    {
        Referrals = new System.Collections.Generic.List<Referral>();
    } 

Answer 2

如果您定义了导航属性，您可以一次性保存公司和推荐人，如下所示：

if (!string.IsNullOrEmpty(viewModel.TempCompany))
{
   var f = new Company
   {
        CompanyName = viewModel.TempCompany
   };
   referral.Company = f;
   _context.SaveChanges();
}

EF 将负责设置 FK。

编辑

已更新代码以包含 cmets 中提到的 _context.Referrals.Add(referral)：

if (!string.IsNullOrEmpty(viewModel.TempCompany))
{
   var f = new Company
   {
        CompanyName = viewModel.TempCompany
   };

   referral.Company = f;
   _context.Referrals.Add(referral);
   _context.SaveChanges();
}