Golang开发一个简单的相册网站

最新推荐文章于 2024-05-21 10:09:25 发布

瑾修

最新推荐文章于 2024-05-21 10:09:25 发布

阅读量1.7k

点赞数 2

分类专栏： golang 文章标签： Golang Go web

本文链接：https://blog.csdn.net/weixin_41093846/article/details/87528036

版权

golang 专栏收录该内容

11 篇文章 0 订阅

订阅专栏

源码来源于《Go语言编程》（许式伟著）第一版，第5章网络编程 5.5.3 开发一个简单的相册网站。

由于学习过程中发现一些源码的小问题。在此把修改后的代码发一下，做个记录。

main.go

package main

import (
	"html/template"
	"io"
	"io/ioutil"
	"log"
	"net/http"
	"os"
	"path"
	"runtime/debug"
)

const (
	ListDir      = 0x0001
	UPLOAD_DIR   = "./uploads"
	TEMPLATE_DIR = "./views"
)

var templates map[string]*template.Template

func init() {
	templates = make(map[string]*template.Template)
	fileInfoArr, err := ioutil.ReadDir(TEMPLATE_DIR)
	check(err)
	var templateName, templatePath string
	for _, fileInfo := range fileInfoArr {
		templateName = fileInfo.Name()
		if ext := path.Ext(templateName); ext != ".html" {
			continue
		}
		templatePath = TEMPLATE_DIR + "/" + templateName
		log.Println("init Loading templatePath:", templatePath)
		t := template.Must(template.ParseFiles(templatePath))
		templates[templateName] = t
	}
}

func check(err error) {
	if err != nil {
		panic(err)
	}
}

func renderHtml(w http.ResponseWriter, tmpl string, locals map[string]interface{}) {
	err := templates[tmpl].Execute(w, locals)
	check(err)
}

func isExists(path string) bool {
	_, err := os.Stat(path)
	if err == nil {
		return true
	}
	return os.IsExist(err)
}

func uploadHandler(w http.ResponseWriter, r *http.Request) {
	if r.Method == "GET" {
		renderHtml(w, "upload.html", nil) //原文中"upload"改为"upload.html"
	}
	if r.Method == "POST" {
		f, h, err := r.FormFile("image")
		check(err)
		filename := h.Filename
		log.Println("uploadHandler :", filename)
		defer f.Close()
		// t, err := ioutil.TempFile(UPLOAD_DIR, filename) //创建临时文件修改为 创建普通文件
		t, err := os.OpenFile(UPLOAD_DIR+"/"+filename, os.O_WRONLY|os.O_CREATE, 0666)
		check(err)
		defer t.Close()
		_, err2 := io.Copy(t, f)
		check(err2)
		http.Redirect(w, r, "/view?id="+filename,
			http.StatusFound)
	}
}

func viewHandler(w http.ResponseWriter, r *http.Request) {
	imageId := r.FormValue("id")
	imagePath := UPLOAD_DIR + "/" + imageId
	log.Println("viewHandler :", imageId, imagePath)
	if exists := isExists(imagePath); !exists {
		http.NotFound(w, r)
		return
	}
	w.Header().Set("Content-Type", "image")
	http.ServeFile(w, r, imagePath)
}

func listHandler(w http.ResponseWriter, r *http.Request) {
	fileInfoArr, err := ioutil.ReadDir("./uploads")
	check(err)
	locals := make(map[string]interface{})
	images := []string{}
	for _, fileInfo := range fileInfoArr {
		images = append(images, fileInfo.Name())
	}
	locals["images"] = images
	renderHtml(w, "list.html", locals) //原文中"list"改为"list.html"
}

func safeHandler(fn http.HandlerFunc) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		defer func() {
			if err, ok := recover().(error); ok {
				http.Error(w, err.Error(), http.StatusInternalServerError)
				// 或者输出自定义的50x错误页面
				// w.WriteHeader(http.StatusInternalServerError)
				// renderHtml(w, "error", e)
				// logging
				log.Println("WARN: panic in %v. - %v", fn, err)
				log.Println(string(debug.Stack()))
			}
		}()
		fn(w, r)
	}
}

func staticDirHandler(mux *http.ServeMux, prefix string, staticDir string, flags int) {
	mux.HandleFunc(prefix, func(w http.ResponseWriter, r *http.Request) {
		file := staticDir + r.URL.Path[len(prefix)-1:]
		if (flags & ListDir) == 0 {
			if exists := isExists(file); !exists {
				http.NotFound(w, r)
				return
			}
		}
		http.ServeFile(w, r, file)
	})
}

func main() {
	mux := http.NewServeMux()
	staticDirHandler(mux, "/assets/", "./public", 0)
	mux.HandleFunc("/", safeHandler(listHandler))
	mux.HandleFunc("/view", safeHandler(viewHandler))
	mux.HandleFunc("/upload", safeHandler(uploadHandler))
	err := http.ListenAndServe(":8080", mux)
	if err != nil {
		log.Fatal("ListenAndServe: ", err.Error())
	}
}

list.html

<!doctype html>
<html>
<head>
<meta charset="utf-8">
<title>List</title>
</head>
<body>
<ol>
{{range $.images}}
<li><a href="/view?id={{.|urlquery}}">{{.|html}}</a></li>
{{end}}
</ol>
</body>
</html>

upload.html

<!doctype html>
<html>
<head>
<meta charset="utf-8">
<title>Upload</title>
</head>
<body>
<form method="POST" action="/upload" enctype="multipart/form-data">
Choose an image to upload: <input name="image" type="file" />
<input type="submit" value="Upload" />
</form>
</body>
</html>

文件路径是这样的

可能是作者这份源码自己没运行过吧，从书上敲下来的代码是跑不起来的，有点问题。然后调试修改的过程，也是学习的过程。

现在改后的应该是可以的。

瑾修

关注

2
点赞
踩
6

收藏

觉得还不错? 一键收藏
0
评论
Golang开发一个简单的相册网站

源码来源于《Go语言编程》（许式伟著）第一版，第5章网络编程 5.5.3开发一个简单的相册网站。由于学习过程中发现一些源码的小问题。在此把修改后的代码发一下，做个记录。main.gopackage mainimport ( "html/template" "io" "io/ioutil" "log" "net/http"...
复制链接

扫一扫

专栏目录