Golang: 创建Web服务

使用 Go 语言,我们可以轻松创建出 Web 服务,这一点比 Node.js 还要简单,今天就来总结一下 Go 语言中 Web 服务的创建方式。

首先,我们需要引入 net/http 这个包来处理 HTTP 请求,然后在指定的端口开启服务,下面来写一个最简单的 Web 服务程序:

// server.go

package main

import (
    "fmt"
    "log"
    "net/http"
)

func main() {
    http.HandleFunc("/", handler)

    log.Println("go server listening at port 3000...")

    err := http.ListenAndServe(":3000", nil)

    if err != nil {
        log.Fatal("err: ", err)
    }
}

func handler(res http.ResponseWriter, req *http.Request) {
    log.Println(req.URL.Path)
    fmt.Fprintf(res, "URL.Path = %q\n", req.URL.Path)
}

上面代码中,我们会处理所有 3000 端口的请求,然后在页面显示请求的路径。另外,我们在代码中引入了 log 包,使用 log.Println(s) 方法打印信息,运行代码后,在浏览器请求 localhost:3000 及 localhost:3000/test,控制台打印信息如下:

$ go run server.go
2018/08/30 13:26:58 go server listening at port 3000...
2018/08/30 13:27:01 /
2018/08/30 13:27:07 /test

接下来,我们希望这个服务能够解析出浏览器的请求信息,返回并显示到页面,下面是经过改进后的代码:

// server.go

package main

import (
    "fmt"
    "log"
    "net/http"
)

func main() {
    http.HandleFunc("/", handler)

    log.Println("go server listening at port 3000...")

    err := http.ListenAndServe(":3000", nil)

    if err != nil {
        log.Fatal("err: ", err)
    }
}

func handler(res http.ResponseWriter, req *http.Request) {
    // 请求方法 请求地址 协议类型
    fmt.Fprintf(res, "%s %s %s\n", req.Method, req.URL, req.Proto)

    // 请求头信息
    for k, v := range req.Header {
        fmt.Fprintf(res, "Header[%q] = %q\n", k, v)
    }

    // 请求的服务器URL & 请求的远程地址
    fmt.Fprintf(res, "Host = %q\n", req.Host)
    fmt.Fprintf(res, "RemoteAddr = %q\n", req.RemoteAddr)

    if err := req.ParseForm(); err != nil {
        log.Print(err)
    }

    // 表单信息
    for k, v := range req.Form {
        fmt.Fprintf(res, "Form[%q] = %q\n", k, v)
    }
}

在 handler 函数中,我们从 http.Request 中取出相应的 HTTP 请求信息,然后返回给浏览器,大家可以亲自运行以上代码,来观察实际效果。

接下来,我们希望做个小功能 - 统计用户的访问次数,这是一项很古老的技术了,在上个世纪的论坛网站中曾广泛应用。😃

我们改进后的代码如下:

// server.go

package main

import (
    "fmt"
    "log"
    "sync"
    "net/http"
)

var mutex sync.Mutex
var count int

func main() {
    http.HandleFunc("/", handler)
    // 添加/count路由及处理函数
    http.HandleFunc("/count", counter)

    log.Println("go server listening at port 3000...")

    err := http.ListenAndServe(":3000", nil)

    if err != nil {
        log.Fatal("err: ", err)
    }
}

// 每次访问时 count递增
func handler(res http.ResponseWriter, req *http.Request) {
    mutex.Lock()
    count++
    mutex.Unlock()

    fmt.Fprintf(res, "URL.Path = %q\n", req.URL.Path)
}

// 显示当前访问次数
func counter(res http.ResponseWriter, req *http.Request) {
    mutex.Lock()
    fmt.Fprintf(res, "Visit Count: %d\n", count)
    mutex.Unlock()
}

这次我们引入了 sync 包,来应对并发访问带来的数据更新问题。sync.Mutex 类提供了 Lock() 和 Unlock() 方法,分别对更新操作进行加锁和解锁,保证 count 的递增操作不会受到并发访问的影响。

运行上面的代码,然后在浏览器中访问除 /count 之外的路径,都可以累积访问次数,最后,在访问 /count 路径时,会返回当前服务被访问的次数。

posted @ 2018-08-10 06:33 liuhe688 阅读(...) 评论(...) 编辑 收藏