franta-hg@22
|
1 |
/**
|
franta-hg@22
|
2 |
* SQL-DK
|
franta-hg@22
|
3 |
* Copyright © 2013 František Kučera (frantovo.cz)
|
franta-hg@22
|
4 |
*
|
franta-hg@22
|
5 |
* This program is free software: you can redistribute it and/or modify
|
franta-hg@22
|
6 |
* it under the terms of the GNU General Public License as published by
|
franta-hg@22
|
7 |
* the Free Software Foundation, either version 3 of the License, or
|
franta-hg@22
|
8 |
* (at your option) any later version.
|
franta-hg@22
|
9 |
*
|
franta-hg@22
|
10 |
* This program is distributed in the hope that it will be useful,
|
franta-hg@22
|
11 |
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
franta-hg@22
|
12 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
franta-hg@22
|
13 |
* GNU General Public License for more details.
|
franta-hg@22
|
14 |
*
|
franta-hg@22
|
15 |
* You should have received a copy of the GNU General Public License
|
franta-hg@22
|
16 |
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
franta-hg@22
|
17 |
*/
|
franta-hg@22
|
18 |
package info.globalcode.sql.dk.formatting;
|
franta-hg@22
|
19 |
|
franta-hg@22
|
20 |
import info.globalcode.sql.dk.Parameter;
|
franta-hg@29
|
21 |
import info.globalcode.sql.dk.configuration.DatabaseDefinition;
|
franta-hg@22
|
22 |
import java.util.EmptyStackException;
|
franta-hg@22
|
23 |
import java.util.EnumSet;
|
franta-hg@22
|
24 |
import java.util.List;
|
franta-hg@22
|
25 |
import java.util.Stack;
|
franta-hg@22
|
26 |
|
franta-hg@22
|
27 |
/**
|
franta-hg@22
|
28 |
*
|
franta-hg@22
|
29 |
* @author Ing. František Kučera (frantovo.cz)
|
franta-hg@22
|
30 |
*/
|
franta-hg@22
|
31 |
public abstract class AbstractFormatter implements Formatter {
|
franta-hg@22
|
32 |
|
franta-hg@22
|
33 |
private Stack<State> state = new Stack<>();
|
franta-hg@24
|
34 |
private FormatterContext formatterContext;
|
franta-hg@22
|
35 |
private ColumnsHeader currentColumnsHeader;
|
franta-hg@22
|
36 |
private String currentQuery;
|
franta-hg@22
|
37 |
private int currentColumnsCount;
|
franta-hg@25
|
38 |
private int currentRowCount;
|
franta-hg@22
|
39 |
|
franta-hg@24
|
40 |
public AbstractFormatter(FormatterContext formatterContext) {
|
franta-hg@24
|
41 |
this.formatterContext = formatterContext;
|
franta-hg@22
|
42 |
state.push(State.ROOT);
|
franta-hg@22
|
43 |
}
|
franta-hg@22
|
44 |
|
franta-hg@22
|
45 |
/*
|
franta-hg@22
|
46 |
* root
|
franta-hg@22
|
47 |
* .database
|
franta-hg@22
|
48 |
* ..resultSet
|
franta-hg@22
|
49 |
* ...@query
|
franta-hg@22
|
50 |
* ...@parameters
|
franta-hg@22
|
51 |
* ...@columnsHeader
|
franta-hg@22
|
52 |
* ...row
|
franta-hg@22
|
53 |
* ....@columnValue
|
franta-hg@22
|
54 |
* ..updatesResult
|
franta-hg@22
|
55 |
* ...@query
|
franta-hg@22
|
56 |
* ...@parameters
|
franta-hg@22
|
57 |
* ...@updatedRowsCount
|
franta-hg@22
|
58 |
* ...generatedKeys
|
franta-hg@22
|
59 |
* ....resultSet (see above)
|
franta-hg@22
|
60 |
*/
|
franta-hg@22
|
61 |
protected enum State {
|
franta-hg@22
|
62 |
|
franta-hg@22
|
63 |
ROOT,
|
franta-hg@22
|
64 |
DATABASE,
|
franta-hg@22
|
65 |
RESULT_SET,
|
franta-hg@22
|
66 |
ROW,
|
franta-hg@22
|
67 |
UPDATES_RESULT,
|
franta-hg@22
|
68 |
GENERATED_KEYS
|
franta-hg@22
|
69 |
}
|
franta-hg@22
|
70 |
|
franta-hg@22
|
71 |
/**
|
franta-hg@22
|
72 |
* Go down in hierarchy.
|
franta-hg@22
|
73 |
* Pushes new state and verifies the old one.
|
franta-hg@22
|
74 |
*
|
franta-hg@22
|
75 |
* @param current the new state – currently entering
|
franta-hg@22
|
76 |
* @param expected expected previous states (any of them is valid)
|
franta-hg@22
|
77 |
* @return previous state
|
franta-hg@22
|
78 |
* @throws IllegalStateException if previous state was not one from expected
|
franta-hg@22
|
79 |
*/
|
franta-hg@22
|
80 |
private State pushState(State current, EnumSet expected) {
|
franta-hg@22
|
81 |
State previous = state.peek();
|
franta-hg@22
|
82 |
|
franta-hg@22
|
83 |
if (expected.contains(previous)) {
|
franta-hg@22
|
84 |
state.push(current);
|
franta-hg@22
|
85 |
return previous;
|
franta-hg@22
|
86 |
} else {
|
franta-hg@22
|
87 |
throw new IllegalStateException("Formatter was in wrong state: " + previous + " when it should be in one of: " + expected);
|
franta-hg@22
|
88 |
}
|
franta-hg@22
|
89 |
}
|
franta-hg@22
|
90 |
|
franta-hg@22
|
91 |
protected State peekState(EnumSet expected) {
|
franta-hg@22
|
92 |
State current = state.peek();
|
franta-hg@22
|
93 |
|
franta-hg@22
|
94 |
if (expected.contains(current)) {
|
franta-hg@22
|
95 |
return current;
|
franta-hg@22
|
96 |
} else {
|
franta-hg@22
|
97 |
throw new IllegalStateException("Formatter is in wrong state: " + current + " when it should be in one of: " + expected);
|
franta-hg@22
|
98 |
}
|
franta-hg@22
|
99 |
|
franta-hg@22
|
100 |
}
|
franta-hg@22
|
101 |
|
franta-hg@22
|
102 |
/**
|
franta-hg@22
|
103 |
* Go up in hierarchy.
|
franta-hg@22
|
104 |
* Pops the superior state/branch.
|
franta-hg@22
|
105 |
*
|
franta-hg@22
|
106 |
* @param expected expected superior state
|
franta-hg@22
|
107 |
* @return the superior state
|
franta-hg@22
|
108 |
* @throws IllegalStateException if superior state was not one from expected or if there is no
|
franta-hg@22
|
109 |
* more superior state (we are at root level)
|
franta-hg@22
|
110 |
*/
|
franta-hg@22
|
111 |
private State popState(EnumSet expected) {
|
franta-hg@22
|
112 |
try {
|
franta-hg@22
|
113 |
State superior = state.pop();
|
franta-hg@22
|
114 |
if (expected.contains(superior)) {
|
franta-hg@22
|
115 |
return superior;
|
franta-hg@22
|
116 |
} else {
|
franta-hg@22
|
117 |
throw new IllegalStateException("Formatter had wrong superior state: " + superior + " when it should be in one of: " + expected);
|
franta-hg@22
|
118 |
}
|
franta-hg@22
|
119 |
} catch (EmptyStackException e) {
|
franta-hg@22
|
120 |
throw new IllegalStateException("Formatter was already at root level – there is nothing above that.", e);
|
franta-hg@22
|
121 |
}
|
franta-hg@22
|
122 |
}
|
franta-hg@22
|
123 |
|
franta-hg@22
|
124 |
@Override
|
franta-hg@29
|
125 |
public void writeStartDatabase(DatabaseDefinition databaseDefinition) {
|
franta-hg@22
|
126 |
pushState(State.DATABASE, EnumSet.of(State.ROOT));
|
franta-hg@22
|
127 |
}
|
franta-hg@22
|
128 |
|
franta-hg@22
|
129 |
@Override
|
franta-hg@22
|
130 |
public void writeEndDatabase() {
|
franta-hg@22
|
131 |
popState(EnumSet.of(State.ROOT));
|
franta-hg@22
|
132 |
}
|
franta-hg@22
|
133 |
|
franta-hg@22
|
134 |
@Override
|
franta-hg@22
|
135 |
public void writeStartResultSet() {
|
franta-hg@22
|
136 |
pushState(State.RESULT_SET, EnumSet.of(State.DATABASE, State.GENERATED_KEYS));
|
franta-hg@25
|
137 |
currentRowCount = 0;
|
franta-hg@22
|
138 |
}
|
franta-hg@22
|
139 |
|
franta-hg@22
|
140 |
@Override
|
franta-hg@22
|
141 |
public void writeEndResultSet() {
|
franta-hg@22
|
142 |
popState(EnumSet.of(State.DATABASE, State.GENERATED_KEYS));
|
franta-hg@22
|
143 |
currentColumnsHeader = null;
|
franta-hg@22
|
144 |
}
|
franta-hg@22
|
145 |
|
franta-hg@22
|
146 |
@Override
|
franta-hg@22
|
147 |
public void writeQuery(String sql) {
|
franta-hg@22
|
148 |
peekState(EnumSet.of(State.RESULT_SET, State.UPDATES_RESULT));
|
franta-hg@22
|
149 |
|
franta-hg@22
|
150 |
if (currentColumnsHeader == null) {
|
franta-hg@22
|
151 |
currentQuery = sql;
|
franta-hg@22
|
152 |
} else {
|
franta-hg@22
|
153 |
throw new IllegalStateException("Query string '" + sql + "' must be set before columns header – was already set: " + currentColumnsHeader);
|
franta-hg@22
|
154 |
}
|
franta-hg@22
|
155 |
}
|
franta-hg@22
|
156 |
|
franta-hg@22
|
157 |
@Override
|
franta-hg@22
|
158 |
public void writeParameters(List<Parameter> parameters) {
|
franta-hg@22
|
159 |
peekState(EnumSet.of(State.RESULT_SET, State.UPDATES_RESULT));
|
franta-hg@22
|
160 |
|
franta-hg@22
|
161 |
if (currentColumnsHeader != null) {
|
franta-hg@22
|
162 |
throw new IllegalStateException("Parameters '" + parameters + "' must be set before columns header – was already set: " + currentColumnsHeader);
|
franta-hg@22
|
163 |
}
|
franta-hg@22
|
164 |
|
franta-hg@22
|
165 |
if (currentQuery == null) {
|
franta-hg@22
|
166 |
throw new IllegalStateException("Parameters '" + parameters + "' must be set after query – was not yet set.");
|
franta-hg@22
|
167 |
}
|
franta-hg@22
|
168 |
}
|
franta-hg@22
|
169 |
|
franta-hg@22
|
170 |
@Override
|
franta-hg@22
|
171 |
public void writeColumnsHeader(ColumnsHeader header) {
|
franta-hg@22
|
172 |
peekState(EnumSet.of(State.RESULT_SET, State.UPDATES_RESULT));
|
franta-hg@22
|
173 |
|
franta-hg@22
|
174 |
if (currentColumnsHeader == null) {
|
franta-hg@22
|
175 |
currentColumnsHeader = header;
|
franta-hg@22
|
176 |
} else {
|
franta-hg@22
|
177 |
throw new IllegalStateException("Columns header can be set only once per result set – was already set: " + currentColumnsHeader);
|
franta-hg@22
|
178 |
}
|
franta-hg@22
|
179 |
}
|
franta-hg@22
|
180 |
|
franta-hg@22
|
181 |
@Override
|
franta-hg@22
|
182 |
public void writeStartRow() {
|
franta-hg@22
|
183 |
pushState(State.ROW, EnumSet.of(State.RESULT_SET));
|
franta-hg@22
|
184 |
currentColumnsCount = 0;
|
franta-hg@25
|
185 |
currentRowCount++;
|
franta-hg@22
|
186 |
}
|
franta-hg@22
|
187 |
|
franta-hg@22
|
188 |
@Override
|
franta-hg@22
|
189 |
public void writeEndRow() {
|
franta-hg@22
|
190 |
popState(EnumSet.of(State.RESULT_SET));
|
franta-hg@22
|
191 |
}
|
franta-hg@22
|
192 |
|
franta-hg@22
|
193 |
@Override
|
franta-hg@22
|
194 |
public void writeColumnValue(Object value) {
|
franta-hg@22
|
195 |
peekState(EnumSet.of(State.ROW));
|
franta-hg@22
|
196 |
currentColumnsCount++;
|
franta-hg@22
|
197 |
|
franta-hg@22
|
198 |
int declaredCount = currentColumnsHeader.getColumnCount();
|
franta-hg@22
|
199 |
if (currentColumnsCount > declaredCount) {
|
franta-hg@22
|
200 |
throw new IllegalStateException("Current columns count is " + currentColumnsCount + " which is more than declared " + declaredCount + " in header.");
|
franta-hg@22
|
201 |
}
|
franta-hg@22
|
202 |
}
|
franta-hg@22
|
203 |
|
franta-hg@22
|
204 |
@Override
|
franta-hg@22
|
205 |
public void writeStartUpdatesResult() {
|
franta-hg@22
|
206 |
pushState(State.RESULT_SET, EnumSet.of(State.DATABASE));
|
franta-hg@22
|
207 |
}
|
franta-hg@22
|
208 |
|
franta-hg@22
|
209 |
@Override
|
franta-hg@22
|
210 |
public void writeEndUpdatesResult() {
|
franta-hg@22
|
211 |
popState(EnumSet.of(State.DATABASE));
|
franta-hg@22
|
212 |
currentColumnsHeader = null;
|
franta-hg@22
|
213 |
}
|
franta-hg@22
|
214 |
|
franta-hg@22
|
215 |
@Override
|
franta-hg@22
|
216 |
public void writeUpdatedRowsCount(int updatedRowsCount) {
|
franta-hg@22
|
217 |
peekState(EnumSet.of(State.UPDATES_RESULT));
|
franta-hg@22
|
218 |
}
|
franta-hg@22
|
219 |
|
franta-hg@22
|
220 |
@Override
|
franta-hg@22
|
221 |
public void writeStartGeneratedKeys() {
|
franta-hg@22
|
222 |
pushState(State.GENERATED_KEYS, EnumSet.of(State.UPDATES_RESULT));
|
franta-hg@22
|
223 |
}
|
franta-hg@22
|
224 |
|
franta-hg@22
|
225 |
@Override
|
franta-hg@22
|
226 |
public void writeEndGeneratedKeys() {
|
franta-hg@22
|
227 |
popState(EnumSet.of(State.UPDATES_RESULT));
|
franta-hg@22
|
228 |
}
|
franta-hg@22
|
229 |
|
franta-hg@24
|
230 |
public FormatterContext getFormatterContext() {
|
franta-hg@24
|
231 |
return formatterContext;
|
franta-hg@22
|
232 |
}
|
franta-hg@22
|
233 |
|
franta-hg@22
|
234 |
protected ColumnsHeader getCurrentColumnsHeader() {
|
franta-hg@22
|
235 |
return currentColumnsHeader;
|
franta-hg@22
|
236 |
}
|
franta-hg@22
|
237 |
|
franta-hg@25
|
238 |
/**
|
franta-hg@25
|
239 |
* @return column number, 1 = first
|
franta-hg@25
|
240 |
*/
|
franta-hg@22
|
241 |
protected int getCurrentColumnsCount() {
|
franta-hg@22
|
242 |
return currentColumnsCount;
|
franta-hg@22
|
243 |
}
|
franta-hg@25
|
244 |
|
franta-hg@25
|
245 |
/**
|
franta-hg@25
|
246 |
* @return row number, 1 = first
|
franta-hg@25
|
247 |
*/
|
franta-hg@25
|
248 |
protected int getCurrentRowCount() {
|
franta-hg@25
|
249 |
return currentRowCount;
|
franta-hg@25
|
250 |
}
|
franta-hg@22
|
251 |
/**
|
franta-hg@22
|
252 |
* TODO: write SQLWarning
|
franta-hg@22
|
253 |
*/
|
franta-hg@22
|
254 |
}
|