FilesExpand file tree

 master

/

progressreader.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

50 lines (46 loc) · 1.46 KB

 master

/

progressreader.go

Copy path

Top

File metadata and controls

• Code
• Blame

50 lines (46 loc) · 1.46 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

package progressreader

import (

"io"

"github.com/docker/docker/pkg/jsonmessage"

"github.com/docker/docker/pkg/streamformatter"

)

// Reader with progress bar

type Config struct {

In io.ReadCloser // Stream to read from

Out io.Writer // Where to send progress bar to

Formatter *streamformatter.StreamFormatter

Size int

Current int

LastUpdate int

NewLines bool

ID string

Action string

}

func New(newReader Config) *Config {

return &newReader

}

func (config *Config) Read(p []byte) (n int, err error) {

read, err := config.In.Read(p)

config.Current += read

updateEvery := 1024 * 512 //512kB

if config.Size > 0 {

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

if increment := int(0.01 * float64(config.Size)); increment < updateEvery {

updateEvery = increment

}

}

if config.Current-config.LastUpdate > updateEvery || err != nil {

config.Out.Write(config.Formatter.FormatProgress(config.ID, config.Action, &jsonmessage.JSONProgress{Current: config.Current, Total: config.Size}))

config.LastUpdate = config.Current

}

// Send newline when complete

if config.NewLines && err != nil && read == 0 {

config.Out.Write(config.Formatter.FormatStatus("", ""))

}

return read, err

}

func (config *Config) Close() error {

config.Current = config.Size

config.Out.Write(config.Formatter.FormatProgress(config.ID, config.Action, &jsonmessage.JSONProgress{Current: config.Current, Total: config.Size}))

return config.In.Close()

}