您的位置:首页 > 理论基础 > 计算机网络

golang实现http服务器处理静态文件示例

2016-07-22 12:01 1016 查看

新版本更精简:

复制代码 代码如下: package main
import (
    "flag"
    "log"
    "net/http"
    "os"
    "io"
    "path"
    "strconv"
)
var dir string
var port int
var staticHandler http.Handler
// 初始化参数
func init() {
    dir = path.Dir(os.Args[0])
    flag.IntVar(&port, "port", 80, "服务器端口")
    flag.Parse()
    staticHandler = http.FileServer(http.Dir(dir))
}
func main() {
    http.HandleFunc("/", StaticServer)
    err := http.ListenAndServe(":"+strconv.Itoa(port), nil)
    if err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}
// 静态文件处理
func StaticServer(w http.ResponseWriter, req *http.Request) {
    if req.URL.Path != "/" {
        staticHandler.ServeHTTP(w, req)
        return
    }
    io.WriteString(w, "hello, world!\n")
}

老版本:

复制代码 代码如下: package main
import (
    "flag"
    "log"
    "net/http"
    "os"
    "path"
    "strconv"
)
var dir string
var port int
var indexs []string
// 初始化参数
func init() {
    dir = path.Dir(os.Args[0])
    flag.IntVar(&port, "port", 80, "服务器端口")
    flag.Parse()
    indexs = []string{"index.html", "index.htm"}
}
func main() {
    http.HandleFunc("/", StaticServer)
    err := http.ListenAndServe(":"+strconv.Itoa(port), nil)
    if err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}
// 静态文件处理
func StaticServer(w http.ResponseWriter, req *http.Request) {
    file := dir + req.URL.Path
    fi, err := os.Stat(file)
    if os.IsNotExist(err) {
        http.NotFound(w, req)
        return
    }
    if err != nil {
        http.Error(w, err.Error(), 500)
        return
    }
    if fi.IsDir() {
        if req.URL.Path[len(req.URL.Path)-1] != '/' {
            http.Redirect(w, req, req.URL.Path+"/", 301)
            return
        }
        for _, index := range indexs {
            fi, err = os.Stat(file + index)
            if err != nil {
                continue
            }
            http.ServeFile(w, req, file+index)
            return
        }
        http.NotFound(w, req)
        return
    }
    http.ServeFile(w, req, file)
}

希望本文所述对大家Go语言程序设计有所帮助。

您可能感兴趣的文章:

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  golang http 服务器