/
AgencyIdFieldMappingFactory.java
78 lines (67 loc) · 3 KB
/
AgencyIdFieldMappingFactory.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
/**
* Copyright (C) 2011 Brian Ferris <bdferris@onebusaway.org>
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.onebusaway.gtfs.serialization.mappings;
import org.onebusaway.csv_entities.CsvEntityContext;
import org.onebusaway.csv_entities.exceptions.MissingRequiredFieldException;
import org.onebusaway.csv_entities.schema.BeanWrapper;
import org.onebusaway.csv_entities.schema.DefaultFieldMapping;
import org.onebusaway.csv_entities.schema.EntitySchemaFactory;
import org.onebusaway.csv_entities.schema.FieldMapping;
import org.onebusaway.csv_entities.schema.FieldMappingFactory;
import org.onebusaway.gtfs.model.AgencyAndId;
import java.util.Collection;
import java.util.Map;
public class AgencyIdFieldMappingFactory implements FieldMappingFactory {
public FieldMapping createFieldMapping(EntitySchemaFactory schemaFactory,
Class<?> entityType, String csvFieldName, String objFieldName,
Class<?> objFieldType, boolean required) {
return new FieldMappingImpl(entityType, csvFieldName, objFieldName,
String.class, required);
}
private class FieldMappingImpl extends DefaultFieldMapping {
public FieldMappingImpl(Class<?> entityType, String csvFieldName,
String objFieldName, Class<?> objFieldType, boolean required) {
super(entityType, csvFieldName, objFieldName, objFieldType, required);
}
@Override
public void getCSVFieldNames(Collection<String> names) {
names.add(_csvFieldName + "_agencyId");
names.add(_csvFieldName + "_id");
}
@Override
public void translateFromObjectToCSV(CsvEntityContext context,
BeanWrapper object, Map<String, Object> csvValues) {
AgencyAndId id = (AgencyAndId) object.getPropertyValue(_objFieldName);
csvValues.put(_csvFieldName + "_agencyId", id.getAgencyId());
csvValues.put(_csvFieldName + "_id", id.getId());
}
@Override
public void translateFromCSVToObject(CsvEntityContext context,
Map<String, Object> csvValues, BeanWrapper object) {
String agencyId = (String) csvValues.get(_csvFieldName + "_agencyId");
String id = (String) csvValues.get(_csvFieldName + "_id");
boolean missing = (agencyId == null || agencyId.isEmpty())
|| (id == null || id.isEmpty());
if (missing) {
if (_required)
throw new MissingRequiredFieldException(_entityType, _csvFieldName);
return;
}
AgencyAndId agencyAndId = new AgencyAndId(agencyId, id);
object.setPropertyValue(_objFieldName, agencyAndId);
}
}
}