golang的net/http三种搭建内部服务实现方式

以你之姓@ 2022-04-15 06:00 176阅读 0赞

golang搭建一个简易的服务器相当的简单，下面介绍三种从内部底层重写的方式  
第一种使用默认的DefaultServeMux和默认的Server的结构  
第二种会使用自定义的handler  
第三种会使用自定义的Server的结构  
从执行效率来说，第三种是执行效率最快的，毕竟是没经过封装的

①通过http.HandleFunc，使用默认的DefaultServeMux

package main
    import (
    	"io"
    	"log"
    	"net/http"
    )
    func main() { 
    	//设置路由规则
    	http.HandleFunc("/", Tmp)
    
    	//使用默认的DefaultServeMux
    	err := http.ListenAndServe(":8080", nil)
    	if err != nil { 
    		log.Fatal(err)
    	}
    }
    
    func Tmp(w http.ResponseWriter, r *http.Request) { 
    	io.WriteString(w, "version 1")
    }

这种方式可以说是最常用的，不用考虑太多，直接调用方法就可以了

②通过实现一个Handler注册到Mux当中，是用Mux来控制路由的访问

package main
    import (
    	"io"
    	"log"
    	"net/http"
    )
    
    type myHandler struct{ }
    
    func (*myHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) { 
    	io.WriteString(w, "URL:"+r.URL.String())
    }
    
    func Tmp(w http.ResponseWriter, r *http.Request) { 
    	io.WriteString(w, "version 2")
    }
    
    func main(){ 
    	mux := http.NewServeMux()
    
    	//以这种形式来注册路由和注册handler
    	mux.Handle("/",&myHandler{ })
    	
    	mux.HandleFunc("/tmp", Tmp)
    	
    	//http.ListenAndServe使用默认的server结构
    	err = http.ListenAndServe(":8080", mux)
    	if err != nil { 
    		log.Fatal(err)
    	}
    }

这种方式需要自己来定义一个handler，而这个自定义的handler需要重写一下`ServeHTTP(ResponseWriter, *Request)`，才能实现

③通过实现Server的结构

package main
    import (
    	"io"
    	"log"
    	"net/http"
    	"time"
    )
    
    //定义一个map来实现路由转发
    var mux map[string]func(http.ResponseWriter, *http.Request) 
    
    func main(){ 
    	server := http.Server{ 
    		Addr: ":8080",
    		Handler: &myHandler{ },
    		ReadTimeout: 5*time.Second,
    	}
    	
    	mux = make(map[string]func(http.ResponseWriter, *http.Request))
    	mux["/tmp"] = Tmp
    
    	err := server.ListenAndServe()
    	if err != nil { 
    		log.Fatal(err)
    	}	
    }
    
    type myHandler struct{ }
    
    func (*myHandler) ServeHTTP(w http.ResponseWriter, r *http.Request){ 
    	//实现路由的转发
    	if h, ok := mux[r.URL.String()];ok{ 
    	//用这个handler实现路由转发，相应的路由调用相应func
    		h(w, r)
    		return
    	}
    	io.WriteString(w, "URL:"+r.URL.String())
    }
    
    func Tmp(w http.ResponseWriter, r *http.Request) { 
    	io.WriteString(w, "version 3")
    }

用自定义的Server来监听和实现http服务

上面三种方式其实就是对封装的解开，用自定义的方式来重写底层结构，可以更加了解http包