示例#1
0
文件: main.go 项目: aybabtme/pv
func main() {
	log.SetPrefix("pv: ")
	log.Printf("streaming!")

	start := time.Now()
	mrw := iocontrol.NewMeasuredReader(os.Stdin)
	pw, pr, sample := iocontrol.Profile(os.Stdout, mrw)

	go func() {
		for range time.NewTicker(5 * time.Second).C {
			profile := sample()
			log.Printf("%s at %sps (read=%v write=%v total=%v)",
				humanize.IBytes(uint64(mrw.Total())),
				humanize.IBytes(mrw.BytesPerSec()),
				profile.WaitRead,
				profile.WaitWrite,
				profile.Total,
			)
		}
	}()

	n, err := io.Copy(pw, spark.ReaderOut(pr, os.Stderr))
	if err != nil {
		log.Fatalf("error after %s: %v", humanize.IBytes(uint64(n)), err)
	}
	log.Printf("done after %s in %v", humanize.IBytes(uint64(n)), time.Since(start))
}
示例#2
0
func TestParse(t *testing.T) {
	testfile := os.Getenv("TESTFILE")
	if testfile == "" {
		t.Skip("No testfile specified. Please set `TESTFILE` environment variable with the file path.")
	}
	f, err := os.Open(testfile)
	ensure.Nil(t, err)
	defer f.Close()
	mr := iocontrol.NewMeasuredReader(f)

	rdr := newMockOSMReader()
	d := NewDecoder(mr)
	err = d.Parse(rdr)
	fmt.Printf("Speed: %v/s, total read: %v\n", humanize.Bytes(mr.BytesPerSec()), humanize.Bytes(uint64(mr.Total())))
	fmt.Printf("Read %v nodes, %v ways, %v relations\n", atomic.LoadUint64(rdr.Nodes), atomic.LoadUint64(rdr.Ways), atomic.LoadUint64(rdr.Relations))
	ensure.Nil(t, err)
}
示例#3
0
func Example_ThrottledReader() {

	totalSize := 10 * iocontrol.KiB
	readPerSec := 100 * iocontrol.KiB
	maxBurst := 10 * time.Millisecond

	input := randBytes(totalSize)
	src := bytes.NewReader(input)
	measured := iocontrol.NewMeasuredReader(src)
	throttled := iocontrol.ThrottledReader(measured, readPerSec, maxBurst)

	done := make(chan []byte)
	go func() {
		start := time.Now()
		output, err := ioutil.ReadAll(throttled)
		fmt.Printf("done in %.1fs", time.Since(start).Seconds())
		if err != nil {
			log.Fatalf("error reading: %v", err)
		}
		done <- output
	}()

	for {
		select {
		case <-time.Tick(time.Millisecond * 10):
			log.Printf("reading at %s/s", humanize.IBytes(measured.BytesPerSec()))

		case output := <-done:
			if !bytes.Equal(input, output) {
				log.Print("==== input ====\n", hex.Dump(input))
				log.Print("==== output ====\n", hex.Dump(output))
				log.Fatalf("mismatch between input and output")
			}
			return
		}
	}

	// Output:
	// done in 0.1s
}