fix(#53): direct impact did not contain itself

graph/file/api_query.go

@@ -10,7 +10,6 @@ import (
 func (fg *Graph) GetById(id string) *Vertex {
 	v, err := fg.G.Vertex(id)
 	if err != nil {
-
 		logrus.Warnf("no vertex: %v", id)
 		return nil
 	}

graph/file/api_query_ctx_test.go

@@ -34,4 +34,10 @@ func TestQuery(t *testing.T) {
 		assert.NotEmpty(t, shouldNotEmpty)
 		log.Infof("outv: %d", len(shouldNotEmpty))
 	})
+
+	t.Run("Entries", func(t *testing.T) {
+		entries := fg.EntryIds(fg.GetById("graph/function/api_query_test.go"))
+		// test case usually is an entry point
+		assert.Len(t, entries, 1)
+	})
 }

graph/function/api_stat.go

@@ -70,6 +70,9 @@ func (fg *Graph) GlobalStat(points []*Vertex) *object.StatGlobal {
 			eachId := sg.UnitMapping[eachReference]
 			directImpactMap[eachId] = struct{}{}
 		}
+		// and itself
+		itselfId := sg.UnitMapping[each.Self.Id()]
+		directImpactMap[itselfId] = struct{}{}
 	}
 	directImpactList := make([]int, 0, len(directImpactMap))
 	for each := range directImpactMap {