refactor(logformat): replace parser switch with registry

1 files changed, 72 insertions, 25 deletions

diff --git a/internal/mapr/logformat/parser.go b/internal/mapr/logformat/parser.go
index 37d7a63..b6ed87d 100644
--- a/internal/mapr/logformat/parser.go
+++ b/internal/mapr/logformat/parser.go
@@ -3,6 +3,8 @@ package logformat
 import (
 	"errors"
 	"fmt"
+	"strings"
+	"sync"
 	"time"
 
 	"github.com/mimecast/dtail/internal/config"
@@ -18,6 +20,62 @@ type Parser interface {
 	MakeFields(string) (map[string]string, error)
 }
 
+// ParserFactory builds a Parser for a specific log format.
+type ParserFactory func(hostname, timeZoneName string, timeZoneOffset int) (Parser, error)
+
+var parserFactories = make(map[string]ParserFactory)
+var parserFactoriesMu sync.RWMutex
+
+func init() {
+	registerBuiltInParsers()
+}
+
+// RegisterParser registers or replaces a parser factory for a log format name.
+func RegisterParser(logFormatName string, factory ParserFactory) error {
+	name := strings.TrimSpace(logFormatName)
+	if name == "" {
+		return errors.New("log format name cannot be empty")
+	}
+	if factory == nil {
+		return errors.New("parser factory cannot be nil")
+	}
+
+	parserFactoriesMu.Lock()
+	defer parserFactoriesMu.Unlock()
+	parserFactories[name] = factory
+	return nil
+}
+
+func getParserFactory(logFormatName string) (ParserFactory, bool) {
+	parserFactoriesMu.RLock()
+	defer parserFactoriesMu.RUnlock()
+	factory, found := parserFactories[logFormatName]
+	return factory, found
+}
+
+func registerBuiltInParsers() {
+	mustRegisterParser("generic", wrapParserFactory(newGenericParser))
+	mustRegisterParser("generickv", wrapParserFactory(newGenericKVParser))
+	mustRegisterParser("csv", wrapParserFactory(newCSVParser))
+	mustRegisterParser("mimecast", wrapParserFactory(newMimecastParser))
+	mustRegisterParser("mimecastgeneric", wrapParserFactory(newMimecastGenericParser))
+	mustRegisterParser("default", wrapParserFactory(newDefaultParser))
+	mustRegisterParser("custom1", wrapParserFactory(newCustom1Parser))
+	mustRegisterParser("custom2", wrapParserFactory(newCustom2Parser))
+}
+
+func mustRegisterParser(logFormatName string, factory ParserFactory) {
+	if err := RegisterParser(logFormatName, factory); err != nil {
+		panic(err)
+	}
+}
+
+func wrapParserFactory[T Parser](factory func(string, string, int) (T, error)) ParserFactory {
+	return func(hostname, timeZoneName string, timeZoneOffset int) (Parser, error) {
+		return factory(hostname, timeZoneName, timeZoneOffset)
+	}
+}
+
 // NewParser returns a new log parser.
 func NewParser(logFormatName string, query *mapr.Query) (Parser, error) {
 	hostname, err := config.Hostname()
@@ -27,30 +85,19 @@ func NewParser(logFormatName string, query *mapr.Query) (Parser, error) {
 	now := time.Now()
 	timeZoneName, timeZoneOffset := now.Zone()
 
-	// Extend this for adding more log formats!
-	switch logFormatName {
-	case "generic":
-		return newGenericParser(hostname, timeZoneName, timeZoneOffset)
-	case "generickv":
-		return newGenericKVParser(hostname, timeZoneName, timeZoneOffset)
-	case "csv":
-		return newCSVParser(hostname, timeZoneName, timeZoneOffset)
-	case "mimecast":
-		return newMimecastParser(hostname, timeZoneName, timeZoneOffset)
-	case "mimecastgeneric":
-		return newMimecastGenericParser(hostname, timeZoneName, timeZoneOffset)
-	case "default":
-		return newDefaultParser(hostname, timeZoneName, timeZoneOffset)
-	case "custom1":
-		return newCustom1Parser(hostname, timeZoneName, timeZoneOffset)
-	case "custom2":
-		return newCustom2Parser(hostname, timeZoneName, timeZoneOffset)
-	default:
-		p, err := newDefaultParser(hostname, timeZoneName, timeZoneOffset)
-		if err != nil {
-			return p, fmt.Errorf("No '%s' mapr log format and problem creating default one: %v",
-				logFormatName, err)
-		}
-		return p, fmt.Errorf("No '%s' mapr log format", logFormatName)
+	if parserFactory, found := getParserFactory(logFormatName); found {
+		return parserFactory(hostname, timeZoneName, timeZoneOffset)
+	}
+
+	defaultFactory, found := getParserFactory("default")
+	if !found {
+		return nil, fmt.Errorf("No '%s' mapr log format and no default parser registered", logFormatName)
+	}
+
+	p, err := defaultFactory(hostname, timeZoneName, timeZoneOffset)
+	if err != nil {
+		return p, fmt.Errorf("No '%s' mapr log format and problem creating default one: %v",
+			logFormatName, err)
 	}
+	return p, fmt.Errorf("No '%s' mapr log format", logFormatName)
 }