forked from sqlkata/querybuilder
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathFirebirdCompiler.cs
More file actions
120 lines (92 loc) · 3.31 KB
/
FirebirdCompiler.cs
File metadata and controls
120 lines (92 loc) · 3.31 KB
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
using System.Collections.Generic;
using System.Linq;
using System.Text.RegularExpressions;
namespace SqlKata.Compilers
{
public class FirebirdCompiler : Compiler
{
public FirebirdCompiler()
{
}
public override string EngineCode { get; } = EngineCodes.Firebird;
protected override string SingleRowDummyTableName => "RDB$DATABASE";
protected override SqlResult CompileInsertQuery(Query query)
{
var ctx = base.CompileInsertQuery(query);
var inserts = ctx.Query.GetComponents<AbstractInsertClause>("insert", EngineCode);
if (inserts.Count > 1)
{
ctx.RawSql = Regex.Replace(ctx.RawSql, @"\)\s+VALUES\s+\(", ") SELECT ");
ctx.RawSql = Regex.Replace(ctx.RawSql, @"\),\s*\(", " FROM RDB$DATABASE UNION ALL SELECT ");
ctx.RawSql = Regex.Replace(ctx.RawSql, @"\)$", " FROM RDB$DATABASE");
}
return ctx;
}
public override string CompileLimit(SqlResult ctx)
{
var limit = ctx.Query.GetLimit(EngineCode);
var offset = ctx.Query.GetOffset(EngineCode);
if (limit > 0 && offset > 0)
{
ctx.Bindings.Add(offset + 1);
ctx.Bindings.Add(limit + offset);
return $"ROWS {parameterPlaceholder} TO {parameterPlaceholder}";
}
return null;
}
protected override string CompileColumns(SqlResult ctx)
{
var compiled = base.CompileColumns(ctx);
var limit = ctx.Query.GetLimit(EngineCode);
var offset = ctx.Query.GetOffset(EngineCode);
if (limit > 0 && offset == 0)
{
ctx.Bindings.Insert(0, limit);
ctx.Query.ClearComponent("limit");
return $"SELECT FIRST {parameterPlaceholder}" + compiled.Substring(6);
}
else if (limit == 0 && offset > 0)
{
ctx.Bindings.Insert(0, offset);
ctx.Query.ClearComponent("offset");
return $"SELECT SKIP {parameterPlaceholder}" + compiled.Substring(6);
}
return compiled;
}
protected override string CompileBasicDateCondition(SqlResult ctx, BasicDateCondition condition)
{
var column = Wrap(condition.Column);
string left;
if (condition.Part == "time")
{
left = $"CAST({column} as TIME)";
}
else if (condition.Part == "date")
{
left = $"CAST({column} as DATE)";
}
else
{
left = $"EXTRACT({condition.Part.ToUpperInvariant()} FROM {column})";
}
var sql = $"{left} {condition.Operator} {Parameter(ctx, condition.Value)}";
if (condition.IsNot)
{
return $"NOT ({sql})";
}
return sql;
}
public override string WrapValue(string value)
{
return base.WrapValue(value).ToUpperInvariant();
}
public override string CompileTrue()
{
return "1";
}
public override string CompileFalse()
{
return "0";
}
}
}