1
package org.hisp.dhis.importexport.dhis14.file.rowhandler;
4
* Copyright (c) 2004-2007, University of Oslo
7
* Redistribution and use in source and binary forms, with or without
8
* modification, are permitted provided that the following conditions are met:
9
* * Redistributions of source code must retain the above copyright notice, this
10
* list of conditions and the following disclaimer.
11
* * Redistributions in binary form must reproduce the above copyright notice,
12
* this list of conditions and the following disclaimer in the documentation
13
* and/or other materials provided with the distribution.
14
* * Neither the name of the HISP project nor the names of its contributors may
15
* be used to endorse or promote products derived from this software without
16
* specific prior written permission.
18
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
19
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
20
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
21
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
22
* ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
23
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
24
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
25
* ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
27
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30
import java.util.HashSet;
33
import org.hisp.dhis.dataelement.CalculatedDataElement;
34
import org.hisp.dhis.dataelement.DataElement;
35
import org.hisp.dhis.dataelement.DataElementCategoryCombo;
36
import org.hisp.dhis.dataelement.DataElementService;
37
import org.hisp.dhis.expression.Expression;
38
import org.hisp.dhis.expression.ExpressionService;
39
import org.hisp.dhis.importexport.GroupMemberType;
40
import org.hisp.dhis.importexport.ImportObjectService;
41
import org.hisp.dhis.importexport.ImportParams;
42
import org.hisp.dhis.importexport.analysis.ImportAnalyser;
43
import org.hisp.dhis.importexport.converter.AbstractCalculatedDataElementConverter;
44
import org.hisp.dhis.importexport.mapping.NameMappingUtil;
45
import org.hisp.dhis.system.util.UUIdUtils;
47
import com.ibatis.sqlmap.client.event.RowHandler;
50
* @author Lars Helge Overland
53
public class CalculatedDataElementRowHandler
54
extends AbstractCalculatedDataElementConverter implements RowHandler
56
private ImportParams params;
58
private DataElementCategoryCombo categoryCombo;
60
private ExpressionService expressionService;
62
private Map<Integer, String> calculatedEntryMap;
64
private Map<Object, Integer> dataElementMapping;
66
private Map<Object, Integer> categoryOptionComboMapping;
68
// -------------------------------------------------------------------------
70
// -------------------------------------------------------------------------
72
public CalculatedDataElementRowHandler( ImportObjectService importObjectService,
73
DataElementService dataElementService,
75
DataElementCategoryCombo categoryCombo,
76
ImportAnalyser importAnalyser,
77
ExpressionService expressionService,
78
Map<Integer, String> calculatedEntryMap,
79
Map<Object, Integer> dataElementMapping,
80
Map<Object, Integer> categoryOptionComboMapping )
82
this.importObjectService = importObjectService;
83
this.dataElementService = dataElementService;
85
this.categoryCombo = categoryCombo;
86
this.importAnalyser = importAnalyser;
87
this.expressionService = expressionService;
88
this.calculatedEntryMap = calculatedEntryMap;
89
this.dataElementMapping = dataElementMapping;
90
this.categoryOptionComboMapping = categoryOptionComboMapping;
93
// -------------------------------------------------------------------------
94
// BatchRowHandler implementation
95
// -------------------------------------------------------------------------
97
public void handleRow( Object object )
99
final CalculatedDataElement dataElement = (CalculatedDataElement) object;
101
NameMappingUtil.addDataElementMapping( dataElement.getId(), dataElement.getName() );
102
NameMappingUtil.addDataElementAggregationOperatorMapping( dataElement.getId(), dataElement.getAggregationOperator() );
104
dataElement.setUuid( UUIdUtils.getUUId() );
105
dataElement.setActive( true );
107
if ( dataElement.getCode() != null && dataElement.getCode().trim().length() == 0 )
109
dataElement.setCode( null );
112
dataElement.setCategoryCombo( categoryCombo );
114
String expression = calculatedEntryMap.get( dataElement.getId() );
115
expression = expressionService.convertExpression( expression, dataElementMapping, categoryOptionComboMapping );
116
dataElement.setExpression( new Expression( expression, null, new HashSet<DataElement>() ) );
118
read( dataElement, GroupMemberType.NONE, params );