FilesExpand file tree

 master

/

progressreader.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

64 lines (56 loc) · 1.55 KB

 master

/

progressreader.go

Copy path

Top

File metadata and controls

• Code
• Blame

64 lines (56 loc) · 1.55 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

60

61

62

63

64

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 {

updateProgress(config)

config.LastUpdate = config.Current

}

if err != nil && read == 0 {

updateProgress(config)

if config.NewLines {

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

}

}

return read, err

}

func (config *Config) Close() error {

if config.Current < config.Size {

//print a full progress bar when closing prematurely

config.Current = config.Size

updateProgress(config)

}

return config.In.Close()

}

func updateProgress(config *Config) {

progress := jsonmessage.JSONProgress{Current: config.Current, Total: config.Size}

fmtMessage := config.Formatter.FormatProgress(config.ID, config.Action, &progress)

config.Out.Write(fmtMessage)

}