golang框架源码中的扩展与二次开发

package ginextension

import (
    "github.com/gin-gonic/gin"
)

type MyMiddleware interface {
    Handle(c *gin.Context)
}

type MyMiddlewareImpl struct {
}

func (m *MyMiddlewareImpl) Handle(c *gin.Context) {
    // 处理自定义标头
    // ...
}

// gin-extension/register.go

package ginextension

import (
    "github.com/gin-gonic/gin"
)

func Register(app *gin.Engine) {
    c := GetMiddlewareConfig()
    app.Use(MyMiddlewareImpl.Handle)
}

// gin-extension/render.go

package ginextension

import (
    "github.com/gin-gonic/gin"
)

type MyRenderer struct{}

func (r *MyRenderer) Render(w io.Writer, data interface{}) error {
    // 输出自定义 JSON 响应
    // ...
}

func RegisterRouter(router *gin.RouterGroup) {
    r := &MyRenderer{}
    router.GET("mypage", func(c *gin.Context) { c.JSON(200, "Hello, World!") })
    // 注册自定义渲染器
    router.Use(gin.RendererWithOptions(r, gin.RendererOptions{IndentJSON: true}))
}

go mod tidy
go build ./...
go run cmd/main.go