将带有参数的函数作为参数传递并返回通用类型

时间:2020-02-21 03:17:56

标签: c# generics lambda sqldatareader

在下面的代码中,我试图将函数func传递给GetData函数。这将使用阅读器对象并将其映射到通用对象。

我希望将GetData对象类型与将数据映射到该对象类型的函数一起传递,这样我就不必重复打开/关闭/释放连接。

这可能吗?或者有人有其他建议吗?

public T GetData<T>(string cmdText,Func<T> func)
{
    using (SqlConnection conn = new SqlConnection(connectionStringBuilder.ConnectionString))
    {
        using (SqlCommand cmd = new SqlCommand(cmdText, conn))
        {
            SqlDataReader reader = cmd.ExecuteReader();
            //return func(reader);
            //  WITHIN THE FUNC FUNCTION:
            //  while (reader.Read())
            //  {
            //  Map function to T e.g
            //  T.property = reader["column"];
            //  Return T
            //  }
        }
    }
}

1 个答案:

答案 0 :(得分:0)

您正在寻找的签名是:

T GetData<T>(string cmdText, Func<SqlDataReader, T> func)

然后您可以继续编写函数,如下所示:

public T GetData<T>(string cmdText, Func<SqlDataReader, T> func)
{
    using (var conn = new SqlConnection(connectionStringBuilder.ConnectionString))
    {
        using (var cmd = new SqlCommand(cmdText, conn))
        {
            var reader = cmd.ExecuteReader();
            return func(reader);
        }
    }
}

您将像这样使用它:

var result = GetData("select * from Foo", dr =>
{
    while (dr.Read())
    {
        return new { property = dr["column"] };
    }
    throw new DataException();
});

现在,这取决于您说的在问题中的使用方式。

但是,在拆分实现时,您对函数的使用有些麻烦-部分在GetData中,而部分在调用代码中。

最好使用此签名:

IEnumerable<T> GetData<T>(string cmdText, Func<SqlDataReader, T> func)

现在您可以编写如下方法:

public IEnumerable<T> GetData<T>(string cmdText, Func<SqlDataReader, T> func)
{
    using (var conn = new SqlConnection(connectionStringBuilder.ConnectionString))
    {
        using (var cmd = new SqlCommand(cmdText, conn))
        {
            var reader = cmd.ExecuteReader();
            while (reader.Read())
            {
                yield return func(reader);
            }
        }
    }
}

现在的优点是调用代码更加简单:

var results = GetData("select * from Foo", dr => new { property = dr["column"] });

这将返回查询返回的数据行。

如果您知道调用代码仅返回一个值,则可以在方法调用的末尾放一个.Single(),以确保只得到一个结果。