svn-gvsig-desktop / trunk / org.gvsig.desktop / org.gvsig.desktop.compat.cdc / org.gvsig.fmap.dal / org.gvsig.fmap.dal.swing / org.gvsig.fmap.dal.swing.impl / src / main / java / org / gvsig / fmap / dal / swing / impl / searchPostProcess / distinctOn / DistinctOn.java @ 46062
History | View | Annotate | Download (9.03 KB)
1 | 45537 | jolivas | /*
|
---|---|---|---|
2 | * To change this license header, choose License Headers in Project Properties.
|
||
3 | * To change this template file, choose Tools | Templates
|
||
4 | * and open the template in the editor.
|
||
5 | */
|
||
6 | package org.gvsig.fmap.dal.swing.impl.searchPostProcess.distinctOn; |
||
7 | |||
8 | import java.io.File; |
||
9 | import java.util.Collection; |
||
10 | import java.util.HashMap; |
||
11 | import java.util.Map; |
||
12 | import java.util.Objects; |
||
13 | 45705 | jolivas | import java.util.UUID; |
14 | 45537 | jolivas | |
15 | import org.gvsig.fmap.dal.DALLocator; |
||
16 | import org.gvsig.fmap.dal.DataManager; |
||
17 | import org.gvsig.fmap.dal.DataStoreParameters; |
||
18 | import org.gvsig.fmap.dal.exception.DataException; |
||
19 | import org.gvsig.fmap.dal.exception.ValidateDataParametersException; |
||
20 | import org.gvsig.fmap.dal.feature.EditableFeature; |
||
21 | 45705 | jolivas | import org.gvsig.fmap.dal.feature.EditableFeatureAttributeDescriptor; |
22 | import org.gvsig.fmap.dal.feature.EditableFeatureType; |
||
23 | 45537 | jolivas | import org.gvsig.fmap.dal.feature.Feature; |
24 | import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor; |
||
25 | 45705 | jolivas | import org.gvsig.fmap.dal.feature.FeatureAttributeEmulator; |
26 | 45537 | jolivas | import org.gvsig.fmap.dal.feature.FeatureQuery; |
27 | import org.gvsig.fmap.dal.feature.FeatureSet; |
||
28 | import org.gvsig.fmap.dal.feature.FeatureStore; |
||
29 | import org.gvsig.fmap.dal.feature.FeatureType; |
||
30 | import org.gvsig.fmap.dal.store.jdbc.JDBCNewStoreParameters; |
||
31 | import org.gvsig.fmap.dal.store.jdbc.JDBCServerExplorer; |
||
32 | import org.gvsig.fmap.dal.store.jdbc.JDBCServerExplorerParameters; |
||
33 | import org.gvsig.fmap.dal.swing.impl.searchPostProcess.distinctOn.AggregateOperation.AggregateOperationFactory; |
||
34 | import org.gvsig.fmap.dal.swing.searchPostProcess.AbstractSearchPostProcess; |
||
35 | import org.gvsig.fmap.dal.swing.searchPostProcess.SearchPostProcessFactory; |
||
36 | import org.gvsig.tools.ToolsLocator; |
||
37 | 45705 | jolivas | import org.gvsig.tools.dataTypes.DataType; |
38 | 46062 | omartinez | import org.gvsig.tools.dispose.DisposeUtils; |
39 | 45537 | jolivas | import org.gvsig.tools.dynobject.DynObject; |
40 | import org.gvsig.tools.folders.FoldersManager; |
||
41 | 46062 | omartinez | import org.gvsig.tools.i18n.I18nManager; |
42 | import org.gvsig.tools.task.SimpleTaskStatus; |
||
43 | 45537 | jolivas | import org.gvsig.tools.util.HasAFile; |
44 | import org.slf4j.LoggerFactory; |
||
45 | |||
46 | /**
|
||
47 | *
|
||
48 | * @author jovivas
|
||
49 | */
|
||
50 | 46062 | omartinez | public class DistinctOn extends AbstractSearchPostProcess { |
51 | 45537 | jolivas | |
52 | 46062 | omartinez | private static final org.slf4j.Logger LOGGER = LoggerFactory.getLogger(DistinctOnFactory.class); |
53 | 45537 | jolivas | |
54 | 46062 | omartinez | public DistinctOn(
|
55 | SearchPostProcessFactory factory, |
||
56 | FeatureStore input, |
||
57 | FeatureQuery query, |
||
58 | DynObject parameters |
||
59 | ) { |
||
60 | super(factory, input, query, parameters);
|
||
61 | 45537 | jolivas | |
62 | 46062 | omartinez | } |
63 | 45537 | jolivas | |
64 | 46062 | omartinez | @Override
|
65 | public SearchPostProcessResult execute(FeatureStore inputStore, FeatureQuery query, DynObject parameters, SimpleTaskStatus status) {
|
||
66 | String uniqueFieldName;
|
||
67 | 45715 | jolivas | |
68 | 46062 | omartinez | uniqueFieldName = (String) parameters.getDynValue("field"); |
69 | 45705 | jolivas | |
70 | 46062 | omartinez | I18nManager i18n = ToolsLocator.getI18nManager(); |
71 | if (status == null) { |
||
72 | status = ToolsLocator.getTaskStatusManager().createDefaultSimpleTaskStatus(i18n.getTranslation("_Executing_post_process"));
|
||
73 | status.setAutoremove(true);
|
||
74 | status.add(); |
||
75 | } else {
|
||
76 | status.push(); |
||
77 | status.setTitle(i18n.getTranslation("_Executing_post_process"));
|
||
78 | } |
||
79 | status.message(i18n.getTranslation("_Creating_data_base"));
|
||
80 | 45537 | jolivas | |
81 | 46062 | omartinez | try {
|
82 | // crear ruta de archivo temporal
|
||
83 | FoldersManager foldersManager = ToolsLocator.getFoldersManager(); |
||
84 | File tempFile = foldersManager.getUniqueTemporaryFile("distinctOn_process_" + UUID.randomUUID().toString()); |
||
85 | 45705 | jolivas | |
86 | 46062 | omartinez | // crear SERVER STORE
|
87 | DataManager dataManager = DALLocator.getDataManager(); |
||
88 | JDBCServerExplorerParameters serverParameters = (JDBCServerExplorerParameters) dataManager.createServerExplorerParameters("H2Spatial");
|
||
89 | ((HasAFile) serverParameters).setFile(tempFile); |
||
90 | JDBCServerExplorer serverExplorer = (JDBCServerExplorer) dataManager.openServerExplorer("H2Spatial", serverParameters);
|
||
91 | 45962 | omartinez | |
92 | 46062 | omartinez | //Crear tablas en server store
|
93 | JDBCNewStoreParameters parametersResults = serverExplorer.getAddParameters(); |
||
94 | parametersResults.setDynValue("Table", "results"); |
||
95 | FeatureType ft_inputStore = inputStore.getDefaultFeatureType(); |
||
96 | EditableFeatureType ft_inputStoreNoContainer = dataManager.createFeatureType(); |
||
97 | ft_inputStoreNoContainer.addAll(ft_inputStore); |
||
98 | for (FeatureAttributeDescriptor attr : ft_inputStore) {
|
||
99 | DataType dataType = attr.getDataType(); |
||
100 | EditableFeatureAttributeDescriptor eattr = ft_inputStoreNoContainer.getEditableAttributeDescriptor(attr.getName()); |
||
101 | if (dataType.isContainer()) {
|
||
102 | ft_inputStoreNoContainer.remove(attr.getName()); |
||
103 | } |
||
104 | if (eattr.isComputed()) {
|
||
105 | eattr.setFeatureAttributeEmulator((FeatureAttributeEmulator) null);
|
||
106 | } |
||
107 | if (eattr.isPrimaryKey()) {
|
||
108 | eattr.setIsPrimaryKey(false);
|
||
109 | } |
||
110 | if (!eattr.allowIndexDuplicateds()) {
|
||
111 | eattr.setAllowIndexDuplicateds(true);
|
||
112 | } |
||
113 | } |
||
114 | Map<String, EditableFeatureAttributeDescriptor> allExtraColumns = new HashMap<>(); |
||
115 | for (EditableFeatureAttributeDescriptor column : ft_inputStore.getExtraColumns().getColumns()) {
|
||
116 | allExtraColumns.put(column.getName(), column); |
||
117 | } |
||
118 | for (EditableFeatureAttributeDescriptor column : query.getExtraColumn().getColumns()) {
|
||
119 | allExtraColumns.put(column.getName(), column); |
||
120 | } |
||
121 | if (allExtraColumns.size() > 0) { |
||
122 | for (EditableFeatureAttributeDescriptor column : allExtraColumns.values()) {
|
||
123 | DataType dataType = column.getDataType(); |
||
124 | if (!dataType.isContainer()) {
|
||
125 | ft_inputStoreNoContainer.add(column.getName(), column.getType(), column.getSize()); |
||
126 | } |
||
127 | } |
||
128 | } |
||
129 | |||
130 | parametersResults.setDefaultFeatureType(ft_inputStoreNoContainer); |
||
131 | |||
132 | serverExplorer.add("H2Spatial", parametersResults, true); |
||
133 | |||
134 | DataStoreParameters storeParametersResults = dataManager.createStoreParameters("H2Spatial");
|
||
135 | storeParametersResults.setDynValue("database_file", tempFile);
|
||
136 | storeParametersResults.setDynValue("Table", "results"); |
||
137 | |||
138 | //Creaci?n del store con los resultados
|
||
139 | FeatureStore storeResults = (FeatureStore) dataManager.openStore("H2Spatial", storeParametersResults);
|
||
140 | |||
141 | // Creaci?n del Map con las funciones de agragado
|
||
142 | Map<String, AggregateOperation> aggregates = new HashMap<>(); |
||
143 | for (FeatureAttributeDescriptor attr : ft_inputStoreNoContainer) {
|
||
144 | if (parameters.hasDynValue(attr.getName())) {
|
||
145 | String operation = (String) parameters.getDynValue(attr.getName()); |
||
146 | aggregates.put( |
||
147 | attr.getName(), |
||
148 | (AggregateOperation) aggregatesOperationFactories.get(operation).create() |
||
149 | ); |
||
150 | } |
||
151 | } |
||
152 | |||
153 | // distictOn
|
||
154 | storeResults.edit(FeatureStore.MODE_APPEND); |
||
155 | |||
156 | FeatureSet features; |
||
157 | |||
158 | status.message(i18n.getTranslation("_Getting_features_from_query"));
|
||
159 | |||
160 | if (query != null) { |
||
161 | features = inputStore.getFeatureSet(query); |
||
162 | } else {
|
||
163 | features = inputStore.getFeatureSet(); |
||
164 | } |
||
165 | EditableFeature aggregateFeature = null;
|
||
166 | |||
167 | status.message(i18n.getTranslation("_Processing_features"));
|
||
168 | status.setRangeOfValues(0, features.size());
|
||
169 | |||
170 | for (Feature feature : features) {
|
||
171 | if (status.isCancellationRequested()) {
|
||
172 | DisposeUtils.disposeQuietly(features); |
||
173 | if (storeResults.isEditing()) {
|
||
174 | storeResults.cancelEditingQuietly(); |
||
175 | } |
||
176 | |||
177 | status.setTitle(i18n.getTranslation("_Post_process_canceled"));
|
||
178 | status.message("");
|
||
179 | status.cancel(); |
||
180 | return null; |
||
181 | } |
||
182 | if (aggregateFeature != null) { |
||
183 | if (Objects.equals(feature.get(uniqueFieldName), aggregateFeature.get(uniqueFieldName))) {
|
||
184 | for (Map.Entry<String, AggregateOperation> entry : aggregates.entrySet()) { |
||
185 | String name = entry.getKey();
|
||
186 | AggregateOperation operation = entry.getValue(); |
||
187 | Object value = feature.get(name);
|
||
188 | if (operation.isApplicable(value)) {
|
||
189 | operation.perform(value); |
||
190 | } |
||
191 | aggregateFeature.set(name, operation.getValue()); |
||
192 | } |
||
193 | } else {
|
||
194 | storeResults.insert(aggregateFeature); |
||
195 | aggregateFeature = null;
|
||
196 | } |
||
197 | } |
||
198 | |||
199 | if (aggregateFeature == null) { |
||
200 | aggregateFeature = storeResults.createNewFeature(feature); |
||
201 | for (Map.Entry<String, AggregateOperation> entry : aggregates.entrySet()) { |
||
202 | String name = entry.getKey();
|
||
203 | AggregateOperation operation = entry.getValue(); |
||
204 | operation.reset(); |
||
205 | Object value = feature.get(name);
|
||
206 | if (operation.isApplicable(value)) {
|
||
207 | operation.perform(value); |
||
208 | } |
||
209 | aggregateFeature.set(name, operation.getValue()); |
||
210 | } |
||
211 | } |
||
212 | status.incrementCurrentValue(); |
||
213 | } |
||
214 | if (aggregateFeature != null) { |
||
215 | storeResults.insert(aggregateFeature); |
||
216 | } |
||
217 | |||
218 | storeResults.finishEditing(); |
||
219 | 45537 | jolivas | // storeUValues.finishEditing();
|
220 | 46062 | omartinez | |
221 | SearchPostProcessResult searchPostProcessResult = new DefaultSearchPostProcessResult(storeResults, null); |
||
222 | status.setTitle(i18n.getTranslation("_Post_process_ended"));
|
||
223 | status.message("");
|
||
224 | return searchPostProcessResult;
|
||
225 | |||
226 | } catch (DataException | ValidateDataParametersException ex) {
|
||
227 | LOGGER.warn("Can't execute distinct on search post process.", ex);
|
||
228 | } finally {
|
||
229 | status.pop(); |
||
230 | } |
||
231 | |||
232 | return null; |
||
233 | } |
||
234 | |||
235 | private static Map<String, AggregateOperationFactory> aggregatesOperationFactories = new HashMap<>(); |
||
236 | |||
237 | public static void registerAggregateOperation(AggregateOperationFactory aggregateOperationFactory) { |
||
238 | aggregatesOperationFactories.put(aggregateOperationFactory.getName(), aggregateOperationFactory); |
||
239 | } |
||
240 | |||
241 | public static Collection<AggregateOperationFactory> getAggregatesOperationFactories() { |
||
242 | return aggregatesOperationFactories.values();
|
||
243 | } |
||
244 | |||
245 | 45537 | jolivas | } |