View Javadoc
1   package fr.ifremer.tutti.service.genericformat.exportactions;
2   
3   /*
4    * #%L
5    * Tutti :: Service
6    * $Id:$
7    * $HeadURL:$
8    * %%
9    * Copyright (C) 2012 - 2015 Ifremer
10   * %%
11   * This program is free software: you can redistribute it and/or modify
12   * it under the terms of the GNU General Public License as
13   * published by the Free Software Foundation, either version 3 of the
14   * License, or (at your option) any later version.
15   * 
16   * This program is distributed in the hope that it will be useful,
17   * but WITHOUT ANY WARRANTY; without even the implied warranty of
18   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19   * GNU General Public License for more details.
20   * 
21   * You should have received a copy of the GNU General Public
22   * License along with this program.  If not, see
23   * <http://www.gnu.org/licenses/gpl-3.0.html>.
24   * #L%
25   */
26  
27  import fr.ifremer.adagio.core.dao.referential.ObjectTypeCode;
28  import fr.ifremer.tutti.persistence.entities.data.Attachment;
29  import fr.ifremer.tutti.persistence.entities.data.CatchBatch;
30  import fr.ifremer.tutti.persistence.entities.data.Cruise;
31  import fr.ifremer.tutti.persistence.entities.data.FishingOperation;
32  import fr.ifremer.tutti.service.PersistenceService;
33  import fr.ifremer.tutti.service.genericformat.GenericFormatExportContext;
34  import fr.ifremer.tutti.service.genericformat.GenericFormatExportOperationContext;
35  import fr.ifremer.tutti.service.genericformat.csv.AttachmentRow;
36  import fr.ifremer.tutti.service.genericformat.csv.OperationRow;
37  import fr.ifremer.tutti.service.genericformat.producer.CsvProducerForAttachment;
38  
39  import java.util.ArrayList;
40  import java.util.List;
41  
42  /**
43   * Created on 3/28/15.
44   *
45   * @author Tony Chemit - chemit@codelutin.com
46   * @since 3.14.3
47   */
48  public class CreateOperationRowsAction extends ExportFishingOperationActionSupport {
49  
50      private final PersistenceService persistenceService;
51  
52      public CreateOperationRowsAction(PersistenceService persistenceService) {
53          this.persistenceService = persistenceService;
54      }
55  
56      @Override
57      public boolean doExecute(GenericFormatExportContext exportContext, GenericFormatExportOperationContext operationContext) {
58          return true;
59      }
60  
61      @Override
62      public void execute(GenericFormatExportContext exportContext, GenericFormatExportOperationContext operationContext) {
63  
64          FishingOperation operation = operationContext.getOperation();
65          Cruise cruise = operationContext.getCruise();
66          CatchBatch catchBatch = operationContext.getCatchBatch();
67  
68          boolean exportSpecies = exportContext.isExportSpecies();
69          boolean exportBenthos = exportContext.isExportBenthos();
70          boolean exportMarineLitter = exportContext.isExportMarineLitter();
71  
72          OperationRow operationRow = exportContext.getProducerForOperation().getDataToExport(cruise, operation, catchBatch);
73  
74          operationRow.computeCatchValues(exportSpecies, exportBenthos, exportMarineLitter);
75  
76          operationContext.setOperationRow(operationRow);
77  
78          if (exportContext.isExportAttachments()) {
79  
80              List<AttachmentRow> attachmentRows = getAttachmentRows(exportContext, operation, catchBatch);
81              operationContext.addAttachmentRows(attachmentRows);
82  
83          }
84  
85      }
86  
87      protected List<AttachmentRow> getAttachmentRows(GenericFormatExportContext exportContext, FishingOperation operation, CatchBatch catchBatch) {
88  
89          List<AttachmentRow> attachmentRows = new ArrayList<>();
90  
91          CsvProducerForAttachment producerForAttachment = exportContext.getProducerForAttachment();
92          {
93  
94              List<Attachment> attachments = persistenceService.getAllAttachments(ObjectTypeCode.OPERATION, operation.getIdAsInt());
95              producerForAttachment.addAttachments(attachments, attachmentRows);
96  
97          }
98  
99          if (catchBatch != null) {
100 
101             List<Attachment> attachments = persistenceService.getAllAttachments(ObjectTypeCode.CATCH_BATCH, catchBatch.getIdAsInt());
102             producerForAttachment.addAttachments(attachments, attachmentRows);
103 
104         }
105 
106         return attachmentRows;
107 
108     }
109 
110 }