feat: ota

services/chore/internal/logic/internalUploadLogic.go

@@ -0,0 +1,56 @@
+package logic
+
+import (
+	"context"
+	"strings"
+
+	"lone-services/pkg/oss"
+	"lone-services/pkg/utils"
+	"lone-services/pkg/validate"
+	chore "lone-services/rpc/chore/pb"
+	"lone-services/services/chore/internal/svc"
+	"lone-services/services/chore/validator"
+
+	"github.com/zeromicro/go-zero/core/logx"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+type InternalUploadLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewInternalUploadLogic(ctx context.Context, svcCtx *svc.ServiceContext) *InternalUploadLogic {
+	return &InternalUploadLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+// InternalUpload 对内：服务间直传 OSS（无 HTTP 暴露）
+func (l *InternalUploadLogic) InternalUpload(in *chore.InternalUploadReq) (*chore.InternalUploadData, error) {
+	var req validator.UploadValidator
+	if msg := validate.ValidateFromProto(in, &req); msg != utils.StringEmpty {
+		return nil, status.Error(codes.InvalidArgument, msg)
+	}
+
+	res, err := oss.Upload(oss.UploadInput{
+		Filename: req.Filename,
+		Content:  req.Content,
+	})
+	if err != nil {
+		l.Errorf("internal oss upload: %v", err)
+		if strings.Contains(err.Error(), "empty file") || strings.Contains(err.Error(), "too large") {
+			return nil, status.Error(codes.InvalidArgument, err.Error())
+		}
+		return nil, status.Error(codes.Internal, utils.Fail.Msg)
+	}
+	return &chore.InternalUploadData{
+		Url:  res.URL,
+		Path: res.Path,
+		Host: res.Host,
+	}, nil
+}

services/chore/internal/logic/uploadLogic.go

@@ -0,0 +1,49 @@
+package logic
+
+import (
+	"context"
+	"strings"
+
+	"lone-services/pkg/oss"
+	"lone-services/pkg/utils"
+	"lone-services/pkg/validate"
+	chore "lone-services/rpc/chore/pb"
+	"lone-services/services/chore/internal/svc"
+	"lone-services/services/chore/validator"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type UploadLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewUploadLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UploadLogic {
+	return &UploadLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *UploadLogic) Upload(in *chore.UploadReq) (*chore.Response, error) {
+	var req validator.UploadValidator
+	if msg := validate.ValidateFromProto(in, &req); msg != utils.StringEmpty {
+		return outResponse(utils.ErrorParams, msg), nil
+	}
+
+	res, err := oss.Upload(oss.UploadInput{
+		Filename: req.Filename,
+		Content:  req.Content,
+	})
+	if err != nil {
+		l.Errorf("oss upload: %v", err)
+		if strings.Contains(err.Error(), "empty file") || strings.Contains(err.Error(), "too large") {
+			return outResponse(utils.ErrorParams, err.Error()), nil
+		}
+		return failResponse(utils.Fail), nil
+	}
+	return okResponse(res), nil
+}

services/chore/internal/server/choreServer.go

@@ -28,6 +28,18 @@ func (s *ChoreServer) Policy(ctx context.Context, in *chore.PolicyReq) (*chore.R
 	return l.Policy(in)
 }
 
+// 对外：服务端直传 OSS
+func (s *ChoreServer) Upload(ctx context.Context, in *chore.UploadReq) (*chore.Response, error) {
+	l := logic.NewUploadLogic(ctx, s.svcCtx)
+	return l.Upload(in)
+}
+
+// 对内：服务间直传 OSS（无 HTTP 暴露）
+func (s *ChoreServer) InternalUpload(ctx context.Context, in *chore.InternalUploadReq) (*chore.InternalUploadData, error) {
+	l := logic.NewInternalUploadLogic(ctx, s.svcCtx)
+	return l.InternalUpload(in)
+}
+
 func (s *ChoreServer) Test(ctx context.Context, in *chore.TestReq) (*chore.Response, error) {
 	l := logic.NewTestLogic(ctx, s.svcCtx)
 	return l.Test(in)