dbal.driver.pgsql.syntax source code

1 module dbal.driver.pgsql.syntax;
2 
3 import dbal;
4 
5 class PgSqlSyntax : SqlSyntax
6 {
7 	SqlBuilder _builder;
8 
9 	this(PgSqlBuilder builder)
10 	{
11 		this._builder = builder;
12 	}
13 
14 	string selectExpr()
15 	{
16 		string select;
17 		if(!_builder.selectKeys.length)
18 			select = " * ";
19 		else {
20 			foreach(v;_builder.selectKeys){
21 				select ~= v ~ ",";
22 			}
23 			select = select[0 .. $-1];
24 		}
25 		return select;
26 	}
27 	string whereExpr()
28 	{
29 		string where;
30 		if(_builder.whereKeys.length){
31 			int i = 0;
32 			where = " WHERE ";
33 			foreach(v;_builder.whereKeys){
34 				i++;
35 				where ~= v.toString;
36 				if(i<_builder.whereKeys.length) where ~= " AND ";
37 			}
38 		}
39 		if (_builder.multiWhereStr.length){
40 			where ~= where.length ? (" AND " ~ _builder.multiWhereStr) : (" WHERE" ~ _builder.multiWhereStr);
41 		}
42 		return where;
43 	}
44 	string joinExpr()
45 	{
46 		string joinstr;
47 		if(_builder.joins.length)
48 			foreach(join;_builder.joins){joinstr ~= join.toString;}
49 		return joinstr;
50 	}
51 	string groupByExpr()
52 	{
53 		if(_builder.groupBy.length)
54 			return  " GROUP BY " ~ _builder.groupBy;
55 		return null;
56 	}
57 	string havingExpr()
58 	{
59 		if(_builder.having.length) 
60 			return  " HAVING " ~ _builder.having;
61 		return null;
62 	}
63 	string orderExpr()
64 	{
65 		if(_builder.orderBy.length && _builder.order.length)
66 			return  " ORDER BY " ~ _builder.orderBy ~ " " ~ _builder.order;
67 		return null;
68 	}
69 	string limitExpr()
70 	{
71 		if(_builder.limit)
72 			return " LIMIT " ~ (_builder.limit).to!string;
73 		return null;
74 	}
75 	string offsetExpr()
76 	{
77 		if(_builder.offset) 
78 			return " OFFSET " ~ (_builder.offset).to!string;
79 		return null;
80 	}
81 
82 	string setExpr()
83 	{
84 		string set;
85 		if(_builder.values.length){
86 			set ~= " SET ";
87 			int i = 0;
88 			foreach(k,v;_builder.values){
89 				i++;
90 				set ~= v.toString;
91 				if(i<_builder.values.length) set ~= " , ";
92 			}
93 		}else{
94 			throw new DbalException("query builder update method have not set values");
95 		}
96 		return set;
97 	}
98 
99 	string autoIncreaseExpr()
100 	{
101 		return _builder.getAutoIncrease.length ? " returning " ~ _builder.getAutoIncrease : "";
102 	}
103 
104 	string insertExpr()
105 	{
106 		if(!_builder.values.length) throw new DbalException("query build insert have not values");
107 		string keys;
108 		string values;
109 		foreach(k,v;_builder.values){
110 			keys ~= k~",";
111 			values ~= v.value~",";
112 		}
113 		return "(" ~ keys[0.. $-1] ~ ") VALUES("~ values[0..$-1]  ~")";
114 	}
115 
116 	override string toString()
117 	{
118 		if(!_builder.tableName.length)
119 			throw new DbalException("query build table name not exists");
120 		string str;
121 		switch(_builder.method){
122 			case Method.Select:
123 				str ~= Method.Select;
124 				str ~= selectExpr();
125 				str ~= " FROM " ~ _builder.tableName ~ " ";
126 				str ~= joinExpr();
127 				str ~= whereExpr();
128 				str ~= groupByExpr();
129 				str ~= havingExpr();
130 				str ~= orderExpr();
131 				str ~= limitExpr();
132 				str ~= offsetExpr();
133 				break;
134 			case Method.Update:
135 				str ~= Method.Update ~ " " ~ _builder.tableName;
136 				str ~= setExpr();
137 				str ~= whereExpr();
138 				str ~= orderExpr();
139 				str ~= limitExpr();
140 				break;
141 			case Method.Delete:
142 				str ~= Method.Delete ~ " " ~ _builder.tableName;
143 				str ~= whereExpr();
144 				str ~= orderExpr();
145 				str ~= limitExpr();
146 				break;
147 			case Method.Insert:
148 				str ~= Method.Insert ~ " into " ~ _builder.tableName;
149 				str ~= insertExpr();
150 				str ~= autoIncreaseExpr();
151 				break;
152 			default:
153 				throw new DbalException("query build method not found");
154 		}
155 		return str;
156 	}
157 }