浏览代码

alarms: support Get of all alarms

Anthony Romano 9 年之前
父节点
当前提交
dd5b73cfee
共有 1 个文件被更改,包括 8 次插入0 次删除
  1. 8 0
      alarm/alarms.go

+ 8 - 0
alarm/alarms.go

@@ -103,6 +103,14 @@ func (a *AlarmStore) Deactivate(id types.ID, at pb.AlarmType) *pb.AlarmMember {
 func (a *AlarmStore) Get(at pb.AlarmType) (ret []*pb.AlarmMember) {
 	a.mu.Lock()
 	defer a.mu.Unlock()
+	if at == pb.AlarmType_NONE {
+		for _, t := range a.types {
+			for _, m := range t {
+				ret = append(ret, m)
+			}
+		}
+		return ret
+	}
 	for _, m := range a.types[at] {
 		ret = append(ret, m)
 	}