自定义实现类似gin的http server

最新推荐文章于 2024-04-02 11:24:45 发布

kankan231

最新推荐文章于 2024-04-02 11:24:45 发布

阅读量488

点赞数

分类专栏： golang 文章标签： gin http golang

本文链接：https://blog.csdn.net/kankan231/article/details/125785240

版权

本文介绍了如何在Golang中实现一个自定义的HTTP服务器，重点在于实现http.Handler接口，通过示例代码展示了如何处理请求，并提供了测试用例。

摘要由CSDN通过智能技术生成

要实现自定义的golang http server，关键要实现http.Hanlder这个接口，这个接口只有一个方法：

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

直接上代码：

package main

import (
	"fmt"
	"net/http"
)
//处理请求的函数类型
type HandleFunc func(http.ResponseWriter, *http.Request)

//自定义的请求处理器，类似gin的Engine
type MyHandler struct{
	Routers map[string]HandleFunc//用于存放路由与处理函数的map
}

//实现http.Handler接口的ServeHTTP方法，有请求到来时就会调用这个方法
func (h *MyHandler) ServeHTTP (w http.ResponseWriter, r *http.Request) {
	path:=r.URL.Path
	if f,ok := h.Routers[path]; ok{//判断路由是否存在
		f(w,r)//调用处理函数
	}else{
		fmt.Fprintf(w, "404 not found")
	}

}

//添加路由与处理函数
func (h *MyHandler) AddRoute(path string,handleFunc HandleFunc)  {
	h.Routers[path] = handleFunc
}

func NewHandle() *MyHandler {
	engine:=&MyHandler{}
	engine.Routers = make(map[string]HandleFunc)
	return engine
}

func main() {
	engine:=NewHandle()