|
|
@@ -807,21 +807,23 @@ func (this *ArticleController) Pdfwatermark() {
|
|
|
}
|
|
|
sliceLink := strings.Split(fileLink, "/")
|
|
|
uploadDir := "static/pdf/"
|
|
|
- err = os.MkdirAll(uploadDir, 777)
|
|
|
- if err != nil {
|
|
|
- br.Msg = "下载失败"
|
|
|
- br.ErrMsg = "存储目录创建失败,Err:" + err.Error()
|
|
|
- return
|
|
|
+ //判断文件夹是否存在，不存在则创建
|
|
|
+ if !utils.FileIsExist(uploadDir) {
|
|
|
+ err = os.MkdirAll(uploadDir, 0755)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "下载失败"
|
|
|
+ br.ErrMsg = "存储目录创建失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
}
|
|
|
var oldFile string
|
|
|
var newFile string
|
|
|
//获取PDF源文件名称
|
|
|
pdfName := sliceLink[len(sliceLink)-1]
|
|
|
- pdfName = utils.MD5(pdfName)
|
|
|
+ pdfName = utils.MD5(pdfName) + ".pdf"
|
|
|
oldFile = uploadDir + pdfName
|
|
|
//判断PDF本地是否存在，不存在则保存到本地
|
|
|
- _, err = os.Stat(oldFile)
|
|
|
- if err != nil {
|
|
|
+ if !utils.FileIsExist(oldFile) {
|
|
|
res, err := nhttp.Get(fileLink)
|
|
|
if err != nil {
|
|
|
br.Msg = "下载失败"
|
|
|
@@ -872,7 +874,6 @@ func (this *ArticleController) Pdfwatermark() {
|
|
|
resourceUrl := fileHost + savePath
|
|
|
defer func() {
|
|
|
os.Remove(newFile)
|
|
|
- //os.Remove(fileLink)
|
|
|
}()
|
|
|
resp.FileLink = resourceUrl
|
|
|
br.Ret = 200
|
xingzai