fix(api): filter on groups on /worker/enabled

ovh · Jan 17, 2020 · 7ec11da · 7ec11da
1 parent 4a2d850
commit 7ec11da
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/engine/api/worker_model_hatchery.go b/engine/api/worker_model_hatchery.go
@@ -93,8 +93,13 @@ func (api *API) getWorkerModelsEnabledHandler() service.Handler {
 		if !ok {
 			return sdk.WithStack(sdk.ErrForbidden)
 		}
-
-		var groupIDs = append(getAPIConsumer(ctx).GetGroupIDs(), group.SharedInfraGroup.ID)
+		consumer := getAPIConsumer(ctx)
+		var groupIDs []int64
+		if len(consumer.GroupIDs) > 0 {
+			groupIDs = consumer.GroupIDs
+		} else {
+			groupIDs = append(getAPIConsumer(ctx).GetGroupIDs(), group.SharedInfraGroup.ID)
+		}
 
 		// for an hatchery the token was generated by a admin for all consumer's groups.
 		models, err := workermodel.LoadAllUsableWithClearPasswordByGroupIDs(api.mustDB(), groupIDs)