Question

我正在使用NHibernate（Fluently）与数据库进行交互，该数据库被锁定到除应用程序角色之外的所有数据库。

我可以使用存储过程直接在SQL Server Management Studio中使用应用程序角色： sp_setapprole 和 sp_unsetapprole 但尝试执行此操作时遇到问题与NHibernate合作：

using (var session = SessionFactory.OpenSession())
{
    byte[] cookie;

    // 1) Set the application role
    using (var command = session.Connection.CreateCommand())
    {
        command.CommandType = CommandType.StoredProcedure;
        command.CommandText = "sp_setapprole";
        command.Parameters.Add(new SqlParameter("@rolename", RoleName));
        command.Parameters.Add(new SqlParameter("@password", RolePassword));
        command.Parameters.Add(new SqlParameter("@fCreateCookie", true));
        command.Parameters.Add(
            new SqlParameter
                {
                    ParameterName = "@cookie",
                    DbType = DbType.Binary,
                    Direction = ParameterDirection.Output,
                    Size = 8000
                });

        // This works perfectly fine
        command.ExecuteNonQuery();
        var outVal = (SqlParameter)command.Parameters["@cookie"];

        // This returns a byte array value for the cookie generated
        cookie = (byte[])outVal.Value;
    }

    // 2) This line dutifully retreives my contrived Person object but renders
    //    part 3) of this saga next to useless because it performs a logout
    //    (identified with SQL Profiler) and the app role/cookie is forgotten.
    session.CreateCriteria(typeof(Person)).List<Person>().FirstOrDefault();

    // 3) Unset the application role
    using (var command = session.Connection.CreateCommand())
    {
        command.CommandType = CommandType.StoredProcedure;
        command.CommandText = "sp_unsetapprole";
        command.Parameters.Add(new SqlParameter("@cookie", cookie));
        command.ExecuteNonQuery();
    }
}

总而言之，在上面的示例中执行1）和3）之间的任何操作都会导致应用程序崩溃。取决于我是否合并，以两种不同的方式：

合并 - 当前命令发生严重错误。结果（如果有的话）应该被丢弃。

不合并 - 无法取消设置应用程序角色，因为没有设置或cookie无效。

我不确定这是否有帮助，但我正在使用以下内容进行连接（通常情况略有不同，但我需要配置Pooling属性）：

private static ISessionFactory CreateSessionFactory()
{
    return Fluently.Configure().Database(
        MsSqlConfiguration.MsSql2008.ConnectionString(
            "Data Source=(local);" +
            "Initial Catalog=PeopleDB;" +
            "Integrated Security=False;" +
            "User ID=someuserid;" +
            "Password=somepassword" +
            "Pooling=False"))
        .Mappings(m => m.FluentMappings.AddFromAssemblyOf<Program>())
        .BuildSessionFactory();
}

仅供参考，在上述代码中连接的用户对数据库完全没有权限，这至少证明sp_setapprole正在工作。

有没有人遇到这个并设法解决它？我有一个好的谷歌，但没有任何结果。

提前致谢，罗布

Answer 1

NH使用IConnectionProvider在需要时创建和关闭连接。如果应用程序中每个使用过的连接都使用相同的身份验证过程，那么最好实现自己的IConnectionProvider并使用MsSqlConfiguration.MsSql2008.Provider<YourOwnConnectionProvider>()进行配置;

class MyConnectionProvider : DriverConnectionProvider
{
    public override IDbConnection GetConnection()
    {
        var connection = base.GetConnection();

        byte[] cooky;
        // TODO: authenticate

        return new MyConnectionWrapper(connection, cooky);
    }

    public override void CloseConnection(IDbConnection conn)
    {
        var myConn = conn as MyConnectionWrapper;
        if (myConn != null)
        {
            // TODO: deregister with myConn.Cooky;
        }
        base.CloseConnection(conn);
    }
}

将NHibernate与SQL Server应用程序角色一起使用

1 个答案: