如何在使用 NHibernate 进行任何插入之前增加 ID

Question

看起来 NH 仅获取 MAX(ID) 一次，首先插入，然后在内部存储该值，这会在其他进程插入数据时给我带来一些问题。然后我没有实际的 ID，并且抛出重复键异常。

让我们假设我们有表 Cats

CREATE TABLE Cats(ID int, Name varchar(25))

然后我们用 FluentNhibernate 完成了相应的映射

public class CatMap : ClassMap<Cat>
{
    public CatMap()
    {
      Id(m=>m.ID).GeneratedBy.Increment();
      Map(m=>.Name);
    }
}

我想要实现的就是使用 SELECT MAX 插入带有 NHibernate 生成的 ID 的 Cat 记录(ID) FROM Cats 在任何之前插入。在任何提交之后执行 Session.Flush 都不起作用。我使用 SQL Server Profiler 进行了一些调查，并且此 sql stetement 仅执行一次（在第一次插入时） - 其他插入不会强制检索实际的 MAX(ID)。我知道像 HiLo 这样的其他算法更好，但我无法取代它。

Answer 1

正如您所发现的，NHibernate Increment id 生成器不适用于多用户环境。您声明使用 HiLo 生成器不是一个选项，因此您只剩下以下选项：

• 使用本机生成器并更改 id 列以使用数据库支持的身份机制

• 使用分配的生成器并编写代码来确定下一个有效 id

• 创建自定义生成器，其中你实现了IIdentifierGenerator 接口来执行您需要的操作

下面是自定义生成器的示例代码它使用通用过程来获取给定表的 ID。这种方法的主要问题是您必须将代码包装在类似 工作单元 模式，以确保“select max(id) ...”和插入由同一数据库事务覆盖。IIdentifierGenerator 链接包含连接此自定义生成器所需的 XML 映射。

using System;
using System.Collections.Generic;
using System.Data;
using NHibernate.Dialect;
using NHibernate.Engine;
using NHibernate.Id;
using NHibernate.Persister.Entity;
using NHibernate.Type;

namespace YourCompany.Stuff
{
    public class IdGenerator : IIdentifierGenerator, IConfigurable
    {
        private string _tableName;
        // The "select max(id) ..." query will go into this proc:
        private const string DefaultProcedureName = "dbo.getId";

        public string ProcedureName { get; protected set; }
        public string TableNameParameter { get; protected set; }
        public string OutputParameter { get; protected set; }

        public IdGenerator()
        {
            ProcedureName = DefaultProcedureName;
            TableNameParameter = "@tableName";
            OutputParameter = "@newID";
        }

        public object Generate(ISessionImplementor session, object obj)
        {
            int newId;
            using (var command = session.Connection.CreateCommand())
            {
                var tableName = GetTableName(session, obj.GetType());

                command.CommandType = CommandType.StoredProcedure;
                command.CommandText = ProcedureName;

                // Set input parameters
                var parm = command.CreateParameter();
                parm.Value = tableName;
                parm.ParameterName = TableNameParameter;
                parm.DbType = DbType.String;

                command.Parameters.Add(parm);

                // Set output parameter
                var outputParameter = command.CreateParameter();
                outputParameter.Direction = ParameterDirection.Output;
                outputParameter.ParameterName = OutputParameter;
                outputParameter.DbType = DbType.Int32;

                command.Parameters.Add(outputParameter);

                // Execute the stored procedure
                command.ExecuteNonQuery();

                var id = (IDbDataParameter)command.Parameters[OutputParameter];

                newId = int.Parse(id.Value.ToString());

                if (newId < 1)
                    throw new InvalidOperationException(
                        string.Format("Could not retrieve a new ID with proc {0} for table {1}",
                                      ProcedureName,
                                      tableName));
            }

            return newId;
        }

        public void Configure(IType type, IDictionary<string, string> parms, Dialect dialect)
        {
            _tableName = parms["TableName"];
        }

        private string GetTableName(ISessionImplementor session, Type objectType)
        {
            if (string.IsNullOrEmpty(_tableName))
            {
                //Not set by configuration, default to the mapped table of the actual type from runtime object:
                var persister = (IJoinable)session.Factory.GetClassMetadata(objectType);

                var qualifiedTableName = persister.TableName.Split('.');
                _tableName = qualifiedTableName[qualifiedTableName.GetUpperBound(0)]; //Get last string
            }

            return _tableName;
        }
    }
}