feat: added runtime version in the traffic collector metrics and info

Author: TangoBeeAkto

apps/database-abstractor/src/main/java/com/akto/action/DbAction.java

@@ -372,7 +372,7 @@ public String updateTrafficCollectorMetrics() {
 
         // update heartbeat
         try {
-            TrafficCollectorInfoDao.instance.updateHeartbeat(trafficCollectorMetrics.getId(), trafficCollectorMetrics.getRuntimeId());
+            TrafficCollectorInfoDao.instance.updateHeartbeat(trafficCollectorMetrics.getId(), trafficCollectorMetrics.getRuntimeId(), trafficCollectorMetrics.getVersion());
         } catch (Exception e) {
             loggerMaker.errorAndAddToDb(e, "Error while updating heartbeat: " + e);
         }

libs/dao/src/main/java/com/akto/dao/traffic_collector/TrafficCollectorInfoDao.java

@@ -47,13 +47,14 @@ public void createIndicesIfAbsent() {
     }
 
 
-    public void updateHeartbeat(String id, String runtimeId) {
+    public void updateHeartbeat(String id, String runtimeId, String version) {
         instance.updateOne(
                 Filters.eq("_id", id),
                 Updates.combine(
                         Updates.set(TrafficCollectorInfo.LAST_HEARTBEAT, Context.now()),
                         Updates.setOnInsert(TrafficCollectorInfo.START_TIME, Context.now()),
-                        Updates.setOnInsert(TrafficCollectorInfo.RUNTIME_ID, runtimeId)
+                        Updates.setOnInsert(TrafficCollectorInfo.RUNTIME_ID, runtimeId),
+                        Updates.set(TrafficCollectorInfo.VERSION, version)
                 )
         );
     }

libs/dao/src/main/java/com/akto/dto/traffic_collector/TrafficCollectorInfo.java

@@ -9,14 +9,17 @@ public class TrafficCollectorInfo {
     private int startTime;
     public static final String  LAST_HEARTBEAT = "lastHeartbeat";
     private int lastHeartbeat;
+    public static final String VERSION = "version";
+    private String version;
 
     public TrafficCollectorInfo() {}
 
-    public TrafficCollectorInfo(String id, String runtimeId, int startTime, int lastHeartbeat) {
+    public TrafficCollectorInfo(String id, String runtimeId, int startTime, int lastHeartbeat, String version) {
         this.id = id;
         this.runtimeId = runtimeId;
         this.startTime = startTime;
         this.lastHeartbeat = lastHeartbeat;
+        this.version = version;
     }
 
     public String getId() {
@@ -50,4 +53,12 @@ public String getRuntimeId() {
     public void setRuntimeId(String runtimeId) {
         this.runtimeId = runtimeId;
     }
+
+    public String getVersion() {
+        return version;
+    }
+
+    public void setVersion(String version) {
+        this.version = version;
+    }
 }

libs/dao/src/main/java/com/akto/dto/traffic_collector/TrafficCollectorMetrics.java

@@ -12,14 +12,17 @@ public class TrafficCollectorMetrics {
     private int bucketStartEpoch;
     public static final String BUCKET_END_EPOCH = "bucketEndEpoch";
     private int bucketEndEpoch;
+    public static final String VERSION = "version";
+    private String version;
 
 
-    public TrafficCollectorMetrics(String id, String runtimeId,  Map<String, Integer> requestsCountMapPerMinute, int bucketStartEpoch, int bucketEndEpoch) {
+    public TrafficCollectorMetrics(String id, String runtimeId,  Map<String, Integer> requestsCountMapPerMinute, int bucketStartEpoch, int bucketEndEpoch, String version) {
         this.id = id;
         this.runtimeId = runtimeId;
         this.requestsCountMapPerMinute = requestsCountMapPerMinute;
         this.bucketStartEpoch = bucketStartEpoch;
         this.bucketEndEpoch = bucketEndEpoch;
+        this.version = version;
     }
 
     public TrafficCollectorMetrics() {
@@ -75,4 +78,12 @@ public String getRuntimeId() {
     public void setRuntimeId(String runtimeId) {
         this.runtimeId = runtimeId;
     }
+
+    public String getVersion() {
+        return version;
+    }
+
+    public void setVersion(String version) {
+        this.version = version;
+    }
 }