How does ServeHTTP work?

前端 未结 5 824
温柔的废话
温柔的废话 2020-12-14 03:55

I am studying web development in Golang (Beginner) I came across some code I played around with and I\'m not too sure why it works, I looked through the library source code

5条回答
  •  既然无缘
    2020-12-14 04:54

    How does this exactly work? How is that ServeHTTP function being accessed?

    To answer this question we need to look how http.ListenAndServe works:

    func ListenAndServe(addr string, handler Handler) error {
        server := &Server{Addr: addr, Handler: handler}
        return server.ListenAndServe()
    }
    

    Here we create a Server with given address and handler and call the ListenAndServer method so let's take a look there:

    func (srv *Server) ListenAndServe() error {
        addr := srv.Addr
        if addr == "" {
            addr = ":http"
        }
        ln, err := net.Listen("tcp", addr)
        if err != nil {
            return err
        }
        return srv.Serve(tcpKeepAliveListener{ln.(*net.TCPListener)})
    }
    

    This method just starts listening given address and calls the Server method with our freshly created listener so let's follow the trail there:

    func (srv *Server) Serve(l net.Listener) error {
        defer l.Close()
    
        ...
    
        for {
            rw, e := l.Accept()
    
            ...
    
            c := srv.newConn(rw)
            c.setState(c.rwc, StateNew) // before Serve can return
            go c.serve(ctx)
        }
    }
    

    From the Serve method we can see that this is the point where we accept the new connection and start handling it in it's own goroutine.

    // Serve a new connection.
    func (c *conn) serve(ctx context.Context) {
        ...
        for {
            w, err := c.readRequest(ctx)
            ...
            serverHandler{c.server}.ServeHTTP(w, w.req)
            ...
        }
    }
    

    Here we finally call the ServeHTTP method but as we can see this is not yet our implementation of that function but something from the standard library so let's take a look what that serverHandler struct contains:

    // serverHandler delegates to either the server's Handler or
    // DefaultServeMux and also handles "OPTIONS *" requests.
    type serverHandler struct {
        srv *Server
    }
    
    func (sh serverHandler) ServeHTTP(rw ResponseWriter, req *Request) {
        handler := sh.srv.Handler
        if handler == nil {
            handler = DefaultServeMux
        }
        if req.RequestURI == "*" && req.Method == "OPTIONS" {
            handler = globalOptionsHandler{}
        }
        handler.ServeHTTP(rw, req)
    }
    

    So here it is finally. If we didn't provide any Handler the DefaultServeMux will be used but since we provided our foo handler ServeHTTP from foo get's called.

    And that's it. All of this can be found from server.go

提交回复
热议问题