Added PatchMethodWithMakeFunc func

undefinedlabs · Jan 22, 2020 · 0044123 · 0044123
1 parent 8db8327
commit 0044123
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/patcher.go b/patcher.go
@@ -65,6 +65,10 @@ func PatchMethodByReflect(target reflect.Method, redirection interface{}) (*Patc
 	}
 	return patch, nil
 }
+func PatchMethodWithMakeFunc(target reflect.Method, fn func(args []reflect.Value) (results []reflect.Value)) (*Patch, error) {
+	rValue := reflect.MakeFunc(target.Type, fn)
+	return PatchMethodByReflect(target, rValue)
+}
 
 func (p *Patch) Patch() error {
 	if p == nil {