从零实现Go语言Web框架Gee：第一天理解http.Handler-优快云博客

从零实现Go语言Web框架Gee：第一天理解http.Handler

7days-golang 7 days golang programs from scratch (web framework Gee, distributed cache GeeCache, object relational mapping ORM framework GeeORM, rpc framework GeeRPC etc) 7天用Go动手写/从零实现系列项目地址: https://gitcode.com/gh_mirrors/7d/7days-golang

前言

在Go语言生态中，Web框架的选择非常丰富，其中Gin以其高性能和简洁的API设计广受欢迎。本文将带大家从零开始实现一个类似Gin的Web框架Gee，第一天我们将聚焦于理解Go标准库中的net/http和http.Handler接口。

Go标准库的Web服务基础

Go语言内置的net/http库提供了构建Web服务的基础组件。让我们先看一个简单的例子：

package main

import (
	"fmt"
	"log"
	"net/http"
)

func main() {
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, "URL.Path = %q\n", r.URL.Path)
	})
	
	http.HandleFunc("/hello", func(w http.ResponseWriter, r *http.Request) {
		for k, v := range r.Header {
			fmt.Fprintf(w, "Header[%q] = %q\n", k, v)
		}
	})
	
	log.Fatal(http.ListenAndServe(":9999", nil))
}

这个简单的Web服务器做了两件事：

处理根路径/的请求，返回请求的路径
处理/hello路径的请求，返回所有的请求头信息

理解http.Handler接口

Go的net/http库的核心是http.Handler接口：

type Handler interface {
    ServeHTTP(ResponseWriter, *Request)
}

任何实现了ServeHTTP方法的类型都可以作为HTTP处理器。这为我们构建Web框架提供了切入点。

自定义处理器实现

我们可以创建一个自定义类型来实现http.Handler接口：

type Engine struct{}

func (e *Engine) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	switch r.URL.Path {
	case "/":
		fmt.Fprintf(w, "URL.Path = %q\n", r.URL.Path)
	case "/hello":
		for k, v := range r.Header {
			fmt.Fprintf(w, "Header[%q] = %q\n", k, v)
		}
	default:
		fmt.Fprintf(w, "404 NOT FOUND: %s\n", r.URL)
	}
}

func main() {
	engine := new(Engine)
	log.Fatal(http.ListenAndServe(":9999", engine))
}

这种实现方式让我们可以：

统一处理所有HTTP请求
集中管理路由逻辑
方便添加统一的预处理和后处理逻辑

构建Gee框架雏形

现在，我们将上述概念组织成一个简单的Web框架结构：

// gee/gee.go
package gee

import (
	"fmt"
	"net/http"
)

type HandlerFunc func(http.ResponseWriter, *http.Request)

type Engine struct {
	router map[string]HandlerFunc
}

func New() *Engine {
	return &Engine{router: make(map[string]HandlerFunc)}
}

func (e *Engine) addRoute(method, pattern string, handler HandlerFunc) {
	key := method + "-" + pattern
	e.router[key] = handler
}

func (e *Engine) GET(pattern string, handler HandlerFunc) {
	e.addRoute("GET", pattern, handler)
}

func (e *Engine) POST(pattern string, handler HandlerFunc) {
	e.addRoute("POST", pattern, handler)
}

func (e *Engine) Run(addr string) error {
	return http.ListenAndServe(addr, e)
}

func (e *Engine) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	key := r.Method + "-" + r.URL.Path
	if handler, ok := e.router[key]; ok {
		handler(w, r)
	} else {
		fmt.Fprintf(w, "404 NOT FOUND: %s\n", r.URL)
	}
}

使用这个框架的示例：

// main.go
package main

import (
	"fmt"
	"net/http"
	"gee"
)

func main() {
	r := gee.New()
	r.GET("/", func(w http.ResponseWriter, req *http.Request) {
		fmt.Fprintf(w, "URL.Path = %q\n", req.URL.Path)
	})
	
	r.GET("/hello", func(w http.ResponseWriter, req *http.Request) {
		for k, v := range req.Header {
			fmt.Fprintf(w, "Header[%q] = %q\n", k, v)
		}
	})
	
	r.Run(":9999")
}