Add OnVisitEnd

raskyer · Dec 13, 2017 · 5631c19 · 5631c19
1 parent 5a43fb5
commit 5631c19
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/event-visitor.go b/event-visitor.go
@@ -1,9 +1,12 @@
 package main
 
-import "github.com/leaklessgfy/asm/asm"
+import (
+	"github.com/leaklessgfy/asm/asm"
+)
 
 type EventVisitor struct {
-	OnVisit []func(version, access int, name, signature, superName string, interfaces []string)
+	OnVisit    []func(version, access int, name, signature, superName string, interfaces []string)
+	OnVisitEnd []func()
 }
 
 func (e EventVisitor) Visit(version, access int, name, signature, superName string, interfaces []string) {
@@ -49,4 +52,7 @@ func (e EventVisitor) VisitMethod(access int, name, descriptor, signature string
 }
 
 func (e EventVisitor) VisitEnd() {
+	for _, callback := range e.OnVisitEnd {
+		callback()
+	}
 }
diff --git a/main.go b/main.go
@@ -32,5 +32,10 @@ func main() {
 				fmt.Println(name, signature, superName, interfaces)
 			},
 		},
+		OnVisitEnd: []func(){
+			func() {
+				fmt.Println("End")
+			},
+		},
 	}, 0)
 }