@@ -116,10 +116,11 @@ public virtual string BuildInsert(InsertStatement insertStatement)
116116
117117 if ( insertStatement . OutputExpressions ? . Count > 0 )
118118 {
119+
119120 insertBuilder
120121 . AppendLine ( )
121122 . Append ( "OUTPUT " )
122- . AppendJoin ( ", " , insertStatement . OutputExpressions . Select ( ColumnExpression ) ) ;
123+ . AppendJoin ( ", " , insertStatement . OutputExpressions . Select ( c => ColumnExpression ( c , "INSERTED" ) ) ) ;
123124 }
124125
125126 insertBuilder
@@ -163,7 +164,7 @@ public virtual string BuildUpdate(UpdateStatement updateStatement)
163164 updateBuilder
164165 . AppendLine ( )
165166 . Append ( "OUTPUT " )
166- . AppendJoin ( ", " , updateStatement . OutputExpressions . Select ( ColumnExpression ) ) ;
167+ . AppendJoin ( ", " , updateStatement . OutputExpressions . Select ( c => ColumnExpression ( c , "INSERTED" ) ) ) ;
167168 }
168169
169170 if ( updateStatement . FromExpressions ? . Count > 0 )
@@ -224,7 +225,7 @@ public virtual string BuildDelete(DeleteStatement deleteStatement)
224225 deleteBuilder
225226 . AppendLine ( )
226227 . Append ( "OUTPUT " )
227- . AppendJoin ( ", " , deleteStatement . OutputExpressions . Select ( ColumnExpression ) ) ;
228+ . AppendJoin ( ", " , deleteStatement . OutputExpressions . Select ( c => ColumnExpression ( c , "DELETED" ) ) ) ;
228229 }
229230
230231 if ( deleteStatement . FromExpressions ? . Count > 0 )
@@ -538,4 +539,14 @@ public virtual string ParseIdentifier(string name)
538539 return name ;
539540 }
540541
542+
543+ private string ColumnExpression ( ColumnExpression columnExpression , string tableAlias )
544+ {
545+ var column = columnExpression with
546+ {
547+ TableAlias = columnExpression . TableAlias ?? tableAlias
548+ } ;
549+
550+ return ColumnExpression ( column ) ;
551+ }
541552}
0 commit comments