hook_test.go

package hook

import (
	"errors"
	"fmt"
	"testing"
)

type bird struct {
}

func (p bird) Name() string {
	return "my is bird hook"
}
func (p bird) Run(t *Context) {
	fmt.Println("[run] my param name --> ", t.GetParam("name").(string))
}
func (p bird) Clean(t *Context) {
	fmt.Println("[rollback] my param name <-- ", t.GetParam("name").(string))
}

type tiger struct {
}

func (p tiger) Run(t *Context) {
	fmt.Println("[run] my param name --> ", t.GetParam("name").(string))
}

type lion struct {
}

func (p lion) Name() string {
	return "my is lion hook"
}
func (p lion) Run(t *Context) {
	fmt.Println("[run] my param name --> ", t.GetParam("name").(string))
}

var (
	birdkey  = &Hookkey{"bird"}
	tigerkey = &Hookkey{"tiger"}
	lionkey  = &Hookkey{"lion"}
)

// 初始化注册 hook 程序启动时需要注册
func initregister() {
	// bird
	Register(birdkey, bird{})
	Register(birdkey, bird{})
	// tiger
	Register(tigerkey, tiger{})
	// lion
	Register(lionkey, lion{})
}
func TestHook(t *testing.T) {
	// 注册hook
	initregister()
	// 初始化调用链参数
	param := &Context{}
	param.SetParam("name", "my is bird")
	// 根据 hookkey 调用hook
	RunHook(birdkey, param)
	if param.IsAbort() {
		// 如果调用链 abort
		return
	}
	// 调整调用链参数
	param.SetParam("name", "my is tiger")
	// 根据 hookkey 调用hook
	RunHook(tigerkey, param)
	if param.IsAbort() {
		// 如果调用链 abort
		return
	}
	// 调整调用链参数
	param.SetParam("name", "my is lion")
	// 根据 hookkey 调用hook
	RunHook(lionkey, param)
	if param.IsAbort() {
		// 如果调用链 abort
		return
	}
	// 主动abort 触发 rollback
	param.AbortErr(errors.New(" test abort "))
	if param.IsAbort() {
		fmt.Println(param.Err().Error())
	}

	PrintMap("")
}