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.AccidentalBatch;
29  import fr.ifremer.tutti.persistence.entities.data.Attachment;
30  import fr.ifremer.tutti.service.PersistenceService;
31  import fr.ifremer.tutti.service.genericformat.GenericFormatExportContext;
32  import fr.ifremer.tutti.service.genericformat.GenericFormatExportOperationContext;
33  import fr.ifremer.tutti.service.genericformat.csv.AccidentalCatchRow;
34  import fr.ifremer.tutti.service.genericformat.csv.AttachmentRow;
35  import fr.ifremer.tutti.service.genericformat.producer.CsvProducerForAttachment;
36  
37  import java.util.ArrayList;
38  import java.util.List;
39  
40  /**
41   * Created on 3/28/15.
42   *
43   * @author Tony Chemit - chemit@codelutin.com
44   * @since 3.14.3
45   */
46  public class CreateAccidentalCatchRowsAction extends ExportFishingOperationActionSupport {
47  
48      private final PersistenceService persistenceService;
49  
50      public CreateAccidentalCatchRowsAction(PersistenceService persistenceService) {
51          this.persistenceService = persistenceService;
52      }
53  
54      @Override
55      public boolean doExecute(GenericFormatExportContext exportContext, GenericFormatExportOperationContext operationContext) {
56          return exportContext.isExportAccidentalCatch() && operationContext.isWithCatchBatch();
57      }
58  
59      @Override
60      public void execute(GenericFormatExportContext exportContext, GenericFormatExportOperationContext operationContext) {
61  
62          List<AccidentalCatchRow> accidentalCatchRows = exportContext.getProducerForAccidentalCatch().getDataToExport(operationContext);
63          operationContext.setAccidentalCatchRows(accidentalCatchRows);
64          exportContext.getProducerForSpecies().prepareAccidentalRows(accidentalCatchRows);
65  
66          if (exportContext.isExportAttachments()) {
67  
68              List<AttachmentRow> attachmentRows = getAttachmentRows(exportContext, operationContext.getAccidentalBatches());
69              operationContext.addAttachmentRows(attachmentRows);
70  
71          }
72  
73      }
74  
75      protected List<AttachmentRow> getAttachmentRows(GenericFormatExportContext exportContext, List<AccidentalBatch> accidentalBatches) {
76  
77          CsvProducerForAttachment producerForAttachment = exportContext.getProducerForAttachment();
78  
79          List<AttachmentRow> attachmentRows = new ArrayList<>();
80  
81          for (AccidentalBatch accidentalBatch : accidentalBatches) {
82              List<Attachment> attachments = persistenceService.getAllAttachments(ObjectTypeCode.SAMPLE, accidentalBatch.getIdAsInt());
83              producerForAttachment.addAttachments(attachments, attachmentRows);
84          }
85  
86          return attachmentRows;
87  
88      }
89  
90  }