Enable aggregation for kubearmor summary data - process/file

src/observability/kubearmor.go

@@ -231,6 +231,41 @@ func deDuplicateServerInOutConn(connList []types.SysObsNwData) []types.SysObsNwD
 	return result
 }
 
+func aggregateProcFileData(data []types.SysObsProcFileData) []types.SysObsProcFileData {
+	if len(data) <= 0 {
+		return nil
+	}
+	res := []types.SysObsProcFileData{}
+
+	for _, locData := range data {
+		locKey := types.SysObsProcFileMapKey{
+			Source:      locData.Source,
+			Destination: locData.Destination[:strings.LastIndex(locData.Destination, "/")+1],
+			Status:      locData.Status,
+		}
+
+		v := ProcFileMap[locKey]
+
+		ProcFileMap[locKey] = types.SysObsProcFileMapValue{
+			Count:       v.Count + locData.Count,
+			UpdatedTime: locData.UpdatedTime,
+		}
+	}
+
+	for k, v := range ProcFileMap {
+		res = append(res, types.SysObsProcFileData{
+			Source:      k.Source,
+			Destination: k.Destination,
+			Status:      k.Status,
+			Count:       v.Count,
+			UpdatedTime: v.UpdatedTime,
+		})
+		delete(ProcFileMap, k)
+	}
+
+	return res
+}
+
 func GetKubearmorSummaryData(req *opb.Request) ([]types.SysObsProcFileData, []types.SysObsProcFileData, []types.SysObsNwData, types.ObsPodDetail) {
 	var err error
 	var processData, fileData []types.SysObsProcFileData
@@ -288,5 +323,10 @@ func GetKubearmorSummaryData(req *opb.Request) ([]types.SysObsProcFileData, []ty
 		nwData = deDuplicateServerInOutConn(nwData)
 	}
 
+	if req.Aggregate {
+		processData = aggregateProcFileData(processData)
+		fileData = aggregateProcFileData(fileData)
+	}
+
 	return processData, fileData, nwData, podInfo
 }

src/observability/observability.go

@@ -33,6 +33,7 @@ var (
 	ObsCronJob *cron.Cron
 	//Kubearmor log map
 	KubeArmorLogMap map[types.KubeArmorLog]int
+	ProcFileMap     map[types.SysObsProcFileMapKey]types.SysObsProcFileMapValue
 )
 
 // =================== //
@@ -51,6 +52,7 @@ func initMutex() {
 
 func initMap() {
 	KubeArmorLogMap = make(map[types.KubeArmorLog]int)
+	ProcFileMap = make(map[types.SysObsProcFileMapKey]types.SysObsProcFileMapValue)
 }
 
 func InitObservability() {