golang http服务器处理静态文件

2012-09-22 01:17  2603人阅读  评论 (0)
Tags: golang

新版本更精简:

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)
}