View Javadoc
1   package fr.ifremer.tutti.service.csv;
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.tutti.persistence.entities.TuttiEntity;
28  import org.apache.commons.lang3.StringUtils;
29  import org.nuiton.csv.ImportRuntimeException;
30  import org.nuiton.csv.ValueParserFormatter;
31  
32  import java.text.ParseException;
33  import java.util.Map;
34  
35  import static org.nuiton.i18n.I18n.t;
36  
37  /**
38   * @param <E>
39   * @author Tony Chemit - chemit@codelutin.com
40   * @since 1.0
41   */
42  public class ForeignKeyParserFormatter<E extends TuttiEntity> implements ValueParserFormatter<E> {
43  
44      protected final String propertyName;
45  
46      protected final Class<E> entityType;
47  
48      protected final Map<String, E> universe;
49  
50      public ForeignKeyParserFormatter(Class<E> entityType, String propertyName, Map<String, E> universe) {
51          this.entityType = entityType;
52          this.propertyName = propertyName;
53          this.universe = universe;
54      }
55  
56      @Override
57      public E parse(String value) throws ParseException {
58          E result = null;
59          if (StringUtils.isNotBlank(value)) {
60  
61              // get entity from universe
62              result = universe.get(value);
63  
64              if (result == null) {
65  
66                  // can not find entity this is a big problem for us...
67                  throw new ImportRuntimeException(t("tutti.service.csv.parse.entityNotFound", entityType.getSimpleName(), propertyName, value));
68              }
69          }
70          return result;
71      }
72  
73      @Override
74      public String format(E e) {
75          String value = "";
76          if (e != null) {
77              value = e.getId();
78          }
79          return value;
80      }
81  }