jy_docs/rpc/partnerlib/test/fileUpload_test.go

package test

import (
	"app.yhyue.com/moapp/jyfs/rpc/filesystem"
	"app.yhyue.com/moapp/jyfs/rpc/filesystemclient"
	"fmt"
	"github.com/gogf/gf/v2/os/gctx"
	"github.com/zeromicro/go-zero/core/discov"
	"github.com/zeromicro/go-zero/zrpc"
	"google.golang.org/grpc"
	"log"
	"testing"
)

var (
	Domain = "jydocs-std"
)

func Test_FileUpload(t *testing.T) {
	log.Println("-------------")
	//获取附件后上传oss
	fr := &filesystem.SaveFileReq{
		Domain: Domain,
		FileId: fmt.Sprintf("%s.%s", "101010", "doc"),
		Meta: map[string]string{
			"docName":   "101020",
			"docSuffix": "doc",
			"docSize":   "24",
		},
		RawFileContent: []byte("我是中国人"),
	}
	if fr != nil && len(fr.RawFileContent) > 0 {
		conf := zrpc.RpcClientConf{
			Etcd: discov.EtcdConf{
				Key:   "moapp.filesystem.rpc",
				Hosts: []string{"192.168.3.206:2379"},
			},
		}
		grpcOpts := []zrpc.ClientOption{
			//zrpc.WithDialOption(grpc.WithWriteBufferSize(40 * 1024 * 1024)), // 设置发送消息的最大字节数，这里是40MB
			//zrpc.WithDialOption(grpc.WithReadBufferSize(40 * 1024 * 1024)),  // 设置接收消息的最大字节数，这里也是40MB
			zrpc.WithDialOption(grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(40 * 1024 * 1024))),
			zrpc.WithDialOption(grpc.WithDefaultCallOptions(grpc.MaxCallSendMsgSize(40 * 1024 * 1024))),
		}
		client := zrpc.MustNewClient(conf, grpcOpts...)
		jyFilelLib := filesystemclient.NewFileSystem(client)
		fRes, err := jyFilelLib.SaveFile(gctx.New(), fr)
		log.Println(fRes, "----------", err)
	}
}

func Test_GetFile(t *testing.T) {
	jyFilelLib := filesystemclient.NewFileSystem(zrpc.MustNewClient(zrpc.RpcClientConf{
		Etcd: discov.EtcdConf{
			Key:   "moapp.filesystem.rpc",
			Hosts: []string{"192.168.3.206:2379"},
		},
	}))
	resp, err := jyFilelLib.GetOssUril(gctx.New(), &filesystem.LoadFileReq{
		Domain: Domain,
		FileId: "335a98b93b81dd3a0488a1eecac55d4a.pdf",
	})
	log.Println(resp, "-----", err)
}