FilesExpand file tree

 master

/

SqlServerQueryWriter.cs

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

50 lines (43 loc) · 1.77 KB

 master

/

SqlServerQueryWriter.cs

Copy path

Top

File metadata and controls

• Code
• Blame

50 lines (43 loc) · 1.77 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

using DataKit.SQL.Providers;

using DataKit.SQL.QueryExpressions;

using DataKit.SQL.SqlServer.QueryExpressions;

namespace DataKit.SQL.SqlServer

{

public class SqlServerQueryWriter : QueryWriter

{

public void Append(string str)

=> queryText.Append(str);

public override void WriteRandomFunction(RandomFunctionCallQueryExpression randomFunctionCallQueryExpression, QueryExpressionVisitor queryExpressionVisitor)

{

queryText.Append($" CAST(FLOOR(RAND() * {int.MaxValue}) AS BIGINT) ");

}

public override void WriteLastInsertedIdFunction(LastInsertedIdFunctionCallExpression lastInsertedIdFunctionCallExpression, QueryExpressionVisitor queryExpressionVisitor)

{

queryText.Append(" CAST(@@IDENTITY AS INT) ");

}

public override void WriteLimitExpressionParameter(LimitQueryExpression limitQueryExpression, QueryExpressionVisitor queryExpressionVisitor)

{

var currentQuery = statementStack.Peek() as SelectStatementQueryExpression;

if (currentQuery.OrderBy == null)

queryText.Append(" ORDER BY (SELECT NULL) ");

if (limitQueryExpression.Offset == null)

{

queryText.Append(" OFFSET 0 ROWS FETCH FIRST ");

}

else

{

queryText.Append(" OFFSET ");

WriteIfNotNull(limitQueryExpression.Offset, queryExpressionVisitor);

queryText.Append(" ROWS FETCH NEXT ");

}

WriteIfNotNull(limitQueryExpression.Limit, queryExpressionVisitor);

queryText.Append(" ROWS ONLY ");

}

public override void WriteExtension(QueryExpression queryExpression, QueryExpressionVisitor queryExpressionVisitor)

{

if (queryExpression is ISqlServerExtension sqlServerExtension)

sqlServerExtension.Write(this, queryExpressionVisitor);

base.WriteExtension(queryExpression, queryExpressionVisitor);

}

}

}