jyfs/rpc/test/jyfs_rpc_test.go

package test

import (
	"context"
	"log"
	"testing"
	"time"

	"app.yhyue.com/moapp/jyfs/rpc/filesystem"
	"app.yhyue.com/moapp/jyfs/rpc/filesystemclient"
	"app.yhyue.com/moapp/jyfs/rpc/internal/config"

	"github.com/tal-tech/go-zero/core/conf"
	"github.com/tal-tech/go-zero/zrpc"
)

var C config.Config

func init() {
	conf.MustLoad("./filesystem.yaml", &C)
}

// 创建域
func Test_CreateDomain(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := filesystemclient.NewFileSystem(zrpc.MustNewClient(C.FileSystemConf))
	req := &filesystem.DomainReq{Name: "xzh", MetaFields: []string{"docName", "docSuffix", "docSize"}}
	res, err := FileSystem.CreateDomain(ctx, req)
	log.Println("err ", err)
	log.Println("req ", res)
}

// 删除域
func Test_DeleteDomain(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := filesystemclient.NewFileSystem(zrpc.MustNewClient(C.FileSystemConf))
	req := &filesystem.DomainReq{Name: "xzh", MetaFields: []string{"docName", "docSuffix", "docSize"}}
	res, err := FileSystem.DeleteDomain(ctx, req)
	log.Println("err ", err)
	log.Println("req ", res)
}

// 删除文件
func Test_DeleteFile(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := filesystemclient.NewFileSystem(zrpc.MustNewClient(C.FileSystemConf))
	req := &filesystem.LoadFileReq{Domain: "xzh", FileId: "que.txt"}
	res, err := FileSystem.DeleteFile(ctx, req)
	log.Println("err ", err)
	log.Println("req ", res)
}

// 获取文件
func Test_GetFile(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := filesystemclient.NewFileSystem(zrpc.MustNewClient(C.FileSystemConf))
	req := &filesystem.LoadFileReq{Domain: "xzh", FileId: "que.txt"}
	res, err := FileSystem.GetFile(ctx, req)
	log.Println("err ", err)
	log.Println("req ", res)
}

// 获取文件元数据
func Test_GetFileMeta(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := filesystemclient.NewFileSystem(zrpc.MustNewClient(C.FileSystemConf))
	req := &filesystem.LoadFileReq{Domain: "xzh", FileId: "que.txt"}
	res, err := FileSystem.GetFileMeta(ctx, req)
	log.Println("err ", err)
	log.Println("req ", res)
}

// 保存文件
func Test_SaveFile(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := filesystemclient.NewFileSystem(zrpc.MustNewClient(C.FileSystemConf))
	file, err := os.Open("./111.txt")
	if err != nil {
		return nil, err
	}
	defer file.Close()
	// FileInfo:
	stats, err := file.Stat()
	if err != nil {
		return nil, err
	}
	// []byte
	data := make([]byte, stats.Size())
	count, err := file.Read(data)
	if err != nil {
		return nil, err
	}
	fmt.Println(count)
	req := &filesystem.SaveFileReq{
		Domain:         "xzh",
		FileId:         "111.txt",
		Meta:           map[string]string{"docName": "111", "docSuffix": ".txt", "docSize": "1024"},
		RawFileContent: data,
	}
	res, err := FileSystem.SaveFile(ctx, req)
	log.Println("err ", err)
	log.Println("req ", res)
}

// 更新文件元数据
func Test_UpdateFileMeta(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := filesystemclient.NewFileSystem(zrpc.MustNewClient(C.FileSystemConf))
	req := &filesystem.UpdateFileMetaReq{
		Domain: "xzh",
		FileId: "que.txt",
		Meta:   map[string]string{"docName": "que", "docSuffix": ".txt", "docSize": "2048"},
	}
	res, err := FileSystem.UpdateFileMeta(ctx, req)
	log.Println("err ", err)
	log.Println("req ", res)
}