Skip to content

Commit ab6d5b4

Browse files
committed
refactor: add write file helper
1 parent e5e7ceb commit ab6d5b4

File tree

1 file changed

+20
-14
lines changed

1 file changed

+20
-14
lines changed

main.go

+20-14
Original file line numberDiff line numberDiff line change
@@ -17,25 +17,31 @@ func NewArchiver(archive *os.File) *Archiver {
1717

1818
func (a *Archiver) Archive(files ...*os.File) error {
1919
for _, file := range files {
20-
info, err := file.Stat()
21-
if err != nil {
22-
return err
23-
}
24-
25-
writer, err := a.w.Create(info.Name())
26-
if err != nil {
27-
return err
28-
}
29-
30-
_, err = io.Copy(writer, file)
31-
if err != nil {
32-
return err
33-
}
20+
a.writeFile(file)
3421
}
3522

3623
a.w.Close()
3724
return nil
3825
}
3926

27+
func (a *Archiver) writeFile(file *os.File) error {
28+
info, err := file.Stat()
29+
if err != nil {
30+
return err
31+
}
32+
33+
writer, err := a.w.Create(info.Name())
34+
if err != nil {
35+
return err
36+
}
37+
38+
_, err = io.Copy(writer, file)
39+
if err != nil {
40+
return err
41+
}
42+
43+
return nil
44+
}
45+
4046
func main() {
4147
}

0 commit comments

Comments
 (0)