Add VisitDirectDeps to module context
Change-Id: I271db2120aba1a5c99f21198f1eff6370e6e2389
diff --git a/module_ctx.go b/module_ctx.go
index 87f37b1..4931b6d 100644
--- a/module_ctx.go
+++ b/module_ctx.go
@@ -126,6 +126,8 @@
OtherModuleName(m Module) string
OtherModuleErrorf(m Module, fmt string, args ...interface{})
+ VisitDirectDeps(visit func(Module))
+ VisitDirectDepsIf(pred func(Module) bool, visit func(Module))
VisitDepsDepthFirst(visit func(Module))
VisitDepsDepthFirstIf(pred func(Module) bool, visit func(Module))
@@ -229,6 +231,14 @@
})
}
+func (m *moduleContext) VisitDirectDeps(visit func(Module)) {
+ m.context.visitDirectDeps(m.module, visit)
+}
+
+func (m *moduleContext) VisitDirectDepsIf(pred func(Module) bool, visit func(Module)) {
+ m.context.visitDirectDepsIf(m.module, pred, visit)
+}
+
func (m *moduleContext) VisitDepsDepthFirst(visit func(Module)) {
m.context.visitDepsDepthFirst(m.module, visit)
}