fix(#53): entry check should contain itself

williamfzc · Jul 27, 2023 · 7e59635 · 7e59635
1 parent 2b63eb5
commit 7e59635
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/graph/file/api_query_ctx.go b/graph/file/api_query_ctx.go
@@ -73,7 +73,8 @@ func (fg *Graph) TransitiveReferenceIds(f *Vertex) []string {
 
 func (fg *Graph) EntryIds(f *Vertex) []string {
 	ret := make([]string, 0)
-	all := fg.TransitiveReferencedIds(f)
+	// and also itself
+	all := append(fg.TransitiveReferencedIds(f), f.Id())
 	for _, eachId := range all {
 		item := fg.IdCache[eachId]
 		if item.ContainTag(TagEntry) {

diff --git a/graph/function/api_query_ctx.go b/graph/function/api_query_ctx.go
@@ -80,7 +80,8 @@ func (fg *Graph) TransitiveReferenceIds(f *Vertex) []string {
 
 func (fg *Graph) EntryIds(f *Vertex) []string {
 	ret := make([]string, 0)
-	all := fg.TransitiveReferencedIds(f)
+	// and also itself
+	all := append(fg.TransitiveReferencedIds(f), f.Id())
 	for _, eachId := range all {
 		item := fg.IdCache[eachId]
 		if item.ContainTag(TagEntry) {