openbilibili/app/service/main/push/server/http/upload.go

package http

import (
	"path/filepath"

	"go-common/app/service/main/push/conf"
	"go-common/library/ecode"
	"go-common/library/log"
	bm "go-common/library/net/http/blademaster"
)

var imgExts = map[string]bool{
	".jpg":  true,
	".jpeg": true,
	".png":  true,
}

func upimg(ctx *bm.Context) {
	f, h, err := ctx.Request.FormFile("file")
	if err != nil {
		log.Error("upimg error(%v)", err)
		ctx.JSON(nil, err)
		return
	}
	defer f.Close()
	if h.Size > conf.Conf.Push.UpimgMaxSize {
		log.Error("filesize error name(%s) size(%d)", h.Filename, h.Size)
		ctx.JSON(nil, ecode.PushServiceFileSizeErr)
		return
	}
	if ok := imgExts[filepath.Ext(h.Filename)]; !ok {
		log.Error("file ext error name(%s)", h.Filename)
		ctx.JSON(nil, ecode.PushServiceFileExtErr)
		return
	}
	url, err := pushSrv.Upimg(ctx, f)
	if err != nil {
		ctx.JSON(nil, err)
		return
	}
	ctx.JSON(map[string]string{"url": url}, nil)
}