add BAM

server/plugin/common/util/FileDownload.go

@@ -2,8 +2,10 @@ package util
 
 import (
 	"bufio"
+	"errors"
 	"os"
 	"path/filepath"
+	"server/config"
 )
 
 /*
@@ -11,11 +13,16 @@ import (
 */
 
 // SaveOnlineFile 保存网络文件, 提供下载url和保存路径, 返回保存后的文件访问url相对路径
-func SaveOnlineFile(url, dir string) (err error) {
+func SaveOnlineFile(url, dir string) (path string, err error) {
 	// 请求获取文件内容
 	r := &RequestInfo{Uri: url}
 	ApiGet(r)
-	// 创建保存文件的目录
+	// 如果请求结果为空则直接跳过当前图片的同步, 等待后续触发时重试
+	if len(r.Resp) <= 0 {
+		err = errors.New("SyncPicture Failed: response is empty")
+		return
+	}
+	// 成功拿到图片数据 则创建保存文件的目录
 	if _, err = os.Stat(dir); os.IsNotExist(err) {
 		err = os.MkdirAll(dir, os.ModePerm)
 		if err != nil {
@@ -29,11 +36,18 @@ func SaveOnlineFile(url, dir string) (err error) {
 		return
 	}
 	defer file.Close()
-	//_, _ = file.Write(r.Resp)
-
+	// 将文件内容写入到file
 	writer := bufio.NewWriter(file)
 	_, err = writer.Write(r.Resp)
 	err = writer.Flush()
-	return
+	return filepath.Base(fileName), err
 
 }
+
+func CreateBaseDir() error {
+	// 如果不存在指定目录则创建该目录
+	if _, err := os.Stat(config.FilmPictureUploadDir); os.IsNotExist(err) {
+		return os.MkdirAll(config.FilmPictureUploadDir, os.ModePerm)
+	}
+	return nil
+}