@Override
    public void handleEntity(Object bean) {

      try {
        NycVehicleLocationRecord record = (NycVehicleLocationRecord) bean;
        AgencyAndId vehicleId = record.getVehicleId();

        NycVehicleLocationRecord prev = _lastRecordsByVehicleId.put(vehicleId, record);

        EntityHandler handler = _entityHandlersByVehicleId.get(vehicleId);

        long time = getTimeForRecord(record);

        if (prev == null || getTimeForRecord(prev) + _maxOffset < time || handler == null) {

          Writer writer = _writersByVehicleId.get(vehicleId);
          if (writer != null) writer.close();

          String timeString = _format.format(new Date(time));
          String fileName = vehicleId.getId() + "-" + timeString + ".txt";
          File outputFile = new File(_outputDirectory, fileName);
          writer = new FileWriter(outputFile);

          handler = _factory.createWriter(NycVehicleLocationRecord.class, writer);
          _entityHandlersByVehicleId.put(vehicleId, handler);
          _writersByVehicleId.put(vehicleId, writer);
        }

        handler.handleEntity(record);
      } catch (IOException ex) {
        throw new IllegalStateException(ex);
      }
    }
コード例 #2
0
 private void applyObjectToAnnotation(Graph graph, AnnotationObject annotation, Object o) {
   if (o instanceof Edge) {
     annotation.edge = graph.getIdForEdge((Edge) o);
   } else if (o instanceof Vertex) {
     annotation.vertex = ((Vertex) o).getLabel();
   } else if (o instanceof String) {
     annotation.message = (String) o;
   } else if (o instanceof IdentityBean) {
     IdentityBean<?> bean = (IdentityBean<?>) o;
     Object id = bean.getId();
     applyObjectToAnnotation(graph, annotation, id);
   } else if (o instanceof AgencyAndId) {
     AgencyAndId id = (AgencyAndId) o;
     annotation.agency = id.getAgencyId();
     annotation.id = id.getId();
   } else if (o instanceof Collection) {
     Collection<?> collection = (Collection<?>) o;
     if (collection.isEmpty()) return;
     Object first = collection.iterator().next();
     applyObjectToAnnotation(graph, annotation, first);
   }
 }