Go语言实现一个HttpServer框架(二)Server的抽象

Tertia ·

更新时间:2024-11-13

· 1358 次阅读

上一篇文章对http库的基本使用做了说明，下面来实现一个简单地httpServer

package main  
import (  
    "fmt"  
    "log"  
    "net/http"  
)  
func home(w http.ResponseWriter, r *http.Request) {  
    fmt.Fprintf(w, "这是主页")  
}  
func user(w http.ResponseWriter, r *http.Request) {  
    fmt.Fprintf(w, "这是用户页面")  
}  
func main() {  
    http.HandleFunc("/", home)  
    http.HandleFunc("/user", user)  
    log.Fatal(http.ListenAndServe(":8080", nil))  
}

此时启动服务并访问8080端口，可以看到下面的页面

但是如果我想同时启动多个server监听不同的端口该怎么办？我们可以从逻辑上抽象出一个Server,可以表示对某个端口进行监听的实体，这样必要的时候，我们就可以启动多个Server来监听不同的端口

首先我们来写一个接口

type Server interface{
    //Route 设定一个路由，命中路由会执行handlerFunc的代码
    Route(pattern string handlerFunc http.HandlerFunc)
    //Start 用于启动服务
    Start(address string) error
}

实现Server接口

func (s *sdkHttpServer) Route(pattern string,handlerFunc http.HandlerFunc){
    http.HandleFunc(pattern,handlerFunc)
}
func (s *sdkHttpServer) Start(address string) error{
    http.ListenAndServer(address,nil)
}
func NewSdkHttpServer(name string) Server {
    return &sdkHttpServer{
        Name:name,
    }
}

现在我们已经可以同时监听多个端口了

func main(){
    server := web.NewSdkHttpServer("my-test-server")
    server.Route("/",home)
    server.Route("/user",user)
    server.Start("8080")
}

虽然我们已经对Server进行了抽象，但是在请求处理的过程中仍然存在问题，就比如我们要实现一个用户注册的接口