init: init refactor

app/file/rpc/internal/logic/getFileByIdLogic.go

@@ -0,0 +1,58 @@
+package logic
+
+import (
+	"context"
+
+	fileModel "sundynix-micro-go/app/file/model"
+	"sundynix-micro-go/app/file/rpc/file"
+	"sundynix-micro-go/app/file/rpc/internal/svc"
+
+	"github.com/zeromicro/go-zero/core/logx"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+	"gorm.io/gorm"
+)
+
+type GetFileByIdLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewGetFileByIdLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetFileByIdLogic {
+	return &GetFileByIdLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+// 根据ID获取文件信息
+func (l *GetFileByIdLogic) GetFileById(in *file.GetFileByIdReq) (*file.GetFileByIdResp, error) {
+	var oss fileModel.SundynixOss
+	err := l.svcCtx.DB.Where("id = ?", in.Id).First(&oss).Error
+	if err != nil {
+		if err == gorm.ErrRecordNotFound {
+			return nil, status.Error(codes.NotFound, "文件不存在")
+		}
+		l.Errorf("查询文件失败: %v", err)
+		return nil, status.Error(codes.Internal, "查询文件失败")
+	}
+
+	return &file.GetFileByIdResp{
+		File: convertOssToProto(&oss),
+	}, nil
+}
+
+func convertOssToProto(oss *fileModel.SundynixOss) *file.FileInfo {
+	return &file.FileInfo{
+		Id:        oss.ID,
+		Name:      oss.Name,
+		Url:       oss.Url,
+		Tag:       oss.Tag,
+		Key:       oss.Key,
+		Suffix:    oss.Suffix,
+		Md5:       oss.MD5,
+		CreatedAt: oss.CreatedAt.Unix(),
+	}
+}

app/file/rpc/internal/logic/getFilesByIdsLogic.go

@@ -0,0 +1,48 @@
+package logic
+
+import (
+	"context"
+
+	fileModel "sundynix-micro-go/app/file/model"
+	"sundynix-micro-go/app/file/rpc/file"
+	"sundynix-micro-go/app/file/rpc/internal/svc"
+
+	"github.com/zeromicro/go-zero/core/logx"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+type GetFilesByIdsLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewGetFilesByIdsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetFilesByIdsLogic {
+	return &GetFilesByIdsLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+// 根据ID列表批量获取文件信息
+func (l *GetFilesByIdsLogic) GetFilesByIds(in *file.GetFilesByIdsReq) (*file.GetFilesByIdsResp, error) {
+	if len(in.Ids) == 0 {
+		return &file.GetFilesByIdsResp{Files: []*file.FileInfo{}}, nil
+	}
+
+	var ossList []fileModel.SundynixOss
+	err := l.svcCtx.DB.Where("id IN ?", in.Ids).Find(&ossList).Error
+	if err != nil {
+		l.Errorf("批量查询文件失败: %v", err)
+		return nil, status.Error(codes.Internal, "批量查询文件失败")
+	}
+
+	files := make([]*file.FileInfo, 0, len(ossList))
+	for _, oss := range ossList {
+		files = append(files, convertOssToProto(&oss))
+	}
+
+	return &file.GetFilesByIdsResp{Files: files}, nil
+}