Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -95,21 +95,27 @@ public void transform(FeedTransformZipTarget zipTarget, MonitorableJob.Status st
LOG.warn("No revenue trips! All trips and related stop times will be removed.");
}

writeToFile(
zipTarget,
"stop_times.txt",
revenueStopTimes.getStopTimeRows().values().stream().flatMap(List::stream).collect(Collectors.toList()),
headersForStopTime,
revenueStopTimes.getDeletedRowCount()
);
// Only update tables and report change if rows are removed.

writeToFile(
zipTarget,
"trips.txt",
revenueTrips.tripRows,
headersForTrips,
revenueTrips.getDeletedRowCount()
);
if (revenueStopTimes.getDeletedRowCount() != 0) {
writeToFile(
zipTarget,
"stop_times.txt",
revenueStopTimes.getStopTimeRows().values().stream().flatMap(List::stream).collect(Collectors.toList()),
headersForStopTime,
revenueStopTimes.getDeletedRowCount()
);
}

if (revenueTrips.getDeletedRowCount() != 0) {
writeToFile(
zipTarget,
"trips.txt",
revenueTrips.tripRows,
headersForTrips,
revenueTrips.getDeletedRowCount()
);
}
} catch (Exception e) {
LOG.error("Unknown error encountered while attempting to remove non revenue trip from zip file.", e);
}
Expand Down Expand Up @@ -231,7 +237,7 @@ private boolean isRevenueStopTime(
private boolean isRevenuePickupDropOff(CsvReader csvReader, Map<String, Integer> fieldIndexes) throws IOException {
String pickup = csvReader.get(fieldIndexes.get(PICKUP_FIELD_NAME));
String dropOff = csvReader.get(fieldIndexes.get(DROP_OFF_FIELD_NAME));
return !"1".equals(pickup) && !"1".equals(dropOff);
return !"1".equals(pickup) || !"1".equals(dropOff);
}

/**
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -107,15 +107,16 @@ static Stream<Arguments> createNonRevenueCases() {
return Stream.of(
Arguments.of(
List.of(
"non-revenue-trip1,1,",
"non-revenue-trip1,1,1",
"non-revenue-trip1,1,1",
"non-revenue-trip1,1,"
"non-revenue-trip1,1,1",
"non-revenue-trip1,1,1"
),
List.of(
"revenue-trip,0,0",
"revenue-trip,1,1",
"revenue-trip,1,1",
"revenue-trip,2,1",
"revenue-trip,1,2",
"revenue-trip,0,0"
),
fieldIndexes
Expand All @@ -131,8 +132,8 @@ static Stream<Arguments> createNonRevenueCases() {
List.of(
"revenue-trip,0,0,2,2",
"revenue-trip,0,0,1,1",
"revenue-trip,0,0,2,2",
"revenue-trip,0,0,2,2"
"revenue-trip,0,0,2,1",
"revenue-trip,0,0,1,2"
),
allFieldIndexes
)
Expand Down
Loading