Merge branch 'master' of http://192.168.5.5:10010/r/aps/SRM

liujiandao

2023-11-16 d20acf38c36c11ee4428c3e74a17f5870dc61b51

 api/v1/test/contract.go

@@ -1,14 +1,20 @@
package test

import (
   "github.com/flipped-aurora/gin-vue-admin/server/global"
   "github.com/flipped-aurora/gin-vue-admin/server/model/common/request"
   "github.com/flipped-aurora/gin-vue-admin/server/model/common/response"
   "github.com/flipped-aurora/gin-vue-admin/server/model/test"
   testReq "github.com/flipped-aurora/gin-vue-admin/server/model/test/request"
   "github.com/flipped-aurora/gin-vue-admin/server/service"
   "bytes"
   "fmt"
   "github.com/gin-gonic/gin"
   "github.com/h2non/filetype"
   "go.uber.org/zap"
   "net/http"
   "srm/global"
   "srm/model/common/request"
   "srm/model/common/response"
   "srm/model/test"
   testReq "srm/model/test/request"
   "srm/service"
   "strconv"
   "time"
)

type ContractApi struct {
@@ -25,9 +31,9 @@
// @Param      file   formData   file   true   "上传文件"
// @Param name formData string true "文件名称"
// @Success 200 {string} string "{"success":true,"data":{},"msg":"获取成功"}"
// @Router /api/con/createContract [post]
// @Router /con/createContract [post]
func (conApi *ContractApi) CreateContract(c *gin.Context) {
   name := c.Param("name")
   name := c.PostForm("name")
   file, err := c.FormFile("file")
   if err != nil {
      c.JSON(400, gin.H{"error": err.Error()})
@@ -39,9 +45,20 @@
   defer f.Close()
   f.Read(fileContent)

   // 重置文件指针
   f.Seek(0, 0)

   // 使用 filetype 库检测文件类型
   kind, _ := filetype.Match(fileContent)
   if kind == filetype.Unknown {
      c.JSON(http.StatusBadRequest, gin.H{"error": "Unknown file type"})
      return
   }

   contract := test.Contract{
      FileName:    name,
      FileContent: fileContent,
      FileType:    kind.MIME.Value,
   }

   if err, id := conService.CreateContract(&contract); err != nil {
@@ -49,7 +66,7 @@
      response.FailWithMessage("创建失败", c)
   } else {
      //response.OkWithMessage("创建成功", c)
      response.OkWithData(gin.H{"id": id}, c)
      response.OkWithData(gin.H{"id": id, "name": name}, c)
   }
}

@@ -61,7 +78,7 @@
// @Produce application/json
// @Param data body test.Contract true "删除Contract"
// @Success 200 {string} string "{"success":true,"data":{},"msg":"删除成功"}"
// @Router /api/con/deleteContract [delete]
// @Router /con/deleteContract [delete]
func (conApi *ContractApi) DeleteContract(c *gin.Context) {
   var con test.Contract
   err := c.ShouldBindJSON(&con)
@@ -85,7 +102,7 @@
// @Produce application/json
// @Param data body request.IdsReq true "批量删除Contract"
// @Success 200 {string} string "{"success":true,"data":{},"msg":"批量删除成功"}"
// @Router /api/con/deleteContractByIds [delete]
// @Router /con/deleteContractByIds [delete]
func (conApi *ContractApi) DeleteContractByIds(c *gin.Context) {
   var IDS request.IdsReq
   err := c.ShouldBindJSON(&IDS)
@@ -109,7 +126,7 @@
// @Produce application/json
// @Param data body test.Contract true "更新Contract"
// @Success 200 {string} string "{"success":true,"data":{},"msg":"更新成功"}"
// @Router /api/con/updateContract [put]
// @Router /con/updateContract [put]
func (conApi *ContractApi) UpdateContract(c *gin.Context) {
   var con test.Contract
   err := c.ShouldBindJSON(&con)
@@ -133,7 +150,7 @@
// @Produce application/json
// @Param data query test.Contract true "用id查询Contract"
// @Success 200 {string} string "{"success":true,"data":{},"msg":"查询成功"}"
// @Router /api/con/findContract [get]
// @Router /con/findContract [get]
func (conApi *ContractApi) FindContract(c *gin.Context) {
   var con test.Contract
   err := c.ShouldBindQuery(&con)
@@ -157,7 +174,7 @@
// @Produce application/json
// @Param data query testReq.ContractSearch true "分页获取Contract列表"
// @Success 200 {string} string "{"success":true,"data":{},"msg":"获取成功"}"
// @Router /api/con/getContractList [get]
// @Router /con/getContractList [get]
func (conApi *ContractApi) GetContractList(c *gin.Context) {
   var pageInfo testReq.ContractSearch
   err := c.ShouldBindQuery(&pageInfo)
@@ -184,22 +201,98 @@
// @Security ApiKeyAuth
// @accept application/json
// @Produce application/json
// @Param data query testReq.ContractSearch true "预览Contract"
// @Param data query test.Contract true "用id查询Contract"
// @Success 200 {string} string "{"success":true,"data":{},"msg":"预览成功"}"
// @Router /api/con/previewContract [get]
// @Router /con/previewContract [get]
func (conApi *ContractApi) PreviewContract(c *gin.Context) {
   var pageInfo testReq.ContractSearch
   err := c.ShouldBindQuery(&pageInfo)
   var con test.Contract
   err := c.ShouldBindQuery(&con)
   if err != nil {
      response.FailWithMessage(err.Error(), c)
      return
   }
   contract, err := conService.GetContract(pageInfo.ID)

   id := c.Query("id")
   if id == "" {
      response.FailWithMessage("id不能为空", c)
      return
   }

   val64, err := strconv.ParseUint(id, 10, 64)
   if err != nil {
      response.FailWithMessage("id格式错误", c)
      return
   }

   // Convert uint64 to uint
   conId := uint(val64)

   contract, err := conService.GetContract(conId)
   if err != nil {
      global.GVA_LOG.Error("获取失败!", zap.Error(err))
      response.FailWithMessage("获取失败", c)
      return
   } else {
      c.Data(200, "application/pdf", contract.FileContent)
      //c.Writer.Header().Set("Content-Type", "application/octect-stream")
      //c.Writer.Header().Set("Content-Disposition", "attachment;filename="+contract.FileName)
      //c.Writer.Write(contract.FileContent)
      //c.Data(200, "application/octect-stream", contract.FileContent)
      //reader := bytes.NewReader(contract.FileContent)
      //c.DataFromReader(http.StatusOK, int64(len(contract.FileContent)), "application/pdf", reader, map[string]string{"Content-Disposition": fmt.Sprintf("attachment; filename=%s", contract.FileName)})
      reader := bytes.NewReader(contract.FileContent)
      c.Header("Content-Type", contract.FileType)
      c.Writer.Header().Set("Content-Disposition", "inline; filename="+contract.FileName)
      http.ServeContent(c.Writer, c.Request, contract.FileName, time.Now(), reader)
   }
}

// DownloadContract 下载Contract
// @Tags Contract
// @Summary 下载Contract
// @Security ApiKeyAuth
// @accept application/json
// @Produce application/json
// @Param data query test.Contract true "用id查询Contract"
// @Success 200 {string} string "{"success":true,"data":{},"msg":"下载成功"}"
// @Router /con/downloadContract [get]
func (conApi *ContractApi) DownloadContract(c *gin.Context) {
   var con test.Contract
   err := c.ShouldBindQuery(&con)
   if err != nil {
      response.FailWithMessage(err.Error(), c)
      return
   }

   id := c.Query("id")
   if id == "" {
      response.FailWithMessage("id不能为空", c)
      return
   }

   val64, err := strconv.ParseUint(id, 10, 64)
   if err != nil {
      response.FailWithMessage("id格式错误", c)
      return
   }

   // Convert uint64 to uint
   conId := uint(val64)

   contract, err := conService.GetContract(conId)
   if err != nil {
      global.GVA_LOG.Error("获取失败!", zap.Error(err))
      response.FailWithMessage("获取失败", c)
      return
   } else {
      reader := bytes.NewReader(contract.FileContent)
      // 设置必要的头信息
      c.Header("Content-Description", "File Transfer")
      c.Header("Content-Transfer-Encoding", "binary")
      c.Header("Content-Disposition", fmt.Sprintf("attachment; filename=%s", contract.FileName))
      c.Header("Content-Type", contract.FileType)

      // 将文件作为响应体发送
      http.ServeContent(c.Writer, c.Request, contract.FileName, time.Now(), reader)
      c.File(contract.FileName)
   }
}