IUserType上的NHibernate QueryOver

时间:2012-01-09 23:59:29

标签: nhibernate queryover iusertype

首先让我为这篇文章的篇幅道歉,这主要是代码,所以我希望你们都忍受我!

我有一个处理遗留数据库的场景,我需要使用NHibernate 3.2编写IUserType来获取2个字符的“status”字段并从中返回一个布尔值。状态字段可以包含3个可能的值:

* 'DI'     // 'Disabled', return false
* '  '     // blank or NULL, return true
* NULL     

这是我简化的内容。

表定义:

CREATE TABLE [dbo].[Client](
    [clnID] [int] IDENTITY(1,1) NOT NULL,
    [clnStatus] [char](2) NULL,
    [clnComment] [varchar](250) NULL,
    [clnDescription] [varchar](150) NULL,
    [Version] [int] NOT NULL
)

Fluent Mapping:

public class ClientMapping : CoreEntityMapping<Client>
{
    public ClientMapping()
    {
        SchemaAction.All().Table("Client");
        LazyLoad();

        Id(x => x.Id, "clnId").GeneratedBy.Identity(); 
        Version(x => x.Version).Column("Version").Generated.Never().UnsavedValue("0").Not.Nullable();
        OptimisticLock.Version();

        Map(x => x.Comment, "clnComment").Length(250).Nullable();
        Map(x => x.Description, "clnDescription").Length(250).Nullable();
        Map(x => x.IsActive, "clnStatus").Nullable().CustomType<StatusToBoolType>();
    }
}

我的IUserType实施:

public class StatusToBoolType : IUserType
{
    public bool IsMutable { get { return false; } }
    public Type ReturnedType { get { return typeof(bool); } }
    public SqlType[] SqlTypes { get {  return new[] { NHibernateUtil.String.SqlType }; } }

    public object DeepCopy(object value)
    {
        return value;
    }
    public object Replace(object original, object target, object owner)
    {
        return original;
    }
    public object Assemble(object cached, object owner)
    {
        return cached;
    }
    public object Disassemble(object value)
    {
        return value;
    }

    public new bool Equals(object x, object y)
    {
        if (ReferenceEquals(x, y)) return true;
        if (x == null || y == null) return false;
           return x.Equals(y);
    }
    public int GetHashCode(object x)
    {
        return x == null ? typeof(bool).GetHashCode() + 473 : x.GetHashCode();
    }

    public object NullSafeGet(IDataReader rs, string[] names, object owner)
    {
        var obj = NHibernateUtil.String.NullSafeGet(rs, names[0]);
        if (obj == null) return true;

        var status = (string)obj;
        if (status == "  ") return true;
        if (status == "DI") return false;
        throw new Exception(string.Format("Expected data to be either empty or 'DI' but was '{0}'.", status));
    }

    public void NullSafeSet(IDbCommand cmd, object value, int index)
    {
        var parameter = ((IDataParameter) cmd.Parameters[index]);
        var active = value == null || (bool) value;
        if (active)
            parameter.Value = "  ";
        else
            parameter.Value = "DI";
    }
}

然而,这不起作用。此单元测试失败,计数不准确。

[TestMethod]
public void GetAllActiveClientsTest()
{
    //ACT
    var count = Session.QueryOver<Client>()
        .Where(x => x.IsActive)
        .SelectList(l => l.SelectCount(x => x.Id))
        .FutureValue<int>().Value;

    //ASSERT
    Assert.AreNotEqual(0, count);
    Assert.AreEqual(1721, count);
}

失败的原因是它生成以下SQL:

SELECT count(this_.clnID) as y0_ FROM Client this_ WHERE this_.clnstatus = @p0;
/* @p0 = '  ' [Type: String (0)] */

但我需要它来生成这个:

SELECT count(this_.clnID) as y0_ FROM Client this_ WHERE (this_.clnstatus = @p0 <b> OR this_.clnstatus IS NULL);</b>

经过一些调试后,我发现我的StatusToBoolType类中的NullSafeSet()方法在生成查询之前被调用,所以我能够通过在该方法中编写一些hackish代码来操作cmd中的SQL来解决这个问题。 CommandText属性。

...
public void NullSafeSet(IDbCommand cmd, object value, int index)
{
    var parameter = ((IDataParameter) cmd.Parameters[index]);
    var active = value == null || (bool) value;
    if (active)
    {
        parameter.Value = "  ";

        if (cmd.CommandText.ToUpper().StartsWith("SELECT") == false) return;
        var paramindex = cmd.CommandText.IndexOf(parameter.ParameterName);
        if (paramindex > 0)
        {
            // Purpose: change [columnName] = @p0  ==> ([columnName] = @p0 OR [columnName] IS NULL) 
            paramindex += parameter.ParameterName.Length;
            var before = cmd.CommandText.Substring(0, paramindex);
            var after = cmd.CommandText.Substring(paramindex);

            //look at the text before the '= @p0' and find the column name...
            var columnSection = before.Split(new[] {"= " + parameter.ParameterName}, StringSplitOptions.RemoveEmptyEntries).Reverse().First();
            var column = columnSection.Substring(columnSection.Trim().LastIndexOf(' ')).Replace("(", "");
            var myCommand = string.Format("({0} = {1} OR {0} IS NULL)", column.Trim(), parameter.ParameterName);

            paramindex -= (parameter.ParameterName.Length + column.Length + 1);
            var orig = before.Substring(0, paramindex);
            cmd.CommandText = orig + myCommand + after;
        }
    }
    else
        parameter.Value = "DI";
}

但这是NHibernate!像这样破解sql语句不可能是处理这个问题的正确方法吗?正确?

因为它是一个共享的遗留数据库,所以我无法将表模式更改为NOT NULL,否则我就会这样做,并避免这种情况。

所以最后在所有这个前奏之后我的问题就是这个,我在哪里可以告诉NHibernate为这个IUserType生成自定义SQL条件语句?

提前谢谢大家!

1 个答案:

答案 0 :(得分:2)

解决了!

在我发布我的问题之后,我又回到了绘图板,我想出了一个解决方案,它不需要在IUserType实现中破解生成的SQL。事实上,这个解决方案根本不需要IUserType!

这就是我所做的。

首先,我更改了IsActive列以使用公式来处理空检查。这解决了QueryOver失败的问题,因为现在每次NHibernate处理IsActive属性时它都会注入我的sql公式来处理null。

这种方法的缺点是,在我输入公式后,我的所有保存测试都失败了。事实证明,公式属性是有效的ReadOnly属性。

因此,为了解决这个问题,我向实体添加了一个protected属性来保存数据库中的状态值。

接下来,我更改了IsActive属性,将protected status属性设置为“”或“DI”。最后我将FluentMapping更改为将受保护的Status属性显示为NHibernate,以便NHibernate可以跟踪它。现在NHibernate知道Status,它可以将它包含在INSERT / UPDATE语句中。

我将在下面提供我的解决方案,以防其他人感兴趣。

客户端类

public class Client 
{
    ...

    protected virtual string Status { get; set; }
    private bool _isActive;
    public virtual bool IsActive
    {
        get { return _isActive; }
        set
        {
            _isActive = value;
            Status = (_isActive) ? "  " : "DI";
        }
    }
}

Fluent Mapping的更改

public class ClientMapping : CoreEntityMapping<Client>
{
    public ClientMapping()
    {
        ....

        Map(Reveal.Member<E>("Status"), colName).Length(2);
        Map(x => x.IsActive).Formula("case when clnStatus is null then '  ' else clnStatus end");
    }
}