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
|
// Copyright (c) Microsoft Open Technologies, Inc. All rights reserved. See License.txt in the project root for license information.
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Text;
using System.Data.SqlClient;
using System.Data.Metadata.Edm;
using System.Data.Common.CommandTrees;
namespace SampleEntityFrameworkProvider
{
/// <summary>
/// This extends StringWriter primarily to add the ability to add an indent
/// to each line that is written out.
/// </summary>
class SqlWriter : StringWriter
{
// We start at -1, since the first select statement will increment it to 0.
int indent = -1;
/// <summary>
/// The number of tabs to be added at the beginning of each new line.
/// </summary>
internal int Indent
{
get { return indent; }
set { indent = value; }
}
bool atBeginningOfLine = true;
/// <summary>
///
/// </summary>
/// <param name="b"></param>
public SqlWriter(StringBuilder b)
: base(b, System.Globalization.CultureInfo.InvariantCulture)
{
}
/// <summary>
/// Reset atBeginningofLine if we detect the newline string.
/// <see cref="SqlBuilder.AppendLine"/>
/// Add as many tabs as the value of indent if we are at the
/// beginning of a line.
/// </summary>
/// <param name="value"></param>
public override void Write(string value)
{
if (value == "\r\n")
{
base.WriteLine();
atBeginningOfLine = true;
}
else
{
if (atBeginningOfLine)
{
if (indent > 0)
{
base.Write(new string('\t', indent));
}
atBeginningOfLine = false;
}
base.Write(value);
}
}
/// <summary>
///
/// </summary>
public override void WriteLine()
{
base.WriteLine();
atBeginningOfLine = true;
}
}
}
|
