处理Web请求

Go定义了HandlerFuncs和一个Handler接口：

// HandlerFunc 实现了Handler接口
type HandlerFunc func(http.ResponseWriter, *http.Request)

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

net/http包对这种操作方式广泛使用。例如路由可以附加到Handler或HandlerFunc接口。本节将探讨如何在处理http.Request之后创建Handler接口，侦听本地端口以及在http.ResponseWriter接口上执行某些操作。 这是建立Go Web应用程序和RESTFul API的基础。

实践

建立 get.go：

package handlers

import (
    "fmt"
    "net/http"
)

// HelloHandler 接收GET请求中的参数"name"
// 在responds中返回 Hello <name>! 文本数据
func HelloHandler(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Content-Type", "text/plain")
    if r.Method != http.MethodGet {
        w.WriteHeader(http.StatusMethodNotAllowed)
        return
    }
    name := r.URL.Query().Get("name")

    w.WriteHeader(http.StatusOK)
    w.Write([]byte(fmt.Sprintf("Hello %s!", name)))
}

建立 post.go：

package handlers

import (
    "encoding/json"
    "net/http"
)

// GreetingResponse 用于序列化GreetingHandler返回的JSON数据
type GreetingResponse struct {
    Payload struct {
        Greeting string `json:"greeting,omitempty"`
        Name     string `json:"name,omitempty"`
        Error    string `json:"error,omitempty"`
    } `json:"payload"`
    Successful bool `json:"successful"`
}

// GreetingHandler 返回GreetingResponse格式的数据
func GreetingHandler(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Content-Type", "application/json")
    if r.Method != http.MethodPost {
        w.WriteHeader(http.StatusMethodNotAllowed)
        return
    }
    var gr GreetingResponse
    if err := r.ParseForm(); err != nil {
        gr.Payload.Error = "bad request"
        if payload, err := json.Marshal(gr); err == nil {
            w.Write(payload)
        }
    }
    name := r.FormValue("name")
    greeting := r.FormValue("greeting")

    w.WriteHeader(http.StatusOK)
    gr.Successful = true
    gr.Payload.Name = name
    gr.Payload.Greeting = greeting
    if payload, err := json.Marshal(gr); err == nil {
        w.Write(payload)
    }
}

建立 main.go：

package main

import (
    "fmt"
    "net/http"

    "github.com/agtorre/go-cookbook/chapter7/handlers"
)

func main() {
    http.HandleFunc("/name", handlers.HelloHandler)
    http.HandleFunc("/greeting", handlers.GreetingHandler)
    fmt.Println("Listening on port :3333")
    err := http.ListenAndServe(":3333", nil)
    panic(err)
}

执行go run main.go会显示

Listening on port :3333

在命令行中测试：

curl "http://localhost:3333/name?name=Reader" -X GET
Hello Reader!

curl "http://localhost:3333/greeting" -X POST -d
'name=Reader;greeting=Goodbye'
{"payload":
{"greeting":"Goodbye","name":"Reader"},"successful":true}

说明

示例中我们对GET请求和POST请求分别进行了处理。注意POST的响应是如何返回JOSN格式数据的。

这里只是简单的演示，更丰富的路由解析、限制、处理关闭等复杂操作，可以挑一些第三方库来看看他们是如何思考的。