新增游戏图标，用户头像上传功能

internal/controller/wx/wx_v1_we_chat_event.go

@@ -7,6 +7,7 @@ import (
 	"server/internal/consts"
 	"server/internal/model"
 	"server/internal/service"
+	"server/utility/wechat"
 	"strings"
 
 	"github.com/gogf/gf/v2/encoding/gjson"
@@ -18,13 +19,18 @@ func (c *ControllerV1) WeChatEvent(ctx context.Context, req *v1.WeChatEventReq)
 	glog.Infof(ctx, "【微信事件】收到事件 | event=%s, msgType=%s, eventKey=%s, fromUserName=%s, toUserName=%s, createTime=%d",
 		req.Event, req.MsgType, req.EventKey, req.FromUserName, req.ToUserName, req.CreateTime,
 	)
+	unionid, err := wechat.GetWeChatClient().GetUserUnionId(ctx, req.FromUserName)
+	if err != nil {
+		glog.Errorf(ctx, "【微信事件】获取用户信息失败 | error=%s", err.Error())
+		return nil, nil
+	}
 	switch req.MsgType {
 	case "event":
 		switch req.Event {
 		case "subscribe":
 			key := strings.TrimPrefix(req.EventKey, "qrscene_")
 			split := strings.Split(req.EventKey, "_")
-			out, err := service.User().Login(ctx, &model.UserLoginIn{OpenId: req.FromUserName, StoreCode: split[0]})
+			out, err := service.User().Login(ctx, &model.UserLoginIn{OpenId: unionid, StoreCode: split[0]})
 			if err != nil {
 				return nil, err
 			}
@@ -34,7 +40,7 @@ func (c *ControllerV1) WeChatEvent(ctx context.Context, req *v1.WeChatEventReq)
 			return nil, nil
 		case "SCAN":
 			split := strings.Split(req.EventKey, "_")
-			out, err := service.User().Login(ctx, &model.UserLoginIn{OpenId: req.FromUserName, StoreCode: split[0]})
+			out, err := service.User().Login(ctx, &model.UserLoginIn{OpenId: unionid, StoreCode: split[0]})
 			if err != nil {
 				return nil, err
 			}

internal/logic/upload/upload.go

@@ -55,3 +55,65 @@ func (s *sUpload) Upload(ctx context.Context, in *model.UploadIn) (res *model.Up
 		Url: uploadRes.Url,
 	}, nil
 }
+
+func (s *sUpload) UploadGameImg(ctx context.Context, in *model.UploadIn) (res *model.UploadOut, err error) {
+	// 判断文件格式
+	fileType := in.File.Header.Get("Content-Type")
+	if !strings.HasPrefix(fileType, "image/") {
+		return nil, gerror.New("图片格式错误")
+	}
+	//判断图片大小
+	if in.File.Size > 10*1024*1024 {
+		return nil, gerror.New("图片大小最多10M")
+	}
+	// 获取图片后缀
+	exit := strings.Split(fileType, "/")[1]
+	// 文件名称
+	fileName := strings.ToLower("gameImg/" + grand.Letters(32) + "." + exit)
+	// 开始上传图片
+	client, ok := oss.GetClient("aliyun")
+	if !ok {
+		return nil, ecode.Fail.Sub("未配置OSS")
+	}
+	uploadRes, err := client.UploadFile(ctx, &model.OssUploadFileInput{
+		Filename: fileName,
+		File:     in.File,
+	})
+	if err != nil {
+		return nil, err
+	}
+	return &model.UploadOut{
+		Url: uploadRes.Url,
+	}, nil
+}
+
+func (s *sUpload) UploadUserImg(ctx context.Context, in *model.UploadIn) (res *model.UploadOut, err error) {
+	// 判断文件格式
+	fileType := in.File.Header.Get("Content-Type")
+	if !strings.HasPrefix(fileType, "image/") {
+		return nil, gerror.New("图片格式错误")
+	}
+	//判断图片大小
+	if in.File.Size > 10*1024*1024 {
+		return nil, gerror.New("图片大小最多10M")
+	}
+	// 获取图片后缀
+	exit := strings.Split(fileType, "/")[1]
+	// 文件名称
+	fileName := strings.ToLower("userImg/" + grand.Letters(32) + "." + exit)
+	// 开始上传图片
+	client, ok := oss.GetClient("aliyun")
+	if !ok {
+		return nil, ecode.Fail.Sub("未配置OSS")
+	}
+	uploadRes, err := client.UploadFile(ctx, &model.OssUploadFileInput{
+		Filename: fileName,
+		File:     in.File,
+	})
+	if err != nil {
+		return nil, err
+	}
+	return &model.UploadOut{
+		Url: uploadRes.Url,
+	}, nil
+}

internal/service/upload.go

@@ -13,6 +13,8 @@ import (
 type (
 	IUpload interface {
 		Upload(ctx context.Context, in *model.UploadIn) (res *model.UploadOut, err error)
+		UploadGameImg(ctx context.Context, in *model.UploadIn) (res *model.UploadOut, err error)
+		UploadUserImg(ctx context.Context, in *model.UploadIn) (res *model.UploadOut, err error)
 	}
 )