feat(pkg/oomstore): check group category in API Push

oom-ai · Jan 28, 2022 · 0b0e7bc · 0b0e7bc
1 parent d25d7d9
commit 0b0e7bc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pkg/oomstore/push.go b/pkg/oomstore/push.go
@@ -24,6 +24,9 @@ func (s *OomStore) Push(ctx context.Context, opt types.PushOpt) error {
 
 	entity := features[0].Entity()
 	group := features[0].Group
+	if group.Category != types.CategoryStream {
+		return errdefs.Errorf("Push API is for streaming features only")
+	}
 	if !stringSliceEqual(features.Names(), featureNames) {
 		return errdefs.Errorf("FeatureNames %v does not match with group's features %v", featureNames, features.Names())
 	}