Gin上传文件到MongoDB gridfs

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103

func imagePost(c *gin.Context) {

	image, err := c.FormFile("file")

	if err != nil {

		errMsg := err.Error()

		logger.Error(errMsg)

		autorestapi.RespErr(c, errMsg)

		return

	}

	fileImage, err := image.Open()

	if err != nil {

		errMsg := err.Error()

		logger.Error(errMsg)

		autorestapi.RespErr(c, errMsg)

		return

	}

	defer fileImage.Close()

	imageGridFS, err := model.GetGridFS("image")

	if err != nil {

		errMsg := err.Error()

		logger.Error(errMsg)

		autorestapi.RespErr(c, errMsg)

		return

	}

	now := self_time.GetNowTimeStamp()

	nowStr := strconv.FormatInt(now, 10)

	file, err := imageGridFS.Create(nowStr)

	if err != nil {

		errMsg := err.Error()

		logger.Error(errMsg)

		autorestapi.RespErr(c, errMsg)

		return

	}

	file.SetName(nowStr)

	_, err = io.Copy(file, fileImage)

	if err != nil {

		errMsg := err.Error()

		logger.Error(errMsg)

		autorestapi.RespErr(c, errMsg)

		return

	}



	err = file.Close()

	if err != nil {

		errMsg := err.Error()

		logger.Error(errMsg)

		autorestapi.RespErr(c, errMsg)

		return

	}



	autorestapi.RespData(c, nowStr)

}

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

func imageGet(c *gin.Context) {

	imagename := c.Param("imagename")

	imageGridFS, err := model.GetGridFS("image")

	if err != nil {

		errMsg := err.Error()

		logger.Error(errMsg)

		autorestapi.RespErr(c, errMsg)

		return

	}

	file, err := imageGridFS.Open(imagename)

	if err != nil {

		errMsg := err.Error()

		logger.Error(errMsg)

		autorestapi.RespErr(c, errMsg)

		return

	}

	size := file.Size()

	io.Copy(c.Writer, file)

	err = file.Close()

	if err != nil {

		errMsg := err.Error()

		logger.Error(errMsg)

		autorestapi.RespErr(c, errMsg)

		return

	}

	c.Writer.WriteHeader(http.StatusOK)

	c.Header("Content-Disposition", fmt.Sprintf("attachment; filename=%s", imagename))

	c.Header("Content-Type", "application/text/plain")

	c.Header("Accept-Length", fmt.Sprintf("%d", size))

}