Function BenchmarkFileTransfer

datanode/bootstrap/bootstrap_server_bm_test.go:93–154

(b *testing.B)

}

func BenchmarkFileTransfer(b *testing.B) {
	testFile := "/tmp/file2m"
	fileSize := 2 * 1024 * 1024

	bigBuff := make([]byte, fileSize)
	ioutil.WriteFile(testFile, bigBuff, 0666)

	metaStore := &metaMocks.MetaStore{}
	diskStore := &diskMocks.DiskStore{}
	grpcServer := grpc.NewServer()
	peerServer := NewPeerDataNodeServer(metaStore, diskStore)
	pb.RegisterPeerDataNodeServer(grpcServer, peerServer)

	lis, err := net.Listen("tcp", fmt.Sprintf(":%d", 0))
	if err != nil {
		b.Error(err)
	}
	go grpcServer.Serve(lis)
	defer grpcServer.Stop()

	testCases := []struct {
		name       string
		chunkSize  int32
		bufferSize int32
	}{
		{"test chunk: 1024, buffer: 0", 1024, 0},
		{"test chunk: 2048, buffer: 0", 2048, 0},
		{"test chunk: 4096, buffer: 0", 4096, 0},
		{"test chunk: 8192, buffer: 0", 8192, 0},
		{"test chunk: 16384, buffer: 0", 16384, 0},
		{"test chunk: 32768, buffer: 0", 32768, 0},

		{"test chunk: 1024, buffer: 16000", 1024, 16 * 1024},
		{"test chunk: 2048, buffer: 16000", 2048, 16 * 1024},
		{"test chunk: 4096, buffer: 16000", 4096, 16 * 1024},
		{"test chunk: 8192, buffer: 16000", 8192, 16 * 1024},
		{"test chunk: 16384, buffer: 16000", 16384, 16 * 1024},
		{"test chunk: 32768, buffer: 16000", 32768, 16 * 1024},

		{"test chunk: 1024, buffer: 32000", 1024, 32 * 1024},
		{"test chunk: 2048, buffer: 32000", 2048, 32 * 1024},
		{"test chunk: 4096, buffer: 32000", 4096, 32 * 1024},
		{"test chunk: 8192, buffer: 32000", 8192, 32 * 1024},
		{"test chunk: 16384, buffer: 32000", 16384, 32 * 1024},
		{"test chunk: 32768, buffer: 32000", 32768, 32 * 1024},

		{"test chunk: 1024, buffer: 64000", 1024, 64 * 1024},
		{"test chunk: 2048, buffer: 64000", 2048, 64 * 1024},
		{"test chunk: 4096, buffer: 64000", 4096, 64 * 1024},
		{"test chunk: 8192, buffer: 64000", 8192, 64 * 1024},
		{"test chunk: 16384, buffer: 64000", 16384, 64 * 1024},
		{"test chunk: 32768, buffer: 64000", 32768, 64 * 1024},
	}

	for _, tc := range testCases {
		b.Run("test1", func(t *testing.B) {
			fileTransfer(tc.name, lis.Addr().String(), testFile, tc.chunkSize, tc.bufferSize, fileSize)
		})