jyfs/rpc/internal/logic/getfilelogic.go

package logic

import (
	"context"

	"fmt"
	"io/ioutil"

	"app.yhyue.com/moapp/jyfs/rpc/filesystem"
	"app.yhyue.com/moapp/jyfs/rpc/internal/svc"
	"github.com/tal-tech/go-zero/core/logx"
)

type GetFileLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewGetFileLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetFileLogic {
	return &GetFileLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

// 获取文件
func (l *GetFileLogic) GetFile(in *filesystem.LoadFileReq) (*filesystem.LoadFileResp, error) {
	// todo: add your logic here and delete this line
	// 获取存储空间。
	bucket, err := l.svcCtx.OssClient.Bucket(in.Domain)
	if err != nil {
		return &filesystem.LoadFileResp{}, err
	}
	fmt.Println("data:", in.Domain)
	fmt.Println("data:", in.FileId)
	// 下载文件到流。
	body, err := bucket.GetObject(in.FileId)
	if err != nil {
		return &filesystem.LoadFileResp{}, err
	}
	// 数据读取完成后，获取的流必须关闭，否则会造成连接泄漏，导致请求无连接可用，程序无法正常工作。
	defer body.Close()

	data, err := ioutil.ReadAll(body)
	if err != nil {
		return &filesystem.LoadFileResp{}, err
	}
	fmt.Println("data:", string(data))
	return &filesystem.LoadFileResp{RawFileContent: data}, nil
}