← Back to branch summary

~mortenoh/+junk/dhis2-detailed-import-export

« back to all changes in this revision

Viewing changes to dhis-2/dhis-services/dhis-service-importexport/src/main/java/org/hisp/dhis/importexport/ixf/exporter/DefaultIXFExportService.java

  • Committer: larshelge at gmail
  • Date: 2009-03-03 16:46:36 UTC
  • Revision ID: larshelge@gmail.com-20090303164636-2sjlrquo7ib1gf7r
Initial check-in

Show diffs side-by-side

added added

removed removed

Lines of Context:
dhis-2/dhis-services/dhis-service-importexport/src/main/java/org/hisp/dhis/importexport/ixf/exporter/DefaultIXFExportService.java
 
1
package org.hisp.dhis.importexport.ixf.exporter;
 
2
 
 
3
/*
 
4
 * Copyright (c) 2004-2007, University of Oslo
 
5
 * All rights reserved.
 
6
 *
 
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.
 
17
 *
 
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.
 
28
 */
 
29
 
 
30
import java.io.BufferedInputStream;
 
31
import java.io.IOException;
 
32
import java.io.InputStream;
 
33
import java.io.PipedInputStream;
 
34
import java.io.PipedOutputStream;
 
35
import java.util.zip.ZipEntry;
 
36
import java.util.zip.ZipOutputStream;
 
37
 
 
38
import org.amplecode.staxwax.factory.XMLFactory;
 
39
import org.amplecode.staxwax.writer.XMLWriter;
 
40
import org.hisp.dhis.datavalue.DataValueService;
 
41
import org.hisp.dhis.importexport.ExportParams;
 
42
import org.hisp.dhis.importexport.ExportPipeThread;
 
43
import org.hisp.dhis.importexport.ExportService;
 
44
import org.hisp.dhis.importexport.ixf.config.IXFConfigurationManager;
 
45
import org.hisp.dhis.importexport.ixf.converter.CountryConverter;
 
46
import org.hisp.dhis.importexport.ixf.converter.DimensionConverter;
 
47
import org.hisp.dhis.importexport.ixf.converter.FileMetaDataConverter;
 
48
import org.hisp.dhis.importexport.ixf.converter.GeoCodeConverter;
 
49
import org.hisp.dhis.importexport.ixf.converter.GeoLevelConverter;
 
50
import org.hisp.dhis.importexport.ixf.converter.GeoTypeConverter;
 
51
import org.hisp.dhis.importexport.ixf.converter.IndicatorConverter;
 
52
import org.hisp.dhis.importexport.ixf.converter.LanguageConverter;
 
53
import org.hisp.dhis.importexport.ixf.converter.SourceConverter;
 
54
import org.hisp.dhis.importexport.ixf.converter.TimePeriodConverter;
 
55
import org.hisp.dhis.organisationunit.OrganisationUnitService;
 
56
 
 
57
/**
 
58
 * @author Lars Helge Overland
 
59
 * @version $Id: DefaultIXFExportService.java 5375 2008-06-11 16:42:54Z larshelg $
 
60
 */
 
61
public class DefaultIXFExportService
 
62
    implements ExportService
 
63
{
 
64
    private static final String ZIP_ENTRY_NAME = "Export.xml";
 
65
    private static final String ROOT_NAME = "aidsSchema";
 
66
    private static final String[] ROOT_PROPERTIES = { "xmlns:xsi", "http://www.w3.org/2001/xmlschema-instance", 
 
67
                                  "xmlns", "http://www.unaids.org/ns/cris",
 
68
                                  "xmlns:cris", "http://www.unaids.org/ns/cris-ext",
 
69
                                  "xmlns:aida", "http://www.idmlinitiative.org/resources/dtds/AIDA22.xsd",
 
70
                                  "schemaVersion", "2.0" };
 
71
    
 
72
    // -------------------------------------------------------------------------
 
73
    // Dependencies
 
74
    // -------------------------------------------------------------------------
 
75
    
 
76
    private IXFConfigurationManager configurationManager;
 
77
 
 
78
    public void setConfigurationManager( IXFConfigurationManager configurationManager )
 
79
    {
 
80
        this.configurationManager = configurationManager;
 
81
    }
 
82
 
 
83
    private OrganisationUnitService organisationUnitService;
 
84
 
 
85
    public void setOrganisationUnitService( OrganisationUnitService organisationUnitService )
 
86
    {
 
87
        this.organisationUnitService = organisationUnitService;
 
88
    }
 
89
 
 
90
    private DataValueService dataValueService;
 
91
 
 
92
    public void setDataValueService( DataValueService dataValueService )
 
93
    {
 
94
        this.dataValueService = dataValueService;
 
95
    }
 
96
 
 
97
    // -------------------------------------------------------------------------
 
98
    // IXFExportService implementation
 
99
    // -------------------------------------------------------------------------
 
100
 
 
101
    public InputStream exportData( ExportParams params )
 
102
    {
 
103
        try
 
104
        {
 
105
            // -------------------------------------------------------------------------
 
106
            // Pipes are input/output pairs. Data written on the output stream shows 
 
107
            // up on the input stream at the other end of the pipe. 
 
108
            // -------------------------------------------------------------------------
 
109
            
 
110
            PipedOutputStream out = new PipedOutputStream();
 
111
            
 
112
            PipedInputStream in = new PipedInputStream( out );
 
113
 
 
114
            ZipOutputStream zipOut = new ZipOutputStream( out );
 
115
            
 
116
            zipOut.putNextEntry( new ZipEntry( ZIP_ENTRY_NAME ) );
 
117
 
 
118
            XMLWriter writer = XMLFactory.getPlainXMLWriter( zipOut );
 
119
            
 
120
            // -------------------------------------------------------------------------
 
121
            // Writes to one end of the pipe 
 
122
            // -------------------------------------------------------------------------
 
123
            
 
124
            ExportPipeThread thread = new ExportPipeThread();
 
125
            
 
126
            thread.setZipOutputStream( zipOut );
 
127
            thread.setParams( params );
 
128
            thread.setWriter( writer );
 
129
            thread.setRootName( ROOT_NAME );
 
130
            thread.setRootProperties( ROOT_PROPERTIES );
 
131
            
 
132
            thread.registerXMLConverter( new FileMetaDataConverter( configurationManager ) );
 
133
            thread.registerXMLConverter( new TimePeriodConverter() );
 
134
            thread.registerXMLConverter( new SourceConverter() );
 
135
            thread.registerXMLConverter( new LanguageConverter() );
 
136
            thread.registerXMLConverter( new GeoLevelConverter( configurationManager, organisationUnitService ) );
 
137
            thread.registerXMLConverter( new GeoTypeConverter() );
 
138
            thread.registerXMLConverter( new CountryConverter( configurationManager ) );
 
139
            thread.registerXMLConverter( new GeoCodeConverter() );
 
140
            thread.registerXMLConverter( new DimensionConverter() );
 
141
            thread.registerXMLConverter( new IndicatorConverter( dataValueService ) );
 
142
            
 
143
            thread.start();
 
144
 
 
145
            // -------------------------------------------------------------------------
 
146
            // Reads at the other end of the pipe 
 
147
            // -------------------------------------------------------------------------
 
148
            
 
149
            InputStream bis = new BufferedInputStream( in );
 
150
                                    
 
151
            return bis;
 
152
        }
 
153
        catch ( IOException ex )
 
154
        {
 
155
            throw new RuntimeException( "Error occured during export to stream", ex );
 
156
        }
 
157
    }
 
158
}