带有更新锁的 ServiceStack OrmLite Select

Question

我有以下代码：

protected static void InsertOrUpdate<T>(
    IDbConnection connection,
    T item,
    Expression<Func<T, bool>> singleItemPredicate,
    Expression<Func<T, object>> updateOnlyFields = null)
{
    var type = item.GetType();
    var idProperty = type.GetProperty("Id");
    var currentId = (int)idProperty.GetValue(item);
    if (currentId != 0)
    {
        throw new ApplicationException("Cannot insert or update with non-zero ID");
    }

    var existingItem = connection.Select(singleItemPredicate).SingleOrDefault();
    if (existingItem == null)
    {
        Insert(connection, item);
        return;
    }

    var existingId = (int)idProperty.GetValue(existingItem);
    idProperty.SetValue(item, existingId);

    var affectedRowCount = connection.UpdateOnly(item, onlyFields: updateOnlyFields, where: singleItemPredicate);
    if (affectedRowCount != 1)
    {
        throw new ApplicationException("Update failed");
    }
}

但是使用默认隔离级别，这可能会在多个线程处理同一个表时产生错误。

我想要的是select-row来指定我想要更新锁，例如：

var existingItem = connection.Select(singleItemPredicate).WithUpdateLock().SingleOrDefault();

• 对于 SQL Server，这将在 SELECT 行的末尾添加WITH UPDLOCK

在 ServiceStack OrmLite 中，我如何最好地解决这个问题？

Answer 1

您可以使用刚刚在 v4.5.5 中添加的新 SqlExpression SQL Filter现在（可在 MyGet 上使用）来自定义生成的 SQL，然后执行自定义的 Typed SqlExpression，例如：

var q = connection.From<T>()
    .Where(singleItemPredicate)
    .WithSqlFilter(sql => sql + " WITH UPDLOCK");

var existingItem = connection.Select(q).SingleOrDefault();
if (existingItem == null)
{
    connection.Insert(item);
    return;
}