github/kubevela
1
0
Fork 0
mirror of https://github.com/kubevela/kubevela.git synced 2026-02-14 18:10:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Feat(velaql): read max depth for query iterator from env. (#6039)

Browse Source 
* Feat(velaql): read max depth for query iterator from env

Signed-off-by: Ahmed Qashlan <ahmedelqashlan@gmail.com>

* Feat(velaql): read max depth for query iterator from env.

Ref #6014

Signed-off-by: Ahmed Qashlan <ahmedelqashlan@gmail.com>

---------

Signed-off-by: Ahmed Qashlan <ahmedelqashlan@gmail.com>
This commit is contained in:
Ahmed Qashlan
2023-05-29 05:49:46 +03:00
committed by GitHub
parent 5ccec581e6
commit a849435db2
1 changed files with 19 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
pkg/velaql/providers/query/tree.go
View File

@@ -20,7 +20,9 @@ import (
"context"
"encoding/json"
"fmt"
"os"
"sort"
"strconv"
"time"
appsv1 "k8s.io/api/apps/v1"
@@ -48,12 +50,15 @@ import (
helmrepoapi "github.com/fluxcd/source-controller/api/v1beta2"
)
const (
// DefaultMaxDepth is the default max depth for query iterator
// check maxDepth function to get the customized val for max depth
DefaultMaxDepth = 5
)
// relationshipKey is the configmap key of relationShip rule
var relationshipKey = "rules"
// set the iterator max depth is 5
var maxDepth = 5
// RuleList the rule list
type RuleList []ChildrenResourcesRule
@@ -937,7 +942,7 @@ func listItemByRule(clusterCTX context.Context, k8sClient client.Client, resourc
}
func iterateListSubResources(ctx context.Context, cluster string, k8sClient client.Client, parentResource types.ResourceTreeNode, depth int, filter func(node types.ResourceTreeNode) bool) ([]*types.ResourceTreeNode, error) {
if depth > maxDepth {
if depth > maxDepth() {
klog.Warningf("listing application resource tree has reached the max-depth %d parentObject is %v", depth, parentResource)
return nil, nil
}
@@ -1062,3 +1067,13 @@ func translateTimestampSince(timestamp v1.Time) string {
return duration.HumanDuration(time.Since(timestamp.Time))
}
// check if max depth is provided or return the default max depth
func maxDepth() int {
maxDepth, err := strconv.Atoi(os.Getenv("KUBEVELA_QUERYTREE_MAXDEPTH"))
if err != nil || maxDepth <= 0 {
return DefaultMaxDepth
}
return maxDepth
}