Search in sources :

Example 1 with BeanWrapper

use of org.onebusaway.csv_entities.schema.BeanWrapper in project onebusaway-gtfs-modules by OneBusAway.

the class DeferredValueSupportTest method testResolveAgencyAndId_ExistingAgencyId.

@Test
public void testResolveAgencyAndId_ExistingAgencyId() {
    Stop stop = new Stop();
    stop.setId(new AgencyAndId("a1", "2"));
    BeanWrapper bean = BeanWrapperFactory.wrap(stop);
    AgencyAndId id = _support.resolveAgencyAndId(bean, "id", "1");
    assertEquals(new AgencyAndId("a1", "1"), id);
}
Also used : BeanWrapper(org.onebusaway.csv_entities.schema.BeanWrapper) AgencyAndId(org.onebusaway.gtfs.model.AgencyAndId) Stop(org.onebusaway.gtfs.model.Stop) Test(org.junit.Test)

Example 2 with BeanWrapper

use of org.onebusaway.csv_entities.schema.BeanWrapper in project onebusaway-gtfs-modules by OneBusAway.

the class ServiceDateFieldMappingFactoryTest method test.

@Test
public void test() {
    ServiceDateFieldMappingFactory factory = new ServiceDateFieldMappingFactory();
    DefaultEntitySchemaFactory schemaFactory = new DefaultEntitySchemaFactory();
    String propName = "date";
    FieldMapping mapping = factory.createFieldMapping(schemaFactory, Dummy.class, propName, propName, ServiceDate.class, true);
    CsvEntityContext context = new CsvEntityContextImpl();
    Map<String, Object> csvValues = new HashMap<String, Object>();
    csvValues.put(propName, "20100212");
    Dummy obj = new Dummy();
    BeanWrapper wrapped = BeanWrapperFactory.wrap(obj);
    mapping.translateFromCSVToObject(context, csvValues, wrapped);
    assertEquals(new ServiceDate(2010, 2, 12), obj.getDate());
    csvValues.clear();
    mapping.translateFromObjectToCSV(context, wrapped, csvValues);
    assertEquals("20100212", csvValues.get(propName));
}
Also used : CsvEntityContextImpl(org.onebusaway.csv_entities.CsvEntityContextImpl) BeanWrapper(org.onebusaway.csv_entities.schema.BeanWrapper) ServiceDate(org.onebusaway.gtfs.model.calendar.ServiceDate) HashMap(java.util.HashMap) FieldMapping(org.onebusaway.csv_entities.schema.FieldMapping) CsvEntityContext(org.onebusaway.csv_entities.CsvEntityContext) DefaultEntitySchemaFactory(org.onebusaway.csv_entities.schema.DefaultEntitySchemaFactory) Test(org.junit.Test)

Example 3 with BeanWrapper

use of org.onebusaway.csv_entities.schema.BeanWrapper in project onebusaway-gtfs-modules by OneBusAway.

the class StopTimeFieldMappingFactoryTest method test.

@Test
public void test() {
    StopTimeFieldMappingFactory factory = new StopTimeFieldMappingFactory();
    DefaultEntitySchemaFactory schemaFactory = new DefaultEntitySchemaFactory();
    String propName = "time";
    FieldMapping mapping = factory.createFieldMapping(schemaFactory, Dummy.class, propName, propName, Integer.class, true);
    CsvEntityContext context = new CsvEntityContextImpl();
    Map<String, Object> csvValues = new HashMap<String, Object>();
    csvValues.put(propName, "1234:23:32");
    Dummy obj = new Dummy();
    BeanWrapper wrapped = BeanWrapperFactory.wrap(obj);
    mapping.translateFromCSVToObject(context, csvValues, wrapped);
    assertEquals(new Integer(1234 * 60 * 60 + 23 * 60 + 32), obj.getTime());
    csvValues.clear();
    mapping.translateFromObjectToCSV(context, wrapped, csvValues);
    assertEquals("1234:23:32", csvValues.get(propName));
}
Also used : CsvEntityContextImpl(org.onebusaway.csv_entities.CsvEntityContextImpl) BeanWrapper(org.onebusaway.csv_entities.schema.BeanWrapper) HashMap(java.util.HashMap) FieldMapping(org.onebusaway.csv_entities.schema.FieldMapping) CsvEntityContext(org.onebusaway.csv_entities.CsvEntityContext) DefaultEntitySchemaFactory(org.onebusaway.csv_entities.schema.DefaultEntitySchemaFactory) StopTimeFieldMappingFactory.getSecondsAsString(org.onebusaway.gtfs.serialization.mappings.StopTimeFieldMappingFactory.getSecondsAsString) Test(org.junit.Test)

Example 4 with BeanWrapper

use of org.onebusaway.csv_entities.schema.BeanWrapper in project onebusaway-gtfs-modules by OneBusAway.

the class TransformFactory method setObjectPropertiesFromJsonUsingCsvFields.

