SqlScriptDOM/SqlScriptDom/ScriptDom/SqlServer/ScriptGenerator/SqlScriptGeneratorVisitor.IndexDefinition.cs at 82a01908707f8c9eadedfe4b912c4cc6deda54f3 · WiseTechGlobal/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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
//------------------------------------------------------------------------------
// <copyright file="SqlScriptGeneratorVisitor.IndexDefinition.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(IndexDefinition node)
        {
            if (_options.NewLineFormattedIndexDefinition)
            {
                AlignmentPoint start = new AlignmentPoint();
                MarkAndPushAlignmentPoint(start);
            }

            GenerateKeyword(TSqlTokenType.Index);

            if (node.Name != null)
            {
                GenerateSpaceAndFragmentIfNotNull(node.Name);
            }

            if (node.Unique)
            {
                if (_options.NewLineFormattedIndexDefinition)
                {
                    NewLineAndIndent();
                }
                else
                {
                    GenerateSpace();
                }
                GenerateKeyword(TSqlTokenType.Unique);
            }

            if (node.IndexType != null)
            {
                switch (node.IndexType.IndexTypeKind)
                {
                    case IndexTypeKind.Clustered:
                        GenerateSpaceAndKeyword(TSqlTokenType.Clustered);
                        break;
                    case IndexTypeKind.NonClustered:
                        GenerateSpaceAndKeyword(TSqlTokenType.NonClustered);
                        break;
                    case IndexTypeKind.NonClusteredHash:
                        GenerateSpaceAndKeyword(TSqlTokenType.NonClustered);
                        GenerateSpaceAndIdentifier(CodeGenerationSupporter.Hash);
						break;
					case IndexTypeKind.ClusteredColumnStore:
                        GenerateSpaceAndKeyword(TSqlTokenType.Clustered);
                        GenerateSpaceAndIdentifier(CodeGenerationSupporter.ColumnStore);
						break;
					case IndexTypeKind.NonClusteredColumnStore:
                        GenerateSpaceAndKeyword(TSqlTokenType.NonClustered);
                        GenerateSpaceAndIdentifier(CodeGenerationSupporter.ColumnStore);
                        break;
                }
            }

            if (node.Columns.Count > 0)
            {
                GenerateSpace();
                GenerateParenthesisedCommaSeparatedList(node.Columns);
            }

            if (node.IncludeColumns.Count > 0)
            {
                if (_options.NewLineFormattedIndexDefinition)
                {
                    NewLineAndIndent();
                }
                else
                {
                    GenerateSpace();
                }
                GenerateIdentifier(CodeGenerationSupporter.Include);
                GenerateSpace();
                GenerateParenthesisedCommaSeparatedList(node.IncludeColumns);
            }

			if (node.FilterPredicate != null)
			{
                if (_options.NewLineFormattedIndexDefinition)
                {
                    NewLineAndIndent();
                }
                else
                {
                    GenerateSpace();
                }
                GenerateKeyword(TSqlTokenType.Where);
                GenerateSpaceAndFragmentIfNotNull(node.FilterPredicate);
            }

            if (node.IndexOptions.Count > 0)
            {
                GenerateIndexOptions(node.IndexOptions);
            }

            if (node.OnFileGroupOrPartitionScheme != null)
            {
                NewLineAndIndent();
                GenerateKeyword(TSqlTokenType.On);

                GenerateSpaceAndFragmentIfNotNull(node.OnFileGroupOrPartitionScheme);
            }

            GenerateFileStreamOn(node);

            if (_options.NewLineFormattedIndexDefinition)
            {
                PopAlignmentPoint();
            }
        }
    }
}