1 、文件的上传
文件的上传是以分块的形式上传,文件的值为缓存目录,存储在缓存路径下。当分块上传完毕后,则会合并文件夹,将缓存文件夹删除。
2、 分块上传
文件的分块已经由前端切割好,发送的请求中req.Action字段会赋值为chunk,网盘只用负责接收请求并根据请求中的动作做出相对应的处理。
为保证文件的完整性,会使用文件的哈希值作为缓存路径的文件夹名,并在该路径下生成缓存文件。
cachePath := req.getCachePath(user.UserID)
if err = fs.Mkdir(cachePath); err != nil {
err = errors.Wrap(err, errors.InternalServerErr)
return
}
if req.isFileExist(filepath.Join(cachePath, req.chunkNumber)) {
return
}
tmpFile, err := ioutil.TempFile(filepath.Join(fs.GetRoot(), cachePath), "temp-")
if err != nil {
err = errors.Wrap(err, errors.InternalServerErr)
return
}
defer tmpFile.Close()
赋值上传文件的内容,重新命名缓存文件。
3. 分块合并
当上传最后一块分块时,req.Action会赋值为merge,插件会根据该字段进行分块合并的操作。
打开缓存路径,统计分块总数是否与请求中的分块总数一致:
fileInfos, err := file.Readdir(-1)
cachePath := req.getCachePath(user.UserID)
file, err := fs.Open(cachePath)
totalChunks, _ := strconv.Atoi(req.TotalChunks)
if len(fileInfos) != totalChunks {
err = errors.New(status.ChunkFileNotExistErr)
return
}
创建临时文件,将缓存目录下的所有的缓存文件写入到临时文件中,并检验临时文件的哈希值。
tempFile, err := ioutil.TempFile(filepat