private void setObjectPropertiesFromJsonUsingCsvFields(Object object, JSONObject json, String line) throws JSONException, TransformSpecificationMissingArgumentException {
    EntitySchemaFactory entitySchemaFactory = _transformer.getReader().getEntitySchemaFactory();
    EntitySchema schema = entitySchemaFactory.getSchema(object.getClass());
    BeanWrapper wrapped = BeanWrapperFactory.wrap(object);
    Map<String, Object> values = new HashMap<String, Object>();
    for (Iterator<?> it = json.keys(); it.hasNext(); ) {
        String key = (String) it.next();
        Object v = json.get(key);
        if (v instanceof JSONArray) {
            JSONArray array = (JSONArray) v;
            List<Object> asList = new ArrayList<Object>();
            for (int i = 0; i < array.length(); ++i) {
                asList.add(array.get(i));
            }
            v = asList;
        }
        values.put(key, v);
    }
    CsvEntityContext context = new CsvEntityContextImpl();
    for (FieldMapping mapping : schema.getFields()) {
        try {
            mapping.translateFromCSVToObject(context, values, wrapped);
        } catch (MissingRequiredFieldException ex) {
            String verboseMessage = "line=" + line + ", context=" + context + ", json=" + json + ", object=" + object;
            _log.error("missing required field; details:" + verboseMessage);
            throw new TransformSpecificationMissingArgumentException(verboseMessage, ex.getFieldName());
        }
    }
}
Also used : MissingRequiredFieldException(org.onebusaway.csv_entities.exceptions.MissingRequiredFieldException) HashMap(java.util.HashMap) FieldMapping(org.onebusaway.csv_entities.schema.FieldMapping) SingleFieldMapping(org.onebusaway.csv_entities.schema.SingleFieldMapping) JSONArray(org.json.JSONArray) ArrayList(java.util.ArrayList) CsvEntityContextImpl(org.onebusaway.csv_entities.CsvEntityContextImpl) BeanWrapper(org.onebusaway.csv_entities.schema.BeanWrapper) EntitySchemaFactory(org.onebusaway.csv_entities.schema.EntitySchemaFactory) CsvEntityContext(org.onebusaway.csv_entities.CsvEntityContext) JSONObject(org.json.JSONObject) TransformSpecificationMissingArgumentException(org.onebusaway.gtfs_transformer.TransformSpecificationMissingArgumentException) EntitySchema(org.onebusaway.csv_entities.schema.EntitySchema)

Example 5 with BeanWrapper

use of org.onebusaway.csv_entities.schema.BeanWrapper in project onebusaway-gtfs-modules by OneBusAway.

the class SimpleModificationStrategy method run.

@Override
public void run(TransformContext context, GtfsMutableRelationalDao dao, Object entity) {
    BeanWrapper wrapper = BeanWrapperFactory.wrap(entity);
    for (Map.Entry<String, ValueSetter> entry : _propertyUpdates.entrySet()) {
        String propertyName = entry.getKey();
        ValueSetter setter = entry.getValue();
        setter.setValue(wrapper, propertyName);
    }
}
Also used : BeanWrapper(org.onebusaway.csv_entities.schema.BeanWrapper) Map(java.util.Map) ValueSetter(org.onebusaway.gtfs_transformer.deferred.ValueSetter)

Aggregations

BeanWrapper (org.onebusaway.csv_entities.schema.BeanWrapper)9 Test (org.junit.Test)4 HashMap (java.util.HashMap)3 CsvEntityContext (org.onebusaway.csv_entities.CsvEntityContext)3 CsvEntityContextImpl (org.onebusaway.csv_entities.CsvEntityContextImpl)3 FieldMapping (org.onebusaway.csv_entities.schema.FieldMapping)3 AgencyAndId (org.onebusaway.gtfs.model.AgencyAndId)3 ArrayList (java.util.ArrayList)2 Map (java.util.Map)2 DefaultEntitySchemaFactory (org.onebusaway.csv_entities.schema.DefaultEntitySchemaFactory)2 Stop (org.onebusaway.gtfs.model.Stop)2 JSONArray (org.json.JSONArray)1 JSONObject (org.json.JSONObject)1 Pair (org.onebusaway.collections.tuple.Pair)1 MissingRequiredFieldException (org.onebusaway.csv_entities.exceptions.MissingRequiredFieldException)1 EntitySchema (org.onebusaway.csv_entities.schema.EntitySchema)1 EntitySchemaFactory (org.onebusaway.csv_entities.schema.EntitySchemaFactory)1 SingleFieldMapping (org.onebusaway.csv_entities.schema.SingleFieldMapping)1 ServiceDate (org.onebusaway.gtfs.model.calendar.ServiceDate)1 StopTimeFieldMappingFactory.getSecondsAsString (org.onebusaway.gtfs.serialization.mappings.StopTimeFieldMappingFactory.getSecondsAsString)1