当前位置：首页 > article >正文

go的hooks如何写

article 2025/3/25 19:08:50

在 Go 语言中，实现 Hooks 的方式多样，具体取决于应用场景。以下是几种常见实现方法及示例：

一、函数式 Hooks（基础实现）

通过函数类型作为参数传递，实现灵活的钩子机制：

// 定义钩子函数类型
type HookFunc func()

// 业务函数接受钩子参数
func DoSomething(hook HookFunc) {
    // 执行前置操作
    fmt.Println("Before operation")
    hook()    // 调用钩子
    // 执行后置操作
    fmt.Println("After operation")
}

// 自定义钩子逻辑
func MyHook() {
    fmt.Println("Custom hook executed")
}

func main() {
    DoSomething(MyHook) // 传递钩子函数
}

特点：简单轻量，适合简单场景。

二、接口式 Hooks（面向对象扩展）

通过接口定义钩子方法，支持多态扩展：

// 定义钩子接口
type Hook interface {
    Run()
}

// 实现具体钩子
type MyHook struct{}

func (h MyHook) Run() {
    fmt.Println("Interface-based hook executed")
}

// 业务函数接受接口类型
func DoSomething(hook Hook) {
    fmt.Println("Before operation")
    hook.Run()
    fmt.Println("After operation")
}

func main() {
    myHook := MyHook{}
    DoSomething(myHook)
}

特点：支持接口组合，便于扩展。

三、事件监听 Hooks（gohook 库）

通过第三方库 gohook 实现系统级事件监听（如键盘、鼠标）：

import "github.com/robotn/gohook"

func main() {
    // 注册键盘事件钩子
    hook.Register(gohook.KeyDown, []string{"q"}, func(e gohook.Event) {
        fmt.Println("Ctrl+Shift+Q pressed, stopping hook")
        hook.End()
    })
    
    // 启动事件循环
    s := hook.Start()
    <-hook.Process(s)
}

依赖安装：

go get github.com/robotn/gohook
CGO_ENABLED=1 go build  # 需启用CGO

特点：适用于系统级事件捕获。

四、Git Hooks 集成（版本控制自动化）

将 Go 脚本集成到 Git 钩子中，实现自动化流程（如代码格式化）：

创建钩子脚本：

#!/bin/sh
# .git/hooks/pre-commit
./path/to/go-format-script

编译 Go 脚本：

go build -o go-format-script format.go

赋予执行权限：

chmod +x .git/hooks/pre-commit

示例 Go 脚本：

// format.go
package main

import (
    "fmt"
    "os/exec"
)

func main() {
    fmt.Println("Running gofmt...")
    exec.Command("gofmt", "-w", ".").Run()
}

特点：适合代码提交前的自动化检查。

五、HTTP 请求 Hooks（上下文扩展）

通过 context.Context 实现请求生命周期的钩子：

type key int

const (
    hookKey key = iota
)

// 自定义钩子类型
type RequestHook struct {
    Log func()
}

// 中间件注入钩子
func WithHook(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        ctx := context.WithValue(r.Context(), hookKey, &RequestHook{Log: func() { fmt.Println("Request started") }})
        next.ServeHTTP(w, r.WithContext(ctx))
    })
}

// 在处理函数中调用钩子
func MyHandler(w http.ResponseWriter, r *http.Request) {
    if hook, ok := r.Context().Value(hookKey).(*RequestHook); ok {
        hook.Log()
    }
    // 处理请求...
}

特点：适用于 HTTP 框架的请求级扩展。

总结

场景	推荐方案	适用性
简单函数扩展	函数式 Hooks	轻量级场景
面向对象扩展	接口式 Hooks	需多态行为的场景
系统事件监听	gohook 库	键盘/鼠标事件捕获
Git 自动化	Git Hooks 集成	代码提交前处理
HTTP 请求生命周期管理	Context Hooks	Web 框架中间件