SqlScriptDOM/SqlScriptDom/ScriptDom/SqlServer/ScriptGenerator/SqlScriptGeneratorVisitor.IfStatement.cs at 692c7877a56d01d1edf878bc6ebd29091e5ef733 · microsoft/SqlScriptDOM · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
//------------------------------------------------------------------------------
// <copyright file="SqlScriptGeneratorVisitor.IfStatement.cs" company="Microsoft">
//         Copyright (c) Microsoft Corporation.  All rights reserved.
// </copyright>
//------------------------------------------------------------------------------
using Microsoft.SqlServer.TransactSql.ScriptDom;

namespace Microsoft.SqlServer.TransactSql.ScriptDom.ScriptGenerator
{
    partial class SqlScriptGeneratorVisitor
    {
        public override void ExplicitVisit(IfStatement node)
        {
            AlignmentPoint branchStatements = new AlignmentPoint();

            GenerateKeyword(TSqlTokenType.If);
            GenerateSpaceAndFragmentIfNotNull(node.Predicate);

            NewLineAndIndent();
            MarkAndPushAlignmentPoint(branchStatements);
            GenerateStatementWithSemiColon(node.ThenStatement);
            PopAlignmentPoint();

            if (node.ElseStatement != null)
            {
                NewLine();
                GenerateKeyword(TSqlTokenType.Else);
                NewLineAndIndent();
                MarkAndPushAlignmentPoint(branchStatements);
                GenerateStatementWithSemiColon(node.ElseStatement);
                PopAlignmentPoint();
            }
        }
    }
}