这可以在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
库。
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
值只能为A
或B
,则可以替换为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);
}