泛型 HashSet

Question

我已经找到了一些关于这个主题的信息，但都是关于典型类的——我不知道如何将下面的类转换为通用版本。

我继承了现有代码，该代码使用 SqlHashSet 类为数据库的参数设置整数值。它看起来像这样：

namespace Services
{
    public class SqlHashSet : HashSet<int>, IEnumerable<SqlDataRecord>
    {
        SqlDataRecord ret = new SqlDataRecord(new SqlMetaData("value", SqlDbType.Int));       
        foreach (var data in this)
        {
            ret.SetValue(0, data);
            yield return ret;
        }
    }
}

由于代码的结构，我必须使用 SqlHashSet。我的问题是我需要一个 strings 的 SqlHashSet。

我在想也许我可以在类声明中使用HashSet<T>，但我还必须检查值是 SqlDbType.int 还是 SqlDbType.VarChar 类型

所以，我的问题是：

1. 我可以这样做吗？

2. 我需要做什么才能使用<T>？

3. 也许没有第三个问题；也许我可以在使用泛型类型后逐步执行代码并弄清楚如何确定 SqlDbType 以将值设置为。

Answer 1

你可以添加一个基类：

public abstract class SqlHashSetBase<T> : HashSet<T>, IEnumerable<SqlDataRecord>
{
    protected abstract SqlDbType DbType { get; }

    IEnumerator<SqlDataRecord> IEnumerable<SqlDataRecord>.GetEnumerator()
    {
        SqlDataRecord ret = new SqlDataRecord(new SqlMetaData("value", this.DbType));
        foreach (T data in this)
        {
            ret.SetValue(0, data);
            yield return ret;
        }
    }
}

然后更改您现有的课程：

public class SqlHashSet : SqlHashSetBase<int> {
    protected override SqlDbType DbType {
        get { return SqlDbType.Int; }
    }
}

并添加一个新的：

public class SqlStringHashSet : SqlHashSetBase<string> {
    protected override SqlDbType DbType {
        get { return SqlDbType.NVarChar; }
    }
}

您可能需要考虑将T => SqlDbType 映射提取到它自己的类中并将其作为参数传递，而不是使用抽象方法。您可能还想考虑使用组合，而不是直接从 HashSet<T> 继承。

Answer 2

（请注意，您发布的代码无法编译；您不能直接在类中编写语句，它们必须在方法中）

您无法创建真正的泛型类，因为您需要根据T 的类型使用不同的SqlDbType。但如果你不介意做一些丑陋的事情，你可以这样做：

public class SqlHashSet<T> : HashSet<T>, IEnumerable<SqlDataRecord>
{
    private static readonly SqlDbType _sqlDbType = GetSqlDbType();
    private static SqlDbType GetSqlDbType()
    {
        if (typeof(T) == typeof(int))
            return SqlDbType.Int;
        if (typeof(T) == typeof(string))
            return SqlDbType.String;

        ...

        throw new InvalidOperationException($"Can't find the SqlDbType for {typeof(T)}");
    }

    ...

    SqlDataRecord ret = new SqlDataRecord(new SqlMetaData("value", _sqlDbType));
}

如果您不介意必须为每种值类型创建单独的类的缺点，Lee 的回答中建议的方法类似但更简洁。