alex/goldsmith
1
Fork 0
Code Releases Activity

Bugfixes

Browse Source
This commit is contained in:
Alex Yatskov 2021-04-25 09:29:53 -07:00
parent 3155de7542
commit 14c9c2065f
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
goldsmith.go
View File

@ -84,14 +84,20 @@ func (goldsmith *Goldsmith) FilterPop() *Goldsmith {
func (goldsmith *Goldsmith) End(targetDir string) []error { func (goldsmith *Goldsmith) End(targetDir string) []error {
goldsmith.targetDir = targetDir goldsmith.targetDir = targetDir
var wg sync.WaitGroup
goldsmith.Chain(&saver{ goldsmith.Chain(&saver{
clean: goldsmith.clean, clean: goldsmith.clean,
wg: &wg,
}) })
wg.Add(1)
for _, context := range goldsmith.contexts { for _, context := range goldsmith.contexts {
go context.step() go context.step()
} }
wg.Wait()
return goldsmith.errors return goldsmith.errors
} }

4
saver.go
View File

@ -3,11 +3,13 @@ package goldsmith
import ( import (
"os" "os"
"path/filepath" "path/filepath"
"sync"
) )
type saver struct { type saver struct {
clean bool clean bool
tokens map[string]bool tokens map[string]bool
wg *sync.WaitGroup
} }
func (*saver) Name() string { func (*saver) Name() string {
@ -29,6 +31,8 @@ func (saver *saver) Process(context *Context, file *File) error {
} }
func (saver *saver) Finalize(context *Context) error { func (saver *saver) Finalize(context *Context) error {
defer saver.wg.Done()
if !saver.clean { if !saver.clean {
return nil return nil
} }