自定义 golang 框架可以提高应用程序开发效率和可维护性。本教程逐步指导你创建自己的框架，包括：创建框架基础定义中间件创建路由器添加依赖注入创建 http 处理程序编写集成测试在实战案例中，我们使用框架创建了一个简单的 api，该 api 允许通过用户 id 获取用户名称。
打造一个用户友好的自定义 Golang 框架
在开发 Go 应用程序时，使用自定义框架可以显著提高效率和可维护性。本教程将指导你逐步创建一个用户友好的 Golang 框架，然后通过一个实战案例展示其应用。
1. 创建框架基础
立即学习“go语言免费学习笔记（深入）”；
创建一个新的 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 处理程序
SiteDynamic企业网站管理系统采用较为成熟的ASP+ACCESS编写，是迄今为止国内较先进的ASP语言企业网站管理系统。系统为企业级网站提供一个框架，能满足企业的基本应用，同时系统开放全部源码，用户可以根据自己的需求扩展出自己需求的模块，如：单页面、新闻、产品展示、下载、友情链接、电子商务、广告、会员、在线支付、人才招聘等。整套系统的设计构造，完全考虑大中小企业类网站的功能要求，网站的后台
处理程序负责处理 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。
-
创建一个新的 Golang 模块并导入框架：
import ( "fmt" "github.com/username/myframework" )
-
创建一个
UserService来处理用户：type UserService struct {} func (s *UserService) Get(id int) (string, error) { return fmt.Sprintf("User ID: %d", id), nil } -
注册服务到
DI：var di = myframework.NewDI() di.Services["userService"] = &UserService{} -
创建处理程序：
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) } -
注册处理程序和路由：
route := myframework.NewRouter() route.HandleFunc("/user", (&UserHandler{di: di}).Handle) -
启动 HTTP 服务器：
log.Fatal(http.ListenAndServe(":8080", route))
运行此模块，你将获得一个可工作的 API，允许你通过 GET /user?id=
