Генерация динамической команды UPDATE из выражения <Func <T, T >> - PullRequest
Новая
       39

Генерация динамической команды UPDATE из выражения <Func <T, T >>

4 голосов
/ 16 марта 2012

Я пытаюсь сгенерировать команду UPDATE на основе деревьев выражений (для пакетного обновления).

Принимая следующую команду UPDATE: 

UPDATE Product
SET ProductTypeId = 123,
ProcessAttempts = ProcessAttempts + 1

Для выражения, подобного этому: 

Expression<Func<Product, Product>> updateExpression = entity =>
    new Product() {
        ProductTypeId = 123,
        ProcessAttempts = entity.ProcessAttempts + 1
    };

Как мне сгенерировать часть команды SET? 

SET ProductTypeId = 123,
ProcessAttempts = ProcessAttempts + 1

1 Ответ

2 голосов
/ 23 марта 2012

Это очень упрощенный подход, но я надеюсь, что он достаточно хорош: 

private static string ConvertToSetCommand<T>(Expression<Func<T, T>> exp)
{
    if (exp.Body.NodeType != ExpressionType.MemberInit)
    {
        throw new ArgumentException("The expression must have an object initializer.", "exp");
    }

    var builder = new StringBuilder("SET ", 100);
    exp = (Expression<Func<T, T>>) new OmitParametersVisitor().Visit(exp);

    var memberInit = (MemberInitExpression) exp.Body;
    foreach (var assignment in memberInit.Bindings)
    {
        builder.Append(assignment.ToString());
        builder.Append(", ");
    }
    builder.Length -= 2; // Remove the last comma

    return builder.ToString();
}

private class OmitParametersVisitor : ExpressionVisitor
{
    protected override Expression VisitMember(MemberExpression node)
    {
        if (node.Expression != null && node.Expression.NodeType == ExpressionType.Parameter)
        {
            return Expression.Parameter(node.Type, node.Member.Name);
        }
        else
        {
            return base.VisitMember(node);
        }
    }
}
...