← Back to branch summary

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

« back to all changes in this revision

Viewing changes to local/in/dhis-web-dashboard/src/main/java/org/hisp/dhis/dashboard/ds/action/GenerateCommentsResultAction.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:
local/in/dhis-web-dashboard/src/main/java/org/hisp/dhis/dashboard/ds/action/GenerateCommentsResultAction.java
 
1
package org.hisp.dhis.dashboard.ds.action;
 
2
 
 
3
import java.sql.Connection;
 
4
import java.sql.PreparedStatement;
 
5
import java.sql.ResultSet;
 
6
import java.util.ArrayList;
 
7
import java.util.Collection;
 
8
import java.util.Collections;
 
9
import java.util.HashMap;
 
10
import java.util.HashSet;
 
11
import java.util.Iterator;
 
12
import java.util.List;
 
13
import java.util.Map;
 
14
import java.util.Set;
 
15
 
 
16
import org.hisp.dhis.dashboard.util.DBConnection;
 
17
import org.hisp.dhis.dashboard.util.DashBoardService;
 
18
import org.hisp.dhis.dataelement.DataElement;
 
19
import org.hisp.dhis.dataelement.DataElementService;
 
20
import org.hisp.dhis.dataset.DataSet;
 
21
import org.hisp.dhis.dataset.DataSetStore;
 
22
import org.hisp.dhis.datavalue.DataValueService;
 
23
import org.hisp.dhis.organisationunit.OrganisationUnit;
 
24
import org.hisp.dhis.organisationunit.OrganisationUnitGroup;
 
25
import org.hisp.dhis.organisationunit.OrganisationUnitGroupService;
 
26
import org.hisp.dhis.organisationunit.OrganisationUnitService;
 
27
import org.hisp.dhis.organisationunit.comparator.OrganisationUnitShortNameComparator;
 
28
import org.hisp.dhis.period.Period;
 
29
import org.hisp.dhis.period.PeriodStore;
 
30
import org.hisp.dhis.period.PeriodType;
 
31
import org.hisp.dhis.source.Source;
 
32
 
 
33
import com.opensymphony.xwork.Action;
 
34
 
 
35
public class GenerateCommentsResultAction implements Action
 
