基于Gin封装Web框架 - 3. 初始化 rum 框架

源码参考: https://github.com/go-jarvis/gin-rum

创建 /rum 目录， 将所有 rum 框架相关的东西都放在这里。

rum

创建 rum 对象, 匿名嵌套 gin.Engine， 这样 Rum 对象就可以直接调用 gin.Engine 的所有方法了。

type Rum struct {
    *gin.Engine
}

创建 Rum 对象提供了 2种方式。 Default 和 自定义。

// Default 创建一个默认的 Engine
func Default() *Rum {
    return &Rum{
        Engine: gin.Default(),
    }
}

不管哪种方式， 在注册的时候， 在初始化 Rum 的时候， 都通过 Engine: gin.Default() 指定字段对象。 因此 匿名嵌套字段 / 命名字段 使用同样的方法方式初始化。这个之前困扰了我一段时间。

解耦 rum 与控制器

之前在控制器中， 是将 *gin.Engine 作为控制器的嵌套字段的， 显然耦合太紧，不合适。 因此要想办法解决。

/classes/index.go ， 在 Index 中， 删除了 *gin.Engine 的嵌套， 取而代之的是将 rum.Rum 在 Build() 方法中从外部传入， 从而达到解耦的目的。

// Index
// 删除 e *gin.Engine ， 删除强耦合关系
type Index struct {
    // e *gin.Engine
}

// Build 控制器的构造器， 创建路由信息
// 1. 通过传参 解耦控制器和 gin server 的关系
// 2. 通过实现 ClassController 接口关联与 rum
func (index *Index) Build(rum *rum.Rum) {
    rum.Handle("GET", "/", handlerIndex)
}

为了管理很多不同的控制器， rum 实现了 class_controller.go ClassController 接口

type ClassController interface {
    Build(rum *Rum)
}

只要满足该接口的对象，都能通过 Mount 方法实现路由注册。

// Mount 挂载控制器
// 1. 关联控制器与 rum
// 2. 返回 *Rum 是为了方便链式调用
func (rum *Rum) Mount(classes ...ClassController) *Rum {
    for _, class := range classes {

        // 将 rum 传入到控制器中
        class.Build(rum)
    }

    return rum
}

main.go 修改

现在就可以使用 rum 代替 gin 了。

func main() {

    // 1. 使用 rum 代替 gin
    g := rum.Default()

    // 2. 注册路由
    g.Mount(
        classes.NewIndex(),
    )

    // 3. 启动 rum server
    g.Launch()
}