Refactor log import function and improve error handling

logs_import/importer/importer.go

@@ -10,31 +10,36 @@ import (
 	"github.com/spf13/viper"
 )
 
-func ImportLogs() {
-	entries, err := parser.ReadLogEntries()
-	if err != nil {
+func Import() {
+	if err := importLogs(); err != nil {
 		log.Println(err)
 		os.Exit(1)
 	}
 
+	fmt.Println("\033[32mLogs successfully imported!\033[0m")
+	os.Exit(0)
+}
+
+func importLogs() error {
+	entries, err := parser.ReadLogEntries()
+	if err != nil {
+		return err
+	}
+
 	db, err := db.NewDBHandler()
 	if err != nil {
-		log.Println(err)
-		os.Exit(1)
+		return err
 	}
 	defer db.Close()
 
 	for _, entry := range entries {
 		if !isBlocked(entry.RequestPath) {
 			if err := db.Add(entry); err != nil {
-				log.Println(err)
-				os.Exit(1)
+				return err
 			}
 		}
 	}
-
-	fmt.Println("\033[32mLogs successfully imported!\033[0m")
-	os.Exit(0)
+	return nil
 }
 
 func isBlocked(host string) bool {

logs_import/main_test.go → logs_import/importer/importer_test.go

@@ -1,4 +1,4 @@
-package main
+package importer
 
 import (
 	"testing"

logs_import/main.go

@@ -18,7 +18,7 @@ func main() {
 
 	fmt.Println("Blocklist:", viper.GetStringSlice("blocklist"))
 
-	go importer.ImportLogs()
+	go importer.Import()
 
 	for {
 		time.Sleep(1 * time.Second)