ossService/oss/grpc.go

package oss

import (
	"jygit.jydev.jianyu360.cn/BaseService/ossClient/constant"
	"jygit.jydev.jianyu360.cn/BaseService/ossClient/entity"
	"jygit.jydev.jianyu360.cn/BaseService/ossClient/pb"
	"log"
)

type myWriter struct {
	resp     pb.Service_DownloadServer
	errorMsg string
}

func (m *myWriter) Write(b []byte) (n int, err error) {
	n = len(b)
	err = m.resp.Send(&pb.Response{Data: b})
	return
}

type Grpc struct {
	pb.UnimplementedServiceServer
}

func (g *Grpc) Download(req *pb.DownloadRequest, resp pb.Service_DownloadServer) error {
	if err := checkArgs(&entity.Args{BucketID: req.BucketID, ObjectName: req.ObjectName}); err != nil {
		log.Println(constant.UploadFail, err)
		return err
	}
	log.Println("grpc方式下载文件", req.BucketID, req.ObjectName)
	return Download(&myWriter{resp: resp, errorMsg: constant.DownloadSuccess}, 0, req.BucketID, req.ObjectName)
}

func (g *Grpc) GetBidDetail(req *pb.DownloadRequest, resp pb.Service_DownloadServer) error {
	if err := checkArgs(&entity.Args{BucketID: req.BucketID, ObjectName: req.ObjectName}); err != nil {
		log.Println(constant.GetBidDetailFail, err)
		return err
	}
	log.Println("grpc方式获取正文", req.BucketID, req.ObjectName)
	req.ObjectName += ".txt"
	GetBidDetail(&myWriter{resp: resp, errorMsg: constant.GetBidDetailSuccess}, req.BucketID, req.ObjectName)
	return nil
}