#1410 Use StreamQuerySrv for case alert stats

TheHive-Project · Jul 22, 2020 · 6f5b879 · 6f5b879
1 parent 56f9be7
commit 6f5b879
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 6 deletions.
diff --git a/frontend/app/scripts/controllers/case/CaseMainCtrl.js b/frontend/app/scripts/controllers/case/CaseMainCtrl.js
@@ -81,6 +81,7 @@
                 }
             });
 
+            // Stats for case tasks counter
             StreamQuerySrv('v1', [
                 {_name: 'getCase', idOrName: caseId},
                 {_name: 'tasks'},
@@ -105,6 +106,7 @@
                 }
             });
 
+            // Stats for case observables counter
             StreamQuerySrv('v1', [
                 {_name: 'getCase', idOrName: caseId},
                 {_name: 'observables'},
@@ -123,15 +125,34 @@
                 }
             });
 
-            $scope.alerts = StreamStatSrv({
+            // Stats for case observables counter
+            StreamQuerySrv('v1', [
+                {_name: 'getCase', idOrName: caseId},
+                {_name: 'alerts'},
+                {_name: 'count'}
+            ], {
                 scope: $scope,
                 rootId: caseId,
-                query: { 'case': caseId },
-                result: {},
                 objectType: 'alert',
-                field: 'type'
+                query: {
+                    params: {
+                        name: 'alert-stats-' + caseId
+                    }
+                },
+                onUpdate: function(updates) {
+                    $scope.alertCount = updates;
+                }
             });
 
+            // $scope.alerts = StreamStatSrv({
+            //     scope: $scope,
+            //     rootId: caseId,
+            //     query: { 'case': caseId },
+            //     result: {},
+            //     objectType: 'alert',
+            //     field: 'type'
+            // });
+
             $scope.$on('tasks:task-removed', function(event, task) {
                 CaseTabsSrv.removeTab('task-' + task._id);
             });

diff --git a/frontend/app/views/partials/case/case.panelinfo.html b/frontend/app/views/partials/case/case.panelinfo.html
@@ -51,11 +51,11 @@ <h3 class="box-title text-primary">
             </strong>
         </span>
 
-        <span class="ml-xxs text-danger" ng-show="alerts.count > 0"uib-tooltip="Related to">
+        <span class="ml-xxs text-danger" ng-show="alertCount > 0"uib-tooltip="Related to">
             <i class="glyphicon glyphicon-link"></i>
             <strong>
                 <a class="text-danger" href ui-sref="app.case.alerts({caseId:caze._id})">
-                  <ng-pluralize count="alerts.count" when="{'one': '1 alert', 'other': '{} alerts'}"></ng-pluralize>
+                  <ng-pluralize count="alertCount" when="{'one': '1 alert', 'other': '{} alerts'}"></ng-pluralize>
                 </a>
             </strong>
         </span>