Added Endpoint to get BSM message Counts

CDOT-CV · Apr 29, 2024 · 7fd4eeb · 7fd4eeb
1 parent b2679d0
commit 7fd4eeb
Show file tree

Hide file tree

Showing 3 changed files with 59 additions and 1 deletion.
diff --git a/...sualizer-api/src/main/java/us/dot/its/jpo/ode/api/accessors/bsm/OdeBsmJsonRepository.java b/...sualizer-api/src/main/java/us/dot/its/jpo/ode/api/accessors/bsm/OdeBsmJsonRepository.java
@@ -6,5 +6,6 @@
 import us.dot.its.jpo.ode.model.OdeBsmData;
 
 public interface OdeBsmJsonRepository extends DataLoader<OdeBsmData>{
-    List<OdeBsmData> findOdeBsmDataGeo(String originIp, String vehicleId, Long startTime, Long endTime, Double longitude, Double latitude, Double distance);  
+    List<OdeBsmData> findOdeBsmDataGeo(String originIp, String vehicleId, Long startTime, Long endTime, Double longitude, Double latitude, Double distance);
+    long countOdeBsmDataGeo(String originIp, String vehicleId, Long startTime, Long endTime, Double longitude, Double latitude, Double distance); 
 }
diff --git a/...izer-api/src/main/java/us/dot/its/jpo/ode/api/accessors/bsm/OdeBsmJsonRepositoryImpl.java b/...izer-api/src/main/java/us/dot/its/jpo/ode/api/accessors/bsm/OdeBsmJsonRepositoryImpl.java
@@ -93,6 +93,41 @@ public List<OdeBsmData> findOdeBsmDataGeo(String originIp, String vehicleId, Lon
         return convertedList;
     }
 
+    public long countOdeBsmDataGeo(String originIp, String vehicleId, Long startTime, Long endTime, Double longitude, Double latitude, Double distance){
+        Query query = new Query();
+
+        if(originIp != null){
+            query.addCriteria(Criteria.where("metadata.originIp").is(originIp));
+        }
+
+        if(vehicleId != null){
+            query.addCriteria(Criteria.where("payload.data.coreData.id").is(vehicleId));
+        }
+
+        String startTimeString = Instant.ofEpochMilli(0).toString();
+        String endTimeString = Instant.now().toString();
+
+        if(startTime != null){
+            startTimeString = Instant.ofEpochMilli(startTime).toString(); 
+        }
+        if(endTime != null){
+            endTimeString = Instant.ofEpochMilli(endTime).toString();
+        }
+	    query.limit(props.getMaximumResponseSize());
+        query.addCriteria(Criteria.where("metadata.odeReceivedAt").gte(startTimeString).lte(endTimeString));
+        query.fields().exclude("recordGeneratedAt");
+
+        if (longitude!=null && latitude!=null && distance!=null){
+            Double[] latitudes = calculateLatitudes(latitude, distance);
+            Double[] longitudes = calculateLongitudes(longitude, latitude, distance);
+
+            query.addCriteria(Criteria.where("payload.data.coreData.position.latitude").gte(Math.min(latitudes[0], latitudes[1])).lte(Math.max(latitudes[0], latitudes[1])));
+            query.addCriteria(Criteria.where("payload.data.coreData.position.longitude").gte(Math.min(longitudes[0], longitudes[1])).lte(Math.max(longitudes[0], longitudes[1])));
+        }
+
+        return mongoTemplate.count(query, Map.class, collectionName);
+    }
+
     @Override
     public void add(OdeBsmData item) {
         mongoTemplate.save(item, collectionName);

diff --git a/...onflictvisualizer-api/src/main/java/us/dot/its/jpo/ode/api/controllers/BsmController.java b/...onflictvisualizer-api/src/main/java/us/dot/its/jpo/ode/api/controllers/BsmController.java
@@ -58,4 +58,26 @@ public ResponseEntity<List<OdeBsmData>> findBSMs(
             return ResponseEntity.ok(geoData);
         }
     }
+
+    @CrossOrigin(origins = "http://localhost:3000")
+    @RequestMapping(value = "/bsm/count", method = RequestMethod.GET, produces = "application/json")
+    @PreAuthorize("hasRole('USER') || hasRole('ADMIN')")
+    public ResponseEntity<Long> countBSMs(
+            @RequestParam(name = "origin_ip", required = false) String originIp,
+            @RequestParam(name = "vehicle_id", required = false) String vehicleId,
+            @RequestParam(name = "start_time_utc_millis", required = false) Long startTime,
+            @RequestParam(name = "end_time_utc_millis", required = false) Long endTime,
+            @RequestParam(name = "latitude", required = false) Double latitude,
+            @RequestParam(name = "longitude", required = false) Double longitude,
+            @RequestParam(name = "distance", required = false) Double distanceInMeters,
+            @RequestParam(name = "test", required = false, defaultValue = "false") boolean testData) {
+
+        if (testData) {
+            return ResponseEntity.ok(10L);
+        } else {
+            long counts =  odeBsmJsonRepo.countOdeBsmDataGeo(originIp, vehicleId, startTime, endTime, longitude, latitude, distanceInMeters);
+            logger.info("Found " + counts + " BSMs");
+            return ResponseEntity.ok(counts);
+        }
+    }
 }