Go语言并发之道学习七 缓冲单元测试

缓冲单元测试

go test 命令,会自动读取源码目录下面名为 *_test.go 的文件,生成并运行测试用的可执行文件。

新建一个 bufio_test.go 的文件:

package main

import (
	"testing"
	"fmt"
	"os"
	"io"
	"io/ioutil"
	"bufio"
)

/*
测试缓冲和未缓冲的写入速度
缓冲写入比未缓冲写入更快。
这是因为在bufio.Writer中,写入在内部的数据,排队到缓冲区中,知道已经积累了足够的块为止,然后块被写出。
这个过程通常称为分块。
*/

//未缓冲
func BenchmarkUnbufferedWrited(b * testing.B){
	performWrite(b,tmpFileOrFatal())
}

//缓冲
func BenchmarkBufferedWrite(b *testing.B){
	bufferedFile := bufio.NewWriter(tmpFileOrFatal())
	performWrite(b,bufio.NewWriter(bufferedFile))
}

func tmpFileOrFatal() *os.File{
	file,err := ioutil.TempFile("","tmp")
	if err != nil{
		fmt.Printf("error: %v\n",err)
	}
	return file
}

func performWrite(b *testing.B,writer io.Writer){
	repeat := func(
		done <-chan interface{},
		vals ...interface{},
		) <-chan interface {}{
		repeatStream := make(chan interface{})
		go func() {
			defer close(repeatStream)
			for{
				for val := range vals{
					select{
					case <- done:
						return
					case repeatStream<-uint8(val): //将val 的byte类型转换成 uint8 类型
					}
				}
			}
		}()
		return repeatStream
	}
	take := func(
		done <-chan interface{},
		valueStream <-chan interface{},
		num int,
		)<-chan interface{}{
		takeStream := make(chan interface{})
		go func() {
			defer close(takeStream)
			for i:=num;i>0||i!=-1;{
				if i != -1{
					i--
				}
				select {
				case <-done:
					return
				case takeStream<-<-valueStream:
				}
			}
		}()
		return takeStream
	}

	done := make(chan interface{})
	defer close(done)

	b.ResetTimer()
	for bt := range take(done,repeat(done,byte(0)),b.N){
		writer.Write([]byte{bt.(byte)})
	}
}

运行测试:

[root@izj6c4jirdug8kh3uo6rdez goroutine]# go test -bench=. bufio_test.go
goos: linux
goarch: amd64
BenchmarkUnbufferedWrited 	  500000	      2938 ns/op
BenchmarkBufferedWrite    	 2000000	       847 ns/op
PASS
ok  	command-line-arguments	4.979s

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值