如何打造一个用户友好的自定义 Golang 框架？

自定义 golang 框架可以提高应用程序开发效率和可维护性。本教程逐步指导你创建自己的框架，包括：创建框架基础定义中间件创建路由器添加依赖注入创建 http 处理程序编写集成测试在实战案例中，我们使用框架创建了一个简单的 api，该 api 允许通过用户 id 获取用户名称。

打造一个用户友好的自定义 Golang 框架

在开发 Go 应用程序时，使用自定义框架可以显著提高效率和可维护性。本教程将指导你逐步创建一个用户友好的 Golang 框架，然后通过一个实战案例展示其应用。

1. 创建框架基础

创建一个新的 Go 模块并导入必要的依赖项：

package myframework

import (
    "fmt"
    "net/http"
)

2. 定义中间件

中间件允许你在处理请求和发送响应之前增强请求处理管道。创建一个 Middleware 接口来定义中间件函数：

type Middleware interface {
    Handle(next http.HandlerFunc) http.HandlerFunc
}

3. 创建路由器

路由器负责将请求映射到处理函数。创建一个 Router 类型，并定义一个 HandleFunc 方法来注册处理程序：

type Router struct {
    routes map[string]*Route
}

func (r *Router) HandleFunc(pattern string, handler http.HandlerFunc) {
    r.routes[pattern] = &Route{pattern: pattern, handler: handler}
}

4. 定义依赖注入（DI）

DI 允许你将依赖项注入到框架中。创建一个 Service 接口来定义服务：

type Service interface {
    Get() string
}

然后创建一个 DI 类型来管理依赖项：

type DI struct {
    services map[string]Service
}

5. 创建 HTTP 处理程序

处理程序负责处理 HTTP 请求。创建一个 Handler 接口来定义处理程序函数：

type Handler interface {
    Handle(w http.ResponseWriter, r *http.Request)
}

然后创建具体处理程序并将其注册到路由器：

type MyHandler struct {
    di *DI
}

func (h *MyHandler) Handle(w http.ResponseWriter, r *http.Request) {
    service := h.di.services["myService"]
    fmt.Fprint(w, service.Get())
}

6. 编写集成测试

单元测试对于确保框架的正确性至关重要。编写集成测试来测试处理程序和中间件的集成：

import (
    "net/http/httptest"
    "testing"
)

func TestMyHandler(t *testing.T) {
    handler := &MyHandler{
        di: &DI{
            services: map[string]Service{
                "myService": &MockService{Get: "Hello World!"},
            },
        },
    }
    
    request := httptest.NewRequest(http.MethodGet, "/", nil)
    response := httptest.NewRecorder()
    handler.Handle(response, request)

    if response.Body.String() != "Hello World!" {
        t.Errorf("Expected 'Hello World!', got %s", response.Body.String())
    }
}

实战案例：创建一个简单的 API

让我们使用自定义框架创建一个简单的 API。

1. 创建一个新的 Golang 模块并导入框架：

   import (
    "fmt"
    
    "<a style='color:#f60; text-decoration:underline;' href="https://www.codesou.cn/" target="_blank">git</a>hub.com/username/myframework"
   )

2. 创建一个 UserService 来处理用户：

   type UserService struct {}
   
   func (s *UserService) Get(id int) (string, error) {
    return fmt.Sprintf("User ID: %d", id), nil
   }

3. 注册服务到 DI：

   var di = myframework.NewDI()
   di.Services["userService"] = &UserService{}

4. 创建处理程序：

   type UserHandler struct {
    di *myframework.DI
   }
   
   func (h *UserHandler) Handle(w http.ResponseWriter, r *http.Request) {
    srv := h.di.Services["userService"].(myframework.Service)
    id, _ := strconv.Atoi(r.URL.Query().Get("id"))
    name, _ := srv.(UserService).Get(id)
    fmt.Fprint(w, name)
   }

5. 注册处理程序和路由：

   route := myframework.NewRouter()
   route.HandleFunc("/user", (&UserHandler{di: di}).Handle)

6. 启动 HTTP 服务器：

   log.Fatal(http.ListenAndServe(":8080", route))

   运行此模块，你将获得一个可工作的 API，允许你通过 GET /user?id= 获取用户名称。

   golang免费学习笔记（深入）：立即学习
   在学习笔记中，你将探索 go语言 的核心概念和高级技巧！