使用NHibernate与SQL Server应用程序角色

本文关键字:应用程序 角色 Server SQL NHibernate 使用 | 更新日期: 2023-09-27 18:17:45

我正在使用NHibernate(流利地)与数据库进行交互,该数据库被锁定为除应用程序角色外的所有角色。

我能够使用SQL Server管理工作室中的应用程序角色使用存储过程:sp_setapprolesp_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)之间的任何操作都会导致应用程序崩溃。以两种不同的方式,这取决于我是否使用池:

Pooling - 当前命令出现严重错误。如果有,则应丢弃。

Not pooling - 不能取消应用程序角色,因为没有设置或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是正常工作的。

有人遇到这个问题并设法解决它吗?我在谷歌上搜了半天,但什么也没搜到。

提前感谢,Rob

使用NHibernate与SQL Server应用程序角色

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);
    }
}