Golang仿云盘项目-2.2保留文件元信息

本文来自博客园，作者：Jayvee，转载请注明原文链接：https://www.cnblogs.com/cenjw/p/16459817.html

目录结构

E:\goproj\FileStorageDisk│  main.go│  program.txt│  ├─handler│      handler.go│      ├─meta│      filemeta.go│      ├─static│  └─view│          index.html│          └─util        util.go

文件元信息接口

文件元信息数据结构：meta\filemeta.go

点击查看代码

package meta// FileMeta: 文件元信息结构type FileMeta struct {	FileSha1 string // unique id	FileName string	FileSize int64	Location string	UploadAt string }// 存储每个上传文件的元信息，key是文件的FileSha1var fileMetas map[string]FileMeta// 初始化func init() {	fileMetas = make(map[string]FileMeta)}// 接口1：更新或新增文件元信息func UpdateFileMeta(fmeta FileMeta) {	fileMetas[fmeta.FileSha1] = fmeta}// 接口2：通过唯一标识获取文件的元信息对象func GetFileMeta(fsha1 string) FileMeta {	return fileMetas[fsha1]}

本文来自博客园，作者：Jayvee，转载请注明原文链接：https://www.cnblogs.com/cenjw/p/16459817.html

获取文件元信息的接口

更新handler\handler.go

点击查看代码

func UploadHandler(w http.ResponseWriter, r *http.Request) {	if r.Method == "GET" {		...	} else if r.Method == "POST" {		...		defer file.Close()		fileMeta := meta.FileMeta{			FileName: head.Filename,			Location: "/tmp/" + head.Filename,			UploadAt: time.Now().Format("2006-01-02 15:04:05"),		}		// newfile, err := os.Create("/tmp/" + head.Filename)		newfile, err := os.Create(fileMeta.Location)		...		fileMeta.FileSize, err = io.Copy(newfile, file)		if err != nil {			fmt.Printf("Failed to save the data to file, err:%s\n", err.Error())			return		}		newfile.Seek(0, 0) // 把文件句柄的位置移到开始位置		fileMeta.FileSha1 = util.FileSha1(newfile) // 计算文件sha1值		meta.UpdateFileMeta(fileMeta) // 更新文件元信息		http.Redirect(w, r, "/file/upload/ok", http.StatusFound)	}}// 获取文件元信息的接口func GetFileMetaHandler(w http.ResponseWriter, r *http.Request) {	// 需要解析客户端发送请求的参数	r.ParseForm()	filehash := r.Form["filehash"][0]  // filehash要与前端对应	// 获取文件元信息对象	fMeta := meta.GetFileMeta(filehash)	// 转成jsonString格式返回客户端	data, err := json.Marshal(fMeta)	if err != nil {		w.WriteHeader(http.StatusInternalServerError)	}	w.Write(data)}

> 本文来自博客园，作者：Jayvee，转载请注明原文链接：https://www.cnblogs.com/cenjw/p/16459817.html

最后，记得到main.go注册handler function

http.HandleFunc("/file/meta", handler.GetFileMetaHandler)

util\util.go 是一个工具包，提供计算文件元信息的函数。

上传示例

1. 任意上传一张图片
   
2. 计算这张图片的sha1值
   
3. 文件上传成功后，访问获取文件元信息的接口

http://localhost:8080/file/meta?filehash=该文件的sha1值