这可以在c#中简化吗?

本文关键字: | 更新日期: 2023-09-27 18:11:46

我有一些重复的代码,但不确定最好的方法来简化它。

private void CheckData(long PKID, int ExpectedResult, string Server)
{
    var a = _ARepo.GetAll();
    var b = _BRepo.GetAll();
    if(Server == "A")
    {
        a.Find(x => x.PKID == PKID).Result.ShouldBe(ExpectedResultId);
    }
    if (Server == "B")
    {
        b.Find(x => x.PKID == PKID).Result.ShouldBe(ExpectedResultId);
    }
}

这是一个单元测试项目,我使用Shouldly库。

这可以在c#中简化吗?

private void CheckData(long PKID, int ExpectedResult, string Server)
{
    //Setting to empty because I don't know what happens if it is not "A" nor "B"
    IEnumerable<YourType> data = Enumerable.Empty<YourType>();
    if(Server == "A")
        data = _ARepo.GetAll();
    else if(Server == "B")
        data = _BRepo.GetAll();
    data.Find(Find(x => x.PKID == PKID).Result.ShouldBe(expectedId);
}

如果Server值只能为AB,则可以替换为if else或更好的?:操作符然后它看起来像:

var data = Server == "A" ? _ARepo.GetAll() : _BRepo.GetAll();
data.Find(Find(x => x.PKID == PKID).Result.ShouldBe(expectedId);

在两个Repo实现相同接口的情况下,更好的设计将是将IRepo作为函数的参数。这样,该函数只有一个角色——检查数据(而不是决定检查哪些数据)

您可以创建一个辅助方法来处理这种重复:

private void CheckFind<T>(List<T> a, int expectedId) where T : IWithId {
    a.Find(x => x.PKID == PKID).Result.ShouldBe(expectedId);
}

CheckData:

调用此方法两次
if(Server == "A") {
    CheckFind(_ARepo.GetAll(), ExpectedResultId);
}
if(Server == "B") {
    CheckFind(_BRepo.GetAll(), ExpectedResultId);
}

如果所有Repos实现相同的接口(如IRepo)。您可以创建一个像这样的辅助方法

    private IRepo GetRepo(string server)
    {
        var repos = new Dictionary<string, IRepo>
        {
            { "A", _ARepo },
            { "B", _BRepo }
        };
        return repos[server];
    }

用法很简单

    private void CheckData(long PKID, int ExpectedResultId, string Server)
    {
        var repo = GetRepo(Server);
        repo.GetAll().Find(x => x.PKID == PKID).Result.ShouldBe(ExpectedResultId);
    }

是的,如果你可以让代码重构你的代码,你的_ARepo.GeAll_BRepo.getAll在返回类型中实现相同的接口,这将是如下所示

private void CheckData(long PKID, int ExpectedResult, string Server, IRepo repo)
{
    var b = repo.GetAll();  
    b.Find(x => x.PKID == PKID).Result.ShouldBe(ExpectedResultId);
}

使用一些简短的if符号,您可以使用以下方法假设下面的_ARepo和_BRepo来自同一个类,但不同的连接字符串

private void CheckData(long PKID, int ExpectedResult, string Server)
{
   BaseType repo = Server == "A" ? _ARepo : _BRepo;
   repo.GetAll().Find(x => x.PKID == PKID).Result.ShouldBe(ExpectedResultId);
}

假设它们不是相同的BaseType,但GetAll()是IEnumerable

private void CheckData(long PKID, int ExpectedResult, string Server)
{
   IEnumerable<YourType> repo = Server == "A" ? _ARepo.GetAll() : _BRepo.GetAll();
   repo.Find(x => x.PKID == PKID).Result.ShouldBe(ExpectedResultId);
}
相关文章:
  • 没有找到相关文章