FilesExpand file tree

 master

/

progressreader.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

59 lines (52 loc) · 1.41 KB

 master

/

progressreader.go

Copy path

Top

File metadata and controls

• Code
• Blame

59 lines (52 loc) · 1.41 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

package progress

import (

"io"

)

// Reader is a Reader with progress bar.

type Reader struct {

in io.ReadCloser // Stream to read from

out Output // Where to send progress bar to

size int64

current int64

lastUpdate int64

id string

action string

}

// NewProgressReader creates a new ProgressReader.

func NewProgressReader(in io.ReadCloser, out Output, size int64, id, action string) *Reader {

return &Reader{

in: in,

out: out,

size: size,

id: id,

action: action,

}

}

func (p *Reader) Read(buf []byte) (n int, err error) {

read, err := p.in.Read(buf)

p.current += int64(read)

updateEvery := int64(1024 * 512) //512kB

if p.size > 0 {

// Update progress for every 1% read if 1% < 512kB

if increment := int64(0.01 * float64(p.size)); increment < updateEvery {

updateEvery = increment

}

}

if p.current-p.lastUpdate > updateEvery || err != nil {

p.updateProgress(err != nil && read == 0)

p.lastUpdate = p.current

}

return read, err

}

// Close closes the progress reader and its underlying reader.

func (p *Reader) Close() error {

if p.current < p.size {

// print a full progress bar when closing prematurely

p.current = p.size

p.updateProgress(false)

}

return p.in.Close()

}

func (p *Reader) updateProgress(last bool) {

p.out.WriteProgress(Progress{ID: p.id, Action: p.action, Current: p.current, Total: p.size, LastUpdate: last})

}