36
{
 
37
 
 
38
    // ---------------------------------------------------------------
 
39
    // Dependencies
 
40
    // ---------------------------------------------------------------
 
41
    private OrganisationUnitService organisationUnitService;
 
42
 
 
43
    public void setOrganisationUnitService( OrganisationUnitService organisationUnitService )
 
44
    {
 
45
        this.organisationUnitService = organisationUnitService;
 
46
    }
 
47
 
 
48
    public OrganisationUnitService getOrganisationUnitService()
 
49
    {
 
50
        return organisationUnitService;
 
51
    }
 
52
    
 
53
    private OrganisationUnitGroupService organisationUnitGroupService;
 
54
 
 
55
    public void setOrganisationUnitGroupService( OrganisationUnitGroupService organisationUnitGroupService )
 
56
    {
 
57
        this.organisationUnitGroupService = organisationUnitGroupService;
 
58
    }
 
59
 
 
60
    private PeriodStore periodStore;
 
61
 
 
62
    public void setPeriodStore( PeriodStore periodStore )
 
63
    {
 
64
        this.periodStore = periodStore;
 
65
    }
 
66
 
 
67
    private DataSetStore dataSetStore;
 
68
 
 
69
    public void setDataSetStore( DataSetStore dataSetStore )
 
70
    {
 
71
        this.dataSetStore = dataSetStore;
 
72
    }
 
73
 
 
74
    public DataSetStore getDataSetStore()
 
75
    {
 
76
        return dataSetStore;
 
77
    }
 
78
 
 
79
    private DashBoardService dashBoardService;
 
80
 
 
81
    public void setDashBoardService( DashBoardService dashBoardService )
 
82
    {
 
83
        this.dashBoardService = dashBoardService;
 
84
    }
 
85
 
 
86
    private DataValueService dataValueService;
 
87
 
 
88
    public void setDataValueService( DataValueService dataValueService )
 
89
    {
 
90
        this.dataValueService = dataValueService;
 
91
    }
 
92
 
 
93
    private DataElementService dataElementService;
 
94
    
 
95
    public void setDataElementService( DataElementService dataElementService )
 
96
    {
 
97
        this.dataElementService = dataElementService;
 
98
    }
 
99
 
 
100
    // ---------------------------------------------------------------
 
101
    // Output Parameters
 
102
    // ---------------------------------------------------------------
 
103
 
 
104
    private List<OrganisationUnit> orgUnitList;
 
105
 
 
106
    public List<OrganisationUnit> getOrgUnitList()
 
107
    {
 
108
        return orgUnitList;
 
109
    }
 
110
 
 
111
    private List<DataSet> dataSetList;
 
112
 
 
113
    public List<DataSet> getDataSetList()
 
114
    {
 
115
        return dataSetList;
 
116
    }
 
117
 
 
118
    private List<Integer> results;
 
119
 
 
120
    public List<Integer> getResults()
 
121
    {
 
122
        return results;
 
123
    }
 
124
 
 
125
    private Map<DataSet, Map<OrganisationUnit, List<Integer>>> dataStatusResult;
 
126
 
 
127
    public Map<DataSet, Map<OrganisationUnit, List<Integer>>> getDataStatusResult()
 
128
    {
 
129
        return dataStatusResult;
 
130
    }
 
131
 
 
132
    private Map<DataSet, Collection<Period>> dataSetPeriods;
 
133
 
 
134
    public Map<DataSet, Collection<Period>> getDataSetPeriods()
 
135
    {
 
136
        return dataSetPeriods;
 
137
    }
 
138
 
 
139
    List<Period> selectedPeriodList;
 
140
 
 
141
    public List<Period> getSelectedPeriodList()
 
142
    {
 
143
        return selectedPeriodList;
 
144
    }
 
145
 
 
146
    List<String> levelNames;
 
147
 
 
148
    public List<String> getLevelNames()
 
149
    {
 
150
        return levelNames;
 
151
    }
 
152
 
 
153
    private int maxOULevel;
 
154
    
 
155
    public int getMaxOULevel()
 
156
    {
 
157
        return maxOULevel;
 
158
    }
 
159
 
 
160
    
 
161
    // ---------------------------------------------------------------
 
162
    // Input Parameters
 
163
    // ---------------------------------------------------------------
 
164
 
 
165
    private String dsId;
 
166
 
 
167
    public void setDsId( String dsId )
 
168
    {
 
169
        this.dsId = dsId;
 
170
    }
 
171
 
 
172
    private String ouId;
 
173
    public void setOuId( String ouId )
 
174
    {
 
175
        this.ouId = ouId;
 
176
    }
 
177
 
 
178
    private String immChildOption;
 
179
 
 
180
    public void setImmChildOption( String immChildOption )
 
181
    {
 
182
        this.immChildOption = immChildOption;
 
183
    }
 
184
 
 
185
    private int sDateLB;
 
186
 
 
187
    public void setSDateLB( int dateLB )
 
188
    {
 
189
        sDateLB = dateLB;
 
190
    }
 
191
 
 
192
    public int getSDateLB()
 
193
    {
 
194
        return sDateLB;
 
195
    }
 
196
    
 
197
    private int eDateLB;
 
198
 
 
199
    public void setEDateLB( int dateLB )
 
200
    {
 
201
        eDateLB = dateLB;
 
202
    }
 
203
 
 
204
    public int getEDateLB()
 
205
    {
 
206
        return eDateLB;
 
207
    }
 
208
 
 
209
    private String facilityLB;
 
210
 
 
211
    public void setFacilityLB( String facilityLB )
 
212
    {
 
213
        this.facilityLB = facilityLB;
 
214
    }
 
215
 
 
216
    private List<String> orgUnitListCB;
 
217
 
 
218
    public void setOrgUnitListCB( List<String> orgUnitListCB )
 
219
    {
 
220
        this.orgUnitListCB = orgUnitListCB;
 
221
    }
 
222
 
 
223
    private List<String> selectedDataSets;
 
224
 
 
225
    public void setSelectedDataSets( List<String> selectedDataSets )
 
226
    {
 
227
        this.selectedDataSets = selectedDataSets;
 
228
    }
 
229
 
 
230
    public List<String> getSelectedDataSets()
 
231
    {
 
232
        return selectedDataSets;
 
233
    }
 
234
 
 
235
    private int minOULevel;
 
236
            
 
237
    public int getMinOULevel()
 
238
    {
 
239
        return minOULevel;
 
240
    }
 
241
 
 
242
    private Connection con = null;
 
243
    String orgUnitInfo;
 
244
    String periodInfo;
 
245
    String deInfo;
 
246
    int orgUnitCount;
 
247
    private String dataTableName;
 
248
    // ---------------------------------------------------------------
 
249
    // Action Implementation
 
250
    // ---------------------------------------------------------------
 
251
    public String execute()
 
252
        throws Exception
 
253
    {
 
254
        con = (new DBConnection()).openConnection();
 
255
        orgUnitCount = 0;
 
256
        dataTableName = "";
 
257
        
 
258
        // Intialization
 
259
        results = new ArrayList<Integer>();
 
260
        maxOULevel = 1;
 
261
        minOULevel = organisationUnitService.getNumberOfOrganisationalLevels();
 
262
        
 
263
        
 
264
        if(immChildOption!= null && immChildOption.equalsIgnoreCase( "yes" ))
 
265
        {
 
266
            orgUnitListCB = new ArrayList<String>();
 
267
            orgUnitListCB.add( ouId );
 
268
            
 
269
            facilityLB = "immChildren";
 
270
            
 
271
            selectedDataSets = new ArrayList<String>();
 
272
            selectedDataSets.add( dsId );
 
273
        }
 
274
        
 
275
        // OrgUnit Related Info
 
276
        OrganisationUnit selectedOrgUnit = new OrganisationUnit();
 
277
        orgUnitList = new ArrayList<OrganisationUnit>();
 
278
        if ( facilityLB.equals( "children" ) )
 
279
        {
 
280
            selectedOrgUnit = organisationUnitService.getOrganisationUnit( Integer.parseInt( orgUnitListCB.get( 0 ) ) );            
 
281
            orgUnitList = getChildOrgUnitTree( selectedOrgUnit );
 
282
        }
 
283
        else if ( facilityLB.equals( "immChildren" ) )
 
284
        {
 
285
            selectedOrgUnit = organisationUnitService.getOrganisationUnit( Integer.parseInt( orgUnitListCB.get( 0 ) ) );
 
286
 
 
287
            // This is hard coded to get District and Corporation List
 
288
            if(organisationUnitService.getLevelOfOrganisationUnit( selectedOrgUnit ) == 1)
 
289
            {                
 
290
                orgUnitList = new ArrayList<OrganisationUnit>();            
 
291
                orgUnitList.add( selectedOrgUnit );
 
292
                
 
293
                OrganisationUnitGroup oug = organisationUnitGroupService.getOrganisationUnitGroupByName( "Districts" );                
 
294
                if(oug != null)
 
295
                {
 
296
                    List<OrganisationUnit> tempOUList = new ArrayList<OrganisationUnit>(oug.getMembers());
 
297
                    Collections.sort( tempOUList, new OrganisationUnitShortNameComparator() );
 
298
                    orgUnitList.addAll( tempOUList );
 
299
                }   
 
300
                
 
301
                oug = organisationUnitGroupService.getOrganisationUnitGroupByName( "Corporations" );
 
302
                if(oug != null)
 
303
                {
 
304
                    List<OrganisationUnit> tempOUList = new ArrayList<OrganisationUnit>(oug.getMembers());
 
305
                    Collections.sort( tempOUList, new OrganisationUnitShortNameComparator() );
 
306
                    orgUnitList.addAll( tempOUList );
 
307
                }   
 
308
            }
 
309
            else
 
310
            {
 
311
                orgUnitList = new ArrayList<OrganisationUnit>();
 
312
                
 
313
                Iterator<String> orgUnitIterator = orgUnitListCB.iterator();                
 
314
                while ( orgUnitIterator.hasNext() )
 
315
                {
 
316
                    OrganisationUnit o = organisationUnitService.getOrganisationUnit( Integer.parseInt( (String) orgUnitIterator.next() ) );
 
317
                    orgUnitList.add( o );
 
318
                    List<OrganisationUnit> organisationUnits = new ArrayList<OrganisationUnit>( o.getChildren() );
 
319
                    Collections.sort( organisationUnits, new OrganisationUnitShortNameComparator() );
 
320
                    orgUnitList.addAll( organisationUnits );
 
321
                }                
 
322
            }
 
323
        }
 
324
        else
 
325
        {
 
326
            Iterator<String> orgUnitIterator = orgUnitListCB.iterator();
 
327
            OrganisationUnit o;
 
328
            while ( orgUnitIterator.hasNext() )
 
329
            {
 
330
                o = organisationUnitService.getOrganisationUnit( Integer
 
331
                    .parseInt( orgUnitIterator.next() ) );
 
332
                orgUnitList.add( o );
 
333
            }
 
334
        }
 
335
 
 
336
        orgUnitInfo = "-1";
 
337
        for(OrganisationUnit ou : orgUnitList)
 
338
        {
 
339
            orgUnitInfo += "," + ou.getId();
 
340
            getOrgUnitInfo(ou);
 
341
        }    
 
342
            
 
343
        // Period Related Info
 
344
        Period startPeriod = periodStore.getPeriod( sDateLB );
 
345
        Period endPeriod = periodStore.getPeriod( eDateLB );
 
346
 
 
347
        selectedPeriodList = dashBoardService.getMonthlyPeriods( startPeriod.getStartDate(), endPeriod.getEndDate() );
 
348
        
 
349
        periodInfo = "-1";
 
350
        for(Period p : selectedPeriodList)            
 
351
            periodInfo +=  "," + p.getId();
 
352
        
 
353
        // DataSet Related Info
 
354
        dataSetList = new ArrayList<DataSet>();
 
355
 
 
356
        deInfo = "-1";
 
357
        for(String ds : selectedDataSets)
 
358
        {
 
359
            DataSet dSet = dataSetStore.getDataSet( Integer.parseInt( ds ) );
 
360
            for(DataElement de : dSet.getDataElements())
 
361
                deInfo += "," + de.getId();
 
362
        }
 
363
 
 
364
        System.out.println(orgUnitInfo);
 
365
        System.out.println(deInfo);
 
366
        System.out.println(periodInfo);
 
367
        
 
368
        dataTableName = dashBoardService.createDataTableForComments(orgUnitInfo, deInfo, periodInfo);
 
369
        
 
370
        dataSetPeriods = new HashMap<DataSet, Collection<Period>>();
 
371
        Iterator dataSetIterator = selectedDataSets.iterator();
 
372
        
 
373
        DataSet ds;
 
374
        Collection<DataElement> dataElements = new ArrayList<DataElement>();
 
375
        PeriodType dataSetPeriodType;
 
376
        Collection<Period> periodList = new ArrayList<Period>();
 
377
        
 
378
        while ( dataSetIterator.hasNext() )
 
379
        {            
 
380
            ds = dataSetStore.getDataSet( Integer.parseInt( (String) dataSetIterator.next() ) );
 
381
            dataSetList.add( ds );
 
382
            dataElements = ds.getDataElements();
 
383
            deInfo = getDEInfo(dataElements);
 
384
                        
 
385
            dataSetPeriodType = ds.getPeriodType();
 
386
            
 
387
            periodList = periodStore.getIntersectingPeriodsByPeriodType( dataSetPeriodType,
 
388
                startPeriod.getStartDate(), endPeriod.getEndDate() );
 
389
            dataSetPeriods.put( ds, periodList );
 
390
 
 
391
            Iterator orgUnitListIterator = orgUnitList.iterator();
 
392
            OrganisationUnit o;
 
393
            Set<Source> dso = new HashSet<Source>();
 
394
            Iterator periodIterator;
 
395
            
 
396
            while ( orgUnitListIterator.hasNext() )
 
397
            {
 
398
                
 
399
                o = (OrganisationUnit) orgUnitListIterator.next();
 
400
                orgUnitInfo = ""+o.getId();
 
401
                
 
402
                if(maxOULevel < organisationUnitService.getLevelOfOrganisationUnit( o ))
 
403
                    maxOULevel = organisationUnitService.getLevelOfOrganisationUnit( o );
 
404
                
 
405
                if(minOULevel > organisationUnitService.getLevelOfOrganisationUnit( o ))
 
406
                    minOULevel = organisationUnitService.getLevelOfOrganisationUnit( o );
 
407
                    
 
408
                dso = ds.getSources();
 
409
                periodIterator = periodList.iterator();
 
410
                
 
411
                Period p;
 
412
                Collection dataValueResult;
 
413
                double dataStatusPercentatge;
 
414
                
 
415
                while ( periodIterator.hasNext() )
 
416
                {                    
 
417
                    p = (Period) periodIterator.next();
 
418
                    periodInfo = ""+p.getId();
 
419
                    
 
420
                    PreparedStatement ps1 = null;
 
421
                    ResultSet rs1 = null;
 
422
                    String query = "";
 
423
                                                                               
 
424
                    if ( dso == null )
 
425
                    {
 
426
                        results.add( -1 );
 
427
                        continue;
 
428
                    }
 
429
                    else if(!dso.contains( o ))
 
430
                    {                        
 
431
                        orgUnitInfo = "-1";
 
432
                        orgUnitCount = 0;
 
433
                        getOrgUnitInfo(o, dso);
 
434
                        
 
435
                        query = "SELECT sourceid, dataelementid, periodid, comment FROM "+ dataTableName +" WHERE dataelementid IN ("+deInfo+") AND sourceid IN ("+orgUnitInfo+") AND periodid IN ("+periodInfo+")";
 
436
                        ps1 = con.prepareStatement(query);
 
437
                        rs1 = ps1.executeQuery(query);
 
438
                        
 
439
                        while(rs1.next())
 
440
                        {
 
441
                            OrganisationUnit ou = organisationUnitService.getOrganisationUnit( rs1.getInt( 1 ) );
 
442
                            DataElement de = dataElementService.getDataElement( rs1.getInt( 2 ));
 
443
                            Period per = periodStore.getPeriod( rs1.getInt(3) );
 
444
                            
 
445
                            String tempStr = ou.getShortName() + " --- " + de.getName(); 
 
446
                        }                            
 
447
                        
 
448
                        
 
449
                        //results.add( new Integer( (int) dataStatusPercentatge ) );
 
450
                        continue;
 
451
                    }
 
452
                    
 
453
                    /*
 
454
                    dataValueResult = dataValueService.getDataValues( o, p, dataElements );
 
455
                    dataStatusPercentatge = ((double) dataValueResult.size() / (double) dataElements.size()) * 100.0;
 
456
                 
 
457
                    */
 
458
                    
 
459
                    orgUnitInfo = ""+ o.getId();
 
460
                    query = "SELECT COUNT(*) FROM "+ dataTableName +" WHERE dataelementid IN ("+deInfo+") AND sourceid IN ("+orgUnitInfo+") AND periodid IN ("+periodInfo+")";
 
461
                    ps1 = con.prepareStatement(query);
 
462
                    rs1 = ps1.executeQuery(query);
 
463
 
 
464
                    if(rs1.next())
 
465
                    {
 
466
                        try
 
467
                        {
 
468
                            dataStatusPercentatge = ((double) rs1.getInt( 1 ) / (double) dataElements.size())*100.0;    
 
469
                        }
 
470
                        catch(Exception e)
 
471
                        {
 
472
                            dataStatusPercentatge = 0.0;
 
473
                        }
 
474
                    }                            
 
475
                    else
 
476
                        dataStatusPercentatge = 0.0;
 
477
                   
 
478
                    if(dataStatusPercentatge > 100.0) dataStatusPercentatge = 100;
 
479
                    
 
480
                    dataStatusPercentatge = Math.round( dataStatusPercentatge * Math.pow( 10, 0 ) ) / Math.pow( 10, 0 );                    
 
481
 
 
482
                    results.add( ( int ) dataStatusPercentatge );
 
483
                    
 
484
                    try
 
485
                    {
 
486
                        
 
487
                    }
 
488
                    finally
 
489
                    {
 
490
                        try
 
491
                        {
 
492
                            if(rs1 != null) rs1.close();
 
493
                            if(ps1 != null) ps1.close();
 
494
                        }
 
495
                        catch( Exception e )
 
496
                        {
 
497
                            System.out.println("Exception while closing Prepared Statement, ResultSet");
 
498
                        }
 
499
                    }              
 
500
 
 
501
                    
 
502
                }
 
503
                // System.out.println( o.getShortName() );
 
504
            }
 
505
        }
 
506
 
 
507
        // For Level Names
 
508
        String ouLevelNames[] = { " ", "State", "District", "Block", "PHC", "Subcentre"};
 
509
        levelNames = new ArrayList<String>();
 
510
        int count1 = minOULevel;
 
511
        while ( count1 <= maxOULevel )
 
512
        {
 
513
            levelNames.add( ouLevelNames[count1] );
 
514
            count1++;
 
515
        }
 
516
 
 
517
        try
 
518
        {
 
519
            
 
520
        }
 
521
        finally
 
522
        {
 
523
            try
 
524
            {
 
525
                if(con != null) con.close();
 
526
            }
 
527
            catch( Exception e )
 
528
            {
 
529
                System.out.println("Exception while closing DB Connections : "+e.getMessage());
 
530
            }
 
531
        }// finally block end
 
532
 
 
533
        dashBoardService.deleteDataTable( dataTableName );
 
534
        
 
535
        return SUCCESS;
 
536
    }
 
537
 
 
538
    // Returns the OrgUnitTree for which Root is the orgUnit
 
539
    public List<OrganisationUnit> getChildOrgUnitTree( OrganisationUnit orgUnit )
 
540
    {
 
541
        List<OrganisationUnit> orgUnitTree = new ArrayList<OrganisationUnit>();
 
542
        orgUnitTree.add( orgUnit );
 
543
 
 
544
        Collection<OrganisationUnit> children = orgUnit.getChildren();
 
545
 
 
546
        Iterator childIterator = children.iterator();
 
547
        OrganisationUnit child;
 
548
        while ( childIterator.hasNext() )
 
549
        {
 
550
            child = (OrganisationUnit) childIterator.next();
 
551
            orgUnitTree.addAll( getChildOrgUnitTree( child ) );
 
552
        }
 
553
        return orgUnitTree;
 
554
    }// getChildOrgUnitTree end
 
555
 
 
556
 
 
557
    private void getOrgUnitInfo(OrganisationUnit organisationUnit)
 
558
    {                
 
559
        Collection<OrganisationUnit> children = organisationUnit.getChildren();
 
560
 
 
561
        Iterator<OrganisationUnit> childIterator = children.iterator();
 
562
        OrganisationUnit child;
 
563
        while ( childIterator.hasNext() )
 
564
        {
 
565
            child = childIterator.next();
 
566
            orgUnitInfo += "," + child.getId();
 
567
            getOrgUnitInfo( child);
 
568
        }               
 
569
    }
 
570
 
 
571
    private void getOrgUnitInfo(OrganisationUnit organisationUnit, Set<Source> dso)
 
572
    {                
 
573
        Collection<OrganisationUnit> children = organisationUnit.getChildren();
 
574
 
 
575
        Iterator<OrganisationUnit> childIterator = children.iterator();
 
576
        OrganisationUnit child;
 
577
        while ( childIterator.hasNext() )
 
578
        {
 
579
            child = childIterator.next();
 
580
            if(dso.contains( child ))
 
581
            {
 
582
                orgUnitInfo += "," + child.getId();
 
583
                orgUnitCount++;
 
584
            }            
 
585
            getOrgUnitInfo( child, dso );
 
586
        }               
 
587
    }
 
588
 
 
589
    
 
590
    private String getDEInfo(Collection<DataElement> dataElements)
 
591
    {
 
592
        StringBuffer deInfo = new StringBuffer( "-1" );
 
593
                        
 
594
        for(DataElement de : dataElements)
 
595
        {
 
596
            deInfo.append( "," ).append( de.getId() );
 
597
        }
 
598
        return deInfo.toString();
 
599
    }
 
600
 
 
601
 
 
602
    
 
603
}// class end