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 @ 45705
History | View | Annotate | Download (9.26 KB)
1 |
/*
|
---|---|
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 |
|
9 |
import java.io.File; |
10 |
import java.util.Collection; |
11 |
import java.util.HashMap; |
12 |
import java.util.Map; |
13 |
import java.util.Objects; |
14 |
import java.util.UUID; |
15 |
import javax.json.JsonObject; |
16 |
|
17 |
import org.gvsig.fmap.dal.DALLocator; |
18 |
import org.gvsig.fmap.dal.DataManager; |
19 |
import org.gvsig.fmap.dal.DataStoreParameters; |
20 |
import org.gvsig.fmap.dal.exception.DataException; |
21 |
import org.gvsig.fmap.dal.exception.ValidateDataParametersException; |
22 |
import org.gvsig.fmap.dal.feature.EditableFeature; |
23 |
import org.gvsig.fmap.dal.feature.EditableFeatureAttributeDescriptor; |
24 |
import org.gvsig.fmap.dal.feature.EditableFeatureType; |
25 |
import org.gvsig.fmap.dal.feature.Feature; |
26 |
import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor; |
27 |
import org.gvsig.fmap.dal.feature.FeatureAttributeEmulator; |
28 |
import org.gvsig.fmap.dal.feature.FeatureExtraColumns; |
29 |
import org.gvsig.fmap.dal.feature.FeatureQuery; |
30 |
import org.gvsig.fmap.dal.feature.FeatureSet; |
31 |
import org.gvsig.fmap.dal.feature.FeatureStore; |
32 |
import org.gvsig.fmap.dal.feature.FeatureType; |
33 |
import org.gvsig.fmap.dal.feature.impl.DefaultEditableFeatureAttributeDescriptor; |
34 |
import org.gvsig.fmap.dal.feature.impl.DefaultFeatureAttributeDescriptor; |
35 |
import org.gvsig.fmap.dal.store.jdbc.JDBCNewStoreParameters; |
36 |
import org.gvsig.fmap.dal.store.jdbc.JDBCServerExplorer; |
37 |
import org.gvsig.fmap.dal.store.jdbc.JDBCServerExplorerParameters; |
38 |
import org.gvsig.fmap.dal.swing.impl.searchPostProcess.distinctOn.AggregateOperation.AggregateOperationFactory; |
39 |
import org.gvsig.fmap.dal.swing.searchPostProcess.AbstractSearchPostProcess; |
40 |
import org.gvsig.fmap.dal.swing.searchPostProcess.SearchPostProcessFactory; |
41 |
import org.gvsig.tools.ToolsLocator; |
42 |
import org.gvsig.tools.dataTypes.DataType; |
43 |
import org.gvsig.tools.dynobject.DynObject; |
44 |
import org.gvsig.tools.dynobject.DynStruct; |
45 |
import org.gvsig.tools.folders.FoldersManager; |
46 |
import org.gvsig.tools.util.HasAFile; |
47 |
import org.slf4j.LoggerFactory; |
48 |
|
49 |
|
50 |
/**
|
51 |
*
|
52 |
* @author jovivas
|
53 |
*/
|
54 |
public class DistinctOn extends AbstractSearchPostProcess{ |
55 |
|
56 |
private static final org.slf4j.Logger LOGGER = LoggerFactory.getLogger(DistinctOnFactory.class); |
57 |
|
58 |
public DistinctOn (
|
59 |
SearchPostProcessFactory factory, |
60 |
FeatureStore input, |
61 |
FeatureQuery query, |
62 |
DynObject parameters |
63 |
|
64 |
){ |
65 |
super( factory, input, query, parameters);
|
66 |
|
67 |
} |
68 |
|
69 |
|
70 |
@Override
|
71 |
public SearchPostProcessResult execute (FeatureStore inputStore, FeatureQuery query, DynObject parameters){
|
72 |
String uniqueFieldName;
|
73 |
|
74 |
uniqueFieldName = (String) parameters.getDynValue("field"); |
75 |
|
76 |
try {
|
77 |
// crear ruta de archivo temporal
|
78 |
FoldersManager foldersManager = ToolsLocator.getFoldersManager(); |
79 |
File tempFile = foldersManager.getUniqueTemporaryFile("distinctOn_process_"+ UUID.randomUUID().toString()); |
80 |
|
81 |
// crear SERVER STORE
|
82 |
DataManager dataManager = DALLocator.getDataManager(); |
83 |
JDBCServerExplorerParameters serverParameters = (JDBCServerExplorerParameters) dataManager.createServerExplorerParameters("H2Spatial");
|
84 |
((HasAFile) serverParameters).setFile(tempFile); |
85 |
JDBCServerExplorer serverExplorer = (JDBCServerExplorer) dataManager.openServerExplorer("H2Spatial",serverParameters);
|
86 |
|
87 |
|
88 |
//Crear tablas en server store
|
89 |
JDBCNewStoreParameters parametersResults = serverExplorer.getAddParameters(); |
90 |
parametersResults.setDynValue("Table", "results"); |
91 |
FeatureType ft_inputStore = inputStore.getDefaultFeatureType(); |
92 |
EditableFeatureType ft_inputStoreNoContainer = dataManager.createFeatureType(); |
93 |
ft_inputStoreNoContainer.addAll(ft_inputStore); |
94 |
for (FeatureAttributeDescriptor attr : ft_inputStore) {
|
95 |
DataType dataType = attr.getDataType(); |
96 |
if (dataType.isContainer()){
|
97 |
ft_inputStoreNoContainer.remove(attr.getName()); |
98 |
} |
99 |
} |
100 |
if (query.getExtraColumn().size()>0) { |
101 |
FeatureExtraColumns extraColumns = query.getExtraColumn(); |
102 |
for (int i = 0; i < extraColumns.size(); i++) { |
103 |
DataType dataType = extraColumns.get(i).getDataType(); |
104 |
if (!dataType.isContainer()){
|
105 |
ft_inputStoreNoContainer.add(extraColumns.get(i).getName(), extraColumns.get(i).getType(), extraColumns.get(i).getSize()); |
106 |
} |
107 |
} |
108 |
} |
109 |
EditableFeatureType ft_outputStore = dataManager.createFeatureType(); |
110 |
ft_outputStore.addAll(ft_inputStoreNoContainer); |
111 |
for (FeatureAttributeDescriptor attr : ft_outputStore) {
|
112 |
if(attr.isComputed()){
|
113 |
((DefaultFeatureAttributeDescriptor) attr).setFeatureAttributeEmulator((FeatureAttributeEmulator) null);
|
114 |
} |
115 |
} |
116 |
parametersResults.setDefaultFeatureType(ft_outputStore); |
117 |
|
118 |
serverExplorer.add("H2Spatial", parametersResults, true); |
119 |
|
120 |
|
121 |
DataStoreParameters storeParametersResults = dataManager.createStoreParameters("H2Spatial");
|
122 |
storeParametersResults.setDynValue("database_file",tempFile);
|
123 |
storeParametersResults.setDynValue("Table","results"); |
124 |
|
125 |
//Creaci?n del store con los resultados
|
126 |
FeatureStore storeResults = (FeatureStore) dataManager.openStore("H2Spatial",storeParametersResults);
|
127 |
|
128 |
// Creaci?n del Map con las funciones de agragado
|
129 |
Map <String, AggregateOperation> aggregates = new HashMap <>(); |
130 |
for (FeatureAttributeDescriptor attr : ft_inputStoreNoContainer) {
|
131 |
if (parameters.hasDynValue(attr.getName())){
|
132 |
String operation = (String) parameters.getDynValue(attr.getName()); |
133 |
aggregates.put( |
134 |
attr.getName(), |
135 |
(AggregateOperation) aggregatesOperationFactories.get(operation).create() |
136 |
); |
137 |
} |
138 |
} |
139 |
|
140 |
// distictOn
|
141 |
|
142 |
storeResults.edit(FeatureStore.MODE_APPEND); |
143 |
|
144 |
FeatureSet features; |
145 |
if (query != null){ |
146 |
features = inputStore.getFeatureSet(query); |
147 |
}else{
|
148 |
features = inputStore.getFeatureSet(); |
149 |
} |
150 |
EditableFeature lastFeature = null;
|
151 |
for (Feature feature : features) {
|
152 |
if (lastFeature == null){ |
153 |
lastFeature=storeResults.createNewFeature(feature); |
154 |
for (AggregateOperation aggregate : aggregates.values()) {
|
155 |
aggregate.reset(); |
156 |
} |
157 |
} |
158 |
if ( Objects.equals(feature.get(uniqueFieldName), lastFeature.get(uniqueFieldName))){
|
159 |
for (Map.Entry<String, AggregateOperation> entry : aggregates.entrySet()) { |
160 |
String name = entry.getKey();
|
161 |
AggregateOperation operation = entry.getValue(); |
162 |
Object value = feature.get(name);
|
163 |
if (operation.isApplicable(value)){
|
164 |
operation.perform(value); |
165 |
} |
166 |
lastFeature.set(name, operation.getValue()); |
167 |
} |
168 |
}else{
|
169 |
storeResults.insert(lastFeature); |
170 |
lastFeature = storeResults.createNewFeature(feature); |
171 |
for (Map.Entry<String, AggregateOperation> entry : aggregates.entrySet()) { |
172 |
String name = entry.getKey();
|
173 |
AggregateOperation operation = entry.getValue(); |
174 |
operation.reset(); |
175 |
Object value = feature.get(name);
|
176 |
if (operation.isApplicable(value)){
|
177 |
operation.perform(value); |
178 |
} |
179 |
lastFeature.set(name, operation.getValue()); |
180 |
} |
181 |
} |
182 |
} |
183 |
|
184 |
storeResults.insert(lastFeature); |
185 |
|
186 |
storeResults.finishEditing(); |
187 |
// storeUValues.finishEditing();
|
188 |
|
189 |
SearchPostProcessResult searchPostProcessResult = new DefaultSearchPostProcessResult(storeResults,null); |
190 |
return searchPostProcessResult;
|
191 |
|
192 |
} catch (DataException | ValidateDataParametersException ex) {
|
193 |
LOGGER.warn("Can't execute distinct on search post process.", ex);
|
194 |
} |
195 |
|
196 |
return null; |
197 |
} |
198 |
|
199 |
private static Map<String, AggregateOperationFactory> aggregatesOperationFactories = new HashMap<>(); |
200 |
public static void registerAggregateOperation(AggregateOperationFactory aggregateOperationFactory){ |
201 |
aggregatesOperationFactories.put(aggregateOperationFactory.getName(), aggregateOperationFactory); |
202 |
} |
203 |
|
204 |
public static Collection<AggregateOperationFactory> getAggregatesOperationFactories(){ |
205 |
return aggregatesOperationFactories.values();
|
206 |
} |
207 |
|
208 |
|
209 |
|
210 |
} |
211 |
|