From 2e84d8a9b871d16390e323a857c84c0ac49a3ee5 Mon Sep 17 00:00:00 2001
From: PrivateGER <privateger@privateger.me>
Date: Sun, 17 Oct 2021 01:58:23 +0200
Subject: [PATCH] Add goroutines for asynchronous IO

---
 DirectoryIndexers/DirectoryIndexer.go | 109 +++++++++++++++-----------
 1 file changed, 64 insertions(+), 45 deletions(-)

diff --git a/DirectoryIndexers/DirectoryIndexer.go b/DirectoryIndexers/DirectoryIndexer.go
index b9cc937..fa8283e 100644
--- a/DirectoryIndexers/DirectoryIndexer.go
+++ b/DirectoryIndexers/DirectoryIndexer.go
@@ -39,55 +39,74 @@ func Index(path string, results chan FileList) {
 	}
 
 	FL.Files = make(map[string]VideoFile)
+	var wg sync.WaitGroup
+
 	for _, video := range fileList {
-		extension := filepath.Ext(video.Name())[1:]
-		// check if extension is one of valid yt-dlp extensions, if not ignore file
-		switch extension {
-		case "3gp",
-			"aac",
-			"flv",
-			"m4a",
-			"mp3",
-			"mp4",
-			"webm":
+		wg.Add(1)
+		go func(video os.FileInfo) {
+
+			extension := filepath.Ext(video.Name())[1:]
+			// check if extension is one of valid yt-dlp extensions, if not ignore file
+			switch extension {
+			case "3gp",
+				"aac",
+				"flv",
+				"m4a",
+				"mp3",
+				"mp4",
+				"webm":
 				break
-		default:
-			continue
-		}
-
-		id := filenameToID(video.Name())
-
-		videoObject := VideoFile{
-			Filename:  video.Name(),
-			Extension: extension,
-			Title:     filenameToTitle(video.Name(), extension),
-			Id:        id,
-		}
-
-		metadataFilename := strings.TrimSuffix(video.Name(), filepath.Ext(video.Name())) + ".info.json"
-		jsonMetadataFile, err := os.Open(path + metadataFilename)
-		if err != nil {
-			FL.Files[id] = videoObject
-			continue
-		}
-		metadataBytes, _ := ioutil.ReadAll(jsonMetadataFile)
-		_ = jsonMetadataFile.Close()
-
-		metadata, err := ParseMetadata(metadataBytes)
-		if err != nil {
-			FL.Files[id] = videoObject
-			continue
-		}
-
-		FL.Files[id] = VideoFile{
-			Filename:  video.Name(),
-			Extension: extension,
-			Title:     filenameToTitle(video.Name(), extension),
-			Id:        id,
-			Metadata:  metadata,
-		}
+			default:
+				wg.Done()
+				return
+			}
+
+			id := filenameToID(video.Name())
+
+			videoObject := VideoFile{
+				Filename:  video.Name(),
+				Extension: extension,
+				Title:     filenameToTitle(video.Name(), extension),
+				Id:        id,
+			}
+
+			metadataFilename := strings.TrimSuffix(video.Name(), filepath.Ext(video.Name())) + ".info.json"
+			jsonMetadataFile, err := os.Open(path + metadataFilename)
+			if err != nil {
+				FL.Lock()
+				FL.Files[id] = videoObject
+				FL.Unlock()
+				wg.Done()
+				return
+			}
+			metadataBytes, _ := ioutil.ReadAll(jsonMetadataFile)
+			_ = jsonMetadataFile.Close()
+
+			metadata, err := ParseMetadata(metadataBytes)
+			if err != nil {
+				FL.Lock()
+				FL.Files[id] = videoObject
+				FL.Unlock()
+				wg.Done()
+				return
+			}
+
+			FL.Lock()
+			FL.Files[id] = VideoFile{
+				Filename:  video.Name(),
+				Extension: extension,
+				Title:     filenameToTitle(video.Name(), extension),
+				Id:        id,
+				Metadata:  metadata,
+			}
+			FL.Unlock()
+
+			wg.Done()
+		}(video)
 	}
 
+	wg.Wait()
+
 	results <- FL
 	close(results)