forked from ServiceStack/ServiceStack.OrmLite
-
Notifications
You must be signed in to change notification settings - Fork 0
/
PostgreSQLExpressionVisitor.cs
129 lines (117 loc) · 3.89 KB
/
PostgreSQLExpressionVisitor.cs
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
122
123
124
125
126
127
128
129
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Collections.ObjectModel;
using System.Linq.Expressions;
using ServiceStack.OrmLite;
namespace ServiceStack.OrmLite.PostgreSQL
{
public class PostgreSQLExpressionVisitor<T>:SqlExpressionVisitor<T>
{
public PostgreSQLExpressionVisitor():base()
{
}
protected override string VisitMethodCall(MethodCallExpression m)
{
List<Object> args = this.VisitExpressionList(m.Arguments);
Object r ;
if( m.Object!=null)
r=Visit(m.Object);
else{
r= args[0];
args.RemoveAt(0);
}
switch(m.Method.Name){
case "ToUpper":
return string.Format("upper({0})",r);
case "ToLower":
return string.Format("lower({0})",r);
case "StartsWith": //LEFT( title, 1) = '#'
return string.Format("upper({0}) like '{1}%'",r,RemoveQuote(args[0].ToString().ToUpper()) );
case "EndsWith":
return string.Format("upper({0}) like '%{1}'",r,RemoveQuote(args[0].ToString().ToUpper()) );
case "Contains":
return string.Format("upper({0}) like '%{1}%'",r,RemoveQuote(args[0].ToString()).ToUpper() );
case "Substring":
var startIndex = Int32.Parse(args[0].ToString() ) +1;
if (args.Count==2){
var length = Int32.Parse( args[1].ToString() );
return string.Format("substring({0} from {1} for {2})",
r,
startIndex,
length);
}
else
return string.Format("substring({0} from {1})",
r,
startIndex);
case "Round":
case "Floor":
case "Ceiling":
case "Coalesce":
case "Abs":
case "Sum":
return string.Format("{0}({1}{2})",
m.Method.Name,
r,
args.Count==1? string.Format(",{0}", args[0]):"" );
case "Concat":
StringBuilder s = new StringBuilder();
foreach(Object e in args ){
s.AppendFormat( " || {0}", e);
}
return string.Format("{0}{1}",r, s.ToString());
case "In":
var member = Expression.Convert(m.Arguments[1], typeof(object));
var lambda = Expression.Lambda<Func<object>>(member);
var getter = lambda.Compile();
var inArgs = getter() as object[];
StringBuilder sIn = new StringBuilder();
foreach(Object e in inArgs ){
if(e.GetType().ToString()!="System.Collections.Generic.List`1[System.Object]"){
sIn.AppendFormat("{0}{1}",
sIn.Length>0 ? ",":"",
OrmLiteConfig.DialectProvider.GetQuotedValue(e, e.GetType()) );
}
else{
var listArgs= e as IList<Object>;
foreach(Object el in listArgs){
sIn.AppendFormat("{0}{1}",
sIn.Length>0 ? ",":"",
OrmLiteConfig.DialectProvider.GetQuotedValue(el, el.GetType()) );
}
}
}
return string.Format("{0} {1} ({2})", r, m.Method.Name, sIn.ToString() );
case "Desc":
return string.Format("{0} DESC", r);
case "As":
return string.Format("{0} As {1}", r,
OrmLiteConfig.DialectProvider.GetQuotedColumnName(RemoveQuoteFromAlias(args[0].ToString())));
case "ToString":
return r.ToString();
default:
StringBuilder s2 = new StringBuilder();
foreach(Object e in args ){
s2.AppendFormat(",{0}",
OrmLiteConfig.DialectProvider.GetQuotedValue(e, e.GetType()) );
}
return string.Format("{0}({1}{2})",m.Method.Name,r, s2.ToString());
}
}
public override string LimitExpression{
get{
if(!Rows.HasValue) return "";
string offset;
if(Skip.HasValue){
offset= string.Format(" OFFSET {0}", Skip.Value );
}
else{
offset=string.Empty;
}
return string.Format("LIMIT {0}{1}", Rows.Value, offset);
}
}
}
}