upload_api.go 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. /*
  2. * @description:
  3. * @Author: CP
  4. * @Date: 2021-03-31 15:06:21
  5. * @FilePath: \design_quantity\web\api\upload_api.go
  6. */
  7. package api
  8. import (
  9. "bytes"
  10. "fmt"
  11. "io"
  12. "log"
  13. "os"
  14. "strings"
  15. "time"
  16. "github.com/kataras/iris/v12"
  17. "go.mod/conf"
  18. "go.mod/services"
  19. )
  20. type UploadApi struct {
  21. //框架-web应用上下文环境
  22. Ctx iris.Context
  23. // // 需要用的service
  24. ServiceContract services.ContractService
  25. }
  26. // Upload : 处理文件上传
  27. func (c *UploadApi) Post() {
  28. errCode := 0
  29. id := "0"
  30. defer func() {
  31. if errCode < 0 {
  32. c.Ctx.JSON(iris.Map{"code": -1, "msg": "上传失败"})
  33. } else {
  34. c.Ctx.JSON(iris.Map{"code": 0, "msg": "上传成功", "data": id})
  35. }
  36. return
  37. }()
  38. file, head, err := c.Ctx.FormFile("file")
  39. if err != nil {
  40. log.Printf("Failed to get form data, err:%s\n", err.Error())
  41. errCode = -1
  42. return
  43. }
  44. defer file.Close()
  45. buf := bytes.NewBuffer(nil)
  46. if _, err := io.Copy(buf, file); err != nil {
  47. log.Printf("Failed to get file data, err:%s\n", err.Error())
  48. errCode = -2
  49. return
  50. }
  51. // 文件元信息-todo
  52. // FileName := head.Filename
  53. FileSize := int64(len(buf.Bytes()))
  54. UploadAt := fmt.Sprintf("%d", time.Now().UnixNano())
  55. Location := conf.MergeLocalRootDir + UploadAt + head.Filename // 存储地址
  56. newFile, err := os.Create(Location)
  57. if err != nil {
  58. log.Printf("Failed to create file, err:%s\n", err.Error())
  59. errCode = -3
  60. return
  61. }
  62. defer newFile.Close()
  63. nByte, err := newFile.Write(buf.Bytes())
  64. if int64(nByte) != FileSize || err != nil {
  65. log.Printf("Failed to save data into file, writtenSize:%d, err:%s\n", nByte, err.Error())
  66. errCode = -4
  67. return
  68. }
  69. // fmt.Println(c.Ctx.FormValue("id"))
  70. // treeVM := &viewmodels.TreeSectionContract{}
  71. // err = c.Ctx.ReadForm(treeVM)
  72. // if err != nil {
  73. // log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  74. // errCode = -6
  75. // return
  76. // }
  77. // fmt.Println(treeVM)
  78. // id, err := utils.GetDecryptId(treeVM.Id)
  79. // if err != nil {
  80. // errCode = -6
  81. // return
  82. // }
  83. ext := strings.Join(head.Header["Content-Type"], "")
  84. id, err = c.ServiceContract.SaveUpload(Location, head.Filename, ext)
  85. if err != nil {
  86. errCode = -6
  87. return
  88. }
  89. }