log.go

@@ -208,37 +208,27 @@ func logMessage(level StoneLevel, format string, args ...interface{}) {
 	}
 }
 
-// getRandomPrefix returns a random prefix based on the provided StoneLevel.
+// getRandomPrefix returns a random prefix string based on the given StoneLevel.
-// It selects a prefix from predefined lists corresponding to the log level.
+// It selects from predefined lists corresponding to the severity of the level.
 func getRandomPrefix(level StoneLevel) string {
-	switch level {
+	// Map StoneLevel to their respective prefix slices
-	case TRACE:
+	prefixMap := map[StoneLevel][]string{
-		idx := rand.Intn(len(stoneTracePrefixes))
+		TRACE: stoneTracePrefixes,
-		return stoneTracePrefixes[idx]
+		DEBUG: stoneDebugPrefixes,
-
+		INFO:  stoneInfoPrefixes,
-	case DEBUG:
+		WARN:  stoneWarnPrefixes,
-		idx := rand.Intn(len(stoneDebugPrefixes))
+		ERROR: stoneErrorPrefixes,
-		return stoneDebugPrefixes[idx]
+		FATAL: stonePanicPrefixes,
-
-	case INFO:
-		idx := rand.Intn(len(stoneInfoPrefixes))
-		return stoneInfoPrefixes[idx]
-
-	case WARN:
-		idx := rand.Intn(len(stoneWarnPrefixes))
-		return stoneWarnPrefixes[idx]
-	case ERROR:
-		idx := rand.Intn(len(stoneErrorPrefixes))
-		return stoneErrorPrefixes[idx]
-
-	case FATAL:
-		idx := rand.Intn(len(stonePanicPrefixes))
-		return stonePanicPrefixes[idx]
-
-	default:
-		idx := rand.Intn(len(stoneInfoPrefixes))
-		return stoneInfoPrefixes[idx]
 	}
+
+	// Default to INFO prefixes if level is not in the map
+	prefixes, exists := prefixMap[level]
+	if !exists {
+		prefixes = stoneInfoPrefixes
+	}
+
+	// Select and return a random prefix
+	return prefixes[rand.Intn(len(prefixes))]
 }
 
 // getRandomQuote returns a random character quote based on success or failure.