gorilla/mux实现http服务示例

ゝ一世哀愁。 2022-06-02 06:38 138阅读 0赞

[gorilla/mux][gorilla_mux] 小巧玲珑而十分高效，兼容go自带的http。下载源码到本地，编写如下示例：

package main
    
    import (
        "encoding/json"
        "fmt"
        "github.com/gorilla/mux"
        "io/ioutil"
        "net/http"
        "runtime/debug"
        "time"
    )
    
    //错误码常量
    const (
        INVALID_BODY = 1
        INVALID_PARA = 2
        INNER_ERROR  = 3
    )
    
    //请求包体
    type Req struct {
        Msg string `json:"msg"`
    }
    
    //回复包体
    type Rsp struct {
        Status int64       `json:"status"`
        Msg    string      `json:"msg"`
        Data   interface{} `json:"data,omitempty"`
    }
    
    //路由器
    var Router *mux.Router
    type Process func(query map[string][]string, body []byte, rsp *Rsp)
    
    func GenHandler(pro Process) func(w http.ResponseWriter, r *http.Request) {
        return func(w http.ResponseWriter, r *http.Request) {
            //url param
            query := r.URL.Query()
            //post data
            sbody, e := ioutil.ReadAll(r.Body)
            defer r.Body.Close()
            if e != nil {
                w.WriteHeader(500)
                return
            }
            //process
            var rsp Rsp
            fmt.Printf("Req : %v\n", string(sbody))
            pro(query, sbody, &rsp)
            fmt.Printf("Rsp : %d, %v\n", rsp.Status, rsp.Data)
    
            //rsp
            buf, e := json.Marshal(&rsp)
            if e != nil {
                w.WriteHeader(500)
            }
            w.Write([]byte(buf))
        }
    }
    
    //服务
    func ProcessEcho(query map[string][]string, body []byte, rsp *Rsp) {
        //解析url参数
        var isEcho bool
        echo := query["echo"]
        if echo != nil {
            //整型值使用strconv转换
            //state, e := strconv.ParseInt(echo[0], 10, 32)
            if echo[0] == "true" {
                isEcho = true
            } else {
                isEcho = false
            }
        }
    
        //解析post数据
        var req Req
        if e := json.Unmarshal(body, &req); e != nil {
            fmt.Printf("unmarshal fail %v [%v]\n", e, string(body))
            rsp.Status = INVALID_BODY
            rsp.Msg = "Invalid Req Body"
            return
        }
    
        //其他服务操作
    
        //回复数据
        if isEcho {
            rsp.Data = req.Msg
        }
    
        return
    }
    
    func main() {
        defer func() {
            if err := recover(); err != nil {
                debug.PrintStack()
            }
        }()
    
        Router = mux.NewRouter()
        //cgi router
        Router.HandleFunc("/echo_svr", GenHandler(ProcessEcho))
        svr := http.Server{
            Addr:         ":38080",
            ReadTimeout: 3 * time.Second,
            WriteTimeout: 3 * time.Second,
            Handler:      Router,
        }
        svr.ListenAndServe()
    }

这个示例足够满足大多数的人的http服务需求，更复杂的需求，参考mux官网使用范例。

[gorilla_mux]: https://github.com/gorilla/mux