fix: concurrency

Author: devmegablaster

pkg/jatt.go

@@ -33,9 +33,6 @@ func NewJatt(cfg config.JattConfig) *Jatt {
 func (j *Jatt) Run() {
 	timeStart := time.Now()
 
-	j.wg.Add(1)
-	go j.writer.CopyStatic(&j.wg)
-
 	files := j.reader.Read()
 	renderedFiles := j.renderer.Render(files)
 
@@ -48,7 +45,12 @@ func (j *Jatt) Run() {
 
 	j.wg.Wait()
 
-	j.writer.WriteRSSFeed(feed)
+	j.wg.Add(1)
+	go j.writer.CopyStatic(&j.wg)
+	j.wg.Add(1)
+	go j.writer.WriteRSSFeed(&j.wg, feed)
+
+	j.wg.Wait()
 
 	timeEnd := time.Now()
 	fmt.Println(styles.StatsStyle.Render(fmt.Sprintf("\n⚡Took %v", timeEnd.Sub(timeStart).Round(time.Millisecond))))

pkg/writer/writer.go

@@ -112,7 +112,9 @@ func (w *Writer) CopyStatic(wg *sync.WaitGroup) error {
 	return nil
 }
 
-func (w *Writer) WriteRSSFeed(feed []byte) {
+func (w *Writer) WriteRSSFeed(wg *sync.WaitGroup, feed []byte) {
+	defer wg.Done()
+
 	outputDir := w.cfg.SiteConfig.OutputDir
 	err := os.WriteFile(outputDir+"/rss.xml", feed, 0644)
 	if err != nil {