Fix: Checking for correct token on worker management endpoints (beam-…

…cloud#603) Resolve BE-1601
cedana · Oct 8, 2024 · 525b5dd · 525b5dd
1 parent 874ea3f
commit 525b5dd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pkg/gateway/services/worker.go b/pkg/gateway/services/worker.go
@@ -195,5 +195,8 @@ func (gws *GatewayService) DrainWorker(ctx context.Context, in *pb.DrainWorkerRe
 
 func isClusterAdmin(ctx context.Context) (bool, error) {
 	authInfo, _ := auth.AuthInfoFromContext(ctx)
-	return authInfo.Token.TokenType == types.TokenTypeClusterAdmin, errors.New("This action is not permitted")
+	if authInfo.Token.TokenType == types.TokenTypeClusterAdmin {
+		return true, nil
+	}
+	return false, errors.New("This action is not permitted")
 }