正确生成唯一发票 ID

Question

有人要求我清理其他人的控制器代码，该代码会生成发票，但我遇到了一些我不知道如何解决的问题。有问题的代码如下（这是使用 EF 6: Code First）：

var invid = db.TransportJobInvoice.Where(c => c.CompanyId == CompanyId)
                .Max(i => i.InvoiceId);
var invoiceId = invid == null ? 1 : (int)invid + 1;

代码应该根据为其创建发票的公司生成invoiceId。所以这个小表可能如下所示：

------------------------------
| Id | CompanyId | InvoiceId |
------------------------------
|  1 |         1 |         1 |
------------------------------
|  2 |         1 |         2 |
------------------------------
|  3 |         1 |         3 |
------------------------------
|  4 |         2 |         1 |
------------------------------
|  5 |         2 |         2 |
------------------------------

如您所见，invoiceId 将根据相关公司的当前发票数量生成。但是，我认为建议两个线程可以在评估此行之前执行查询是合理的：

var invoiceId = invid == null ? 1 : (int)invid + 1;

这将导致为两张不同的发票生成相同的invoiceId。

是否有一个简单的解决方案，可能利用 Entity Framework 自动执行此操作？

Answer 1

我建议使用身份作为主键，非常重要！

然后，我会为“CustomerInvoiceID”添加一列，并在 CustomerID 和 CustomerInvoiceID 上放置一个复合唯一键。

然后，创建一个存储过程，在插入字段 CustomerInvoiceID 后填充它，这里是一些伪代码：

CREATE PROCEDURE usp_PopulateCustomerInvoiceID 
    @PrimaryKey INT, --this is your primary key identity column
    @CustomerID INT
AS
BEGIN
    SET NOCOUNT ON;

    DECLARE @cnt INT;

    SELECT @CNT = COUNT(1)
    FROM TBL 
    WHERE CustomerID = @CustomerID
        AND PrimaryKeyColumn <= @PrimaryKey

    UPDATE tbl
    SET CustomerInvoiceID = @cnt + 1
    WHERE PrimaryKeyColumn = @PrimaryKey
END

Answer 2

两种可能：

服务器端：不要在客户端计算 max(ID)+1。相反，作为 INSERT 语句的一部分，通过 INSERT..SELECT 语句计算 max(ID)+1。

客户端：在客户端生成一个 GUID，而不是递增的 int，并将其用作您的 InvoiceID。

Answer 3

如果您在 SQL Server 中使用 Identity 字段，则会自动处理。

Answer 4

我不知道您是否可以自动生成发票 ID，除非它被威胁为外键（我认为不是）。

您的多线程问题可以使用 lock 语句来解决。

lock (myLock)
{
     var invid = db.TransportJobInvoice.Where(c => c.CompanyId == CompanyId)
            .Max(i => i.InvoiceId);
     var invoiceId = invid == null ? 1 : (int)invid + 1;
}

这将保证只有线程在执行这些语句。

但要小心，当这些语句被并行执行很多并且查询需要一些很长的时间来执行时，这可能会导致性能问题。

Answer 5

一种完全不同的方法是为每个CustomerId 创建一个带有NextId 的单独表。随着新客户的添加，您将在此表中添加一个新行。它的优点是即使您允许删除发票，分配给发票的编号也可以保持唯一。

create procedure GetInvoiceIdForCustomer
  @CustomerId as Int,
  @InvoiceId as Int Output
as
begin
  set nocount on

  begin transaction

  update CustomerInvoiceNumbers 
    set @InvoiceId = NextId, NextId += 1
    where CustomerId = @CustomerId

  if @@RowCount = 0
    begin
    set @InvoiceId = 1
    insert into CustomerInvoiceNumbers ( CustomerId, NextId ) values ( @CustomerId, @InvoiceId + 1 )
    end

  commit transaction
end       

end