blog_backend_api/controller/file.go

package controller

import (
	"fmt"
	"github.com/gin-gonic/gin"
	uuid "github.com/satori/go.uuid"
	"io"
	"io/ioutil"
	"log"
	"net/http"
	"os"
	"strings"
)

type FileController struct {
}

func (this *FileController) OnUpload(c *gin.Context) {
	uid,e := uuid.NewV1()
	if nil != e{
		log.Print(e.Error())
		return
	}

	file, _, err := c.Request.FormFile("image")
	if nil != err || nil == file{
		log.Print(err.Error())
		return
	}
	bytes,err := ioutil.ReadAll(file)
	if nil != err{
		log.Print(err.Error())
		return
	}
	header := make([]byte,512)
	copy(header,bytes)
	types := http.DetectContentType(header)
	log.Print(types)
	// jpg
	if strings.Contains(types,"jpeg"){
		defer file.Close()
		out, err := os.Create("G://image//" + uid.String() + ".jpg")
		if err != nil {
			log.Fatal(err)
		}
		defer out.Close()
		out.Write(bytes)
		log.Print(len(bytes))
		c.JSON(200, map[string] interface{}{"url":uid.String()})
	}
}

func (this *FileController) OnDownLoad(c *gin.Context) {
	file, header, err := c.Request.FormFile("upload")
	filename := header.Filename
	fmt.Println(header.Filename)
	out, err := os.Create("G://GoPath//image//" + filename + ".png")
	if err != nil {
		log.Fatal(err)
	}
	defer out.Close()
	_, err = io.Copy(out, file)
	if err != nil {
		log.Fatal(err)
	}
}