Merge branch 'dev' of http://git.ycjcjy.com/SpaceOfCheng/services into dev

controllers/file.go

@@ -121,3 +121,28 @@ func (c *BaseController) uploadStringToAliOSS(fStr string) (string, error) {
 
 	return fileURL, nil
 }
+
+// SaveNetFilesToZip 压缩远程文件并下载
+func (c *BaseController) SaveNetFilesToZip(files []utils.NetFile, zf string) {
+	var buf bytes.Buffer
+
+	err := utils.ZipNetFiles(files, &buf)
+	if err != nil {
+		c.ResponseError(errors.New("下载 zip 文件失败"))
+	}
+
+	c.Ctx.Output.Header("Content-Disposition", "attachment; filename="+url.QueryEscape(zf))
+	c.Ctx.Output.Header("Content-Description", "File Transfer")
+	c.Ctx.Output.ContentType("application/zip")
+	c.Ctx.Output.Header("Content-Transfer-Encoding", "binary")
+	c.Ctx.Output.Header("Expires", "0")
+	c.Ctx.Output.Header("Cache-Control", "must-revalidate")
+	c.Ctx.Output.Header("Pragma", "public")
+
+	r := bytes.NewReader(buf.Bytes())
+	http.ServeContent(c.Ctx.ResponseWriter, c.Ctx.Request, zf, time.Now().Local(), r)
+
+	c.destroyContext(true)
+	c.StopRun()
+
+}

utils/zip.go

@@ -0,0 +1,54 @@
+package utils
+
+import (
+	"archive/zip"
+	"compress/flate"
+	"io"
+	"io/ioutil"
+	"net/http"
+)
+
+// NetFile 远程文件
+type NetFile struct {
+	Name string
+	URI  string
+}
+
+// ZipNetFiles 压缩远程文件
+func ZipNetFiles(files []NetFile, w io.Writer) error {
+	if files == nil || len(files) == 0 {
+		return nil
+	}
+
+	z := zip.NewWriter(w)
+	defer z.Close()
+
+	z.RegisterCompressor(zip.Deflate, func(out io.Writer) (io.WriteCloser, error) {
+		return flate.NewWriter(out, flate.BestCompression)
+	})
+
+	for _, f := range files {
+		resp, err := http.Get(f.URI)
+		if err != nil {
+			LogError("获取远程文件失败: ", err)
+			return err
+		}
+
+		dt, err := ioutil.ReadAll(resp.Body)
+		defer resp.Body.Close()
+		if err != nil {
+			LogError("读取远程文件内容失败: ", err)
+			return err
+		}
+
+		fd, err := z.Create(f.Name)
+		if err != nil {
+			LogError("压缩远程文件失败: ", err)
+			return err
+		}
+
+		fd.Write(dt)
+	}
+
+	return nil
+}