Golang系列文章：创建Web服务

Posted 2021-01-04 js-rocker

使用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
", 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
", req.Method, req.URL, req.Proto)

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

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

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

    // 表单信息
    for k, v := range req.Form {
        fmt.Fprintf(res, "Form[%q] = %q
", 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
", req.URL.Path)
}

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

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

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