motiejus/undocker
1
Fork 0
Code Packages Releases Activity

handle stream closing better

Browse Source
This commit is contained in:
Motiejus Jakštys 2021-05-24 00:11:58 +03:00
parent 7f9d150a33
commit 8ea717e6a9
2 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
internal/cmdrootfs/cmdrootfs.go
View File

@ -40,23 +40,26 @@ func (c *Command) Execute(args []string) (err error) {
}
defer func() { err = multierr.Append(err, rd.Close()) }()
var out io.WriteCloser
outf := string(c.PositionalArgs.Outfile)
if outf == "-" {
out = os.Stdout
var out io.Writer
var outf *os.File
if fname := string(c.PositionalArgs.Outfile); fname == "-" {
outf = os.Stdout
} else {
out, err = os.Create(outf)
outf, err = os.Create(fname)
if err != nil {
return err
}
}
defer func() { err = multierr.Append(err, out.Close()) }()
out = outf
defer func() { err = multierr.Append(err, outf.Close()) }()
if c.Xz {
if out, err = xz.NewWriter(out); err != nil {
outz, err := xz.NewWriter(out)
if err != nil {
return err
}
defer func() { err = multierr.Append(err, out.Close()) }()
defer func() { err = multierr.Append(err, outz.Close()) }()
out = outz
}
if _, err := c.rootfsNew(rd).WriteTo(out); err != nil {

1
internal/cmdrootfs/cmdrootfs_test.go
View File

@ -3,5 +3,4 @@ package cmdrootfs
import "testing"
func TestExecute(t *testing.T) {
}