⚡ fix: optimize message rendering in chat component

Refactor the message rendering logic in the chat component to utilize goroutines for parallel processing. This improves performance by reducing rendering time for large message sets. Also, update the MarkdownRenderer to use a sync.Map for caching, improving access efficiency and thread safety.

Author: nathfavour

cmd/vibeaura/chat.go

@@ -1191,22 +1191,35 @@ func (m *model) renderMessages() string {
 	// Sync renderer width with viewport
 	m.md.SetWidth(m.viewport.Width)
 
-	var sb strings.Builder
+	// Use goroutines to render messages in parallel
+	rendered := make([]string, len(m.messages))
+	var wg sync.WaitGroup
+
 	for i, msg := range m.messages {
-		content := msg
-		// If it's an AI message (starts with the VibeAuracle label), render markdown for the content part
-		if strings.HasPrefix(msg, aiStyle.Render("VibeAuracle: ")) {
-			rawContent := strings.TrimPrefix(msg, aiStyle.Render("VibeAuracle: "))
-			// Only render markdown if it's not currently streaming (too slow/flickery)
-			if !strings.HasSuffix(rawContent, subtleStyle.Render("▌")) {
-				content = aiStyle.Render("VibeAuracle:") + "\n" + m.md.Render(rawContent)
+		wg.Add(1)
+		go func(idx int, raw string) {
+			defer wg.Done()
+			
+			content := raw
+			// If it's an AI message, render markdown
+			if strings.HasPrefix(raw, aiStyle.Render("VibeAuracle: ")) {
+				rawContent := strings.TrimPrefix(raw, aiStyle.Render("VibeAuracle: "))
+				// Only render markdown if it's not currently streaming
+				if !strings.HasSuffix(rawContent, subtleStyle.Render("▌")) {
+					content = aiStyle.Render("VibeAuracle:") + "\n" + m.md.Render(rawContent)
+				}
 			}
-		}
 
-		// Use lipgloss to wrap the message to the viewport width precisely.
-		wrapped := lipgloss.NewStyle().Width(m.viewport.Width).Render(content)
-		sb.WriteString(wrapped)
-		if i < len(m.messages)-1 {
+			// Wrap the message
+			rendered[idx] = lipgloss.NewStyle().Width(m.viewport.Width).Render(content)
+		}(i, msg)
+	}
+	wg.Wait()
+
+	var sb strings.Builder
+	for i, r := range rendered {
+		sb.WriteString(r)
+		if i < len(rendered)-1 {
 			sb.WriteString("\n\n")
 		}
 	}

internal/reactor/markdown.go

@@ -7,47 +7,48 @@ import (
 )
 
 type MarkdownRenderer struct {
-	renderer *glamour.TermRenderer
-	cache    map[string]string
-	width    int
-	mu       sync.RWMutex
+	cache sync.Map
+	width int
+	mu    sync.Mutex // Protects width and pool recreation
+	pool  *sync.Pool
 }
 
 func NewMarkdownRenderer(width int) *MarkdownRenderer {
-	r, _ := glamour.NewTermRenderer(
-		glamour.WithAutoStyle(),
-		glamour.WithWordWrap(width-4),
-	)
-	return &MarkdownRenderer{
-		renderer: r,
-		cache:    make(map[string]string),
-		width:    width,
+	mr := &MarkdownRenderer{
+		width: width,
 	}
+	mr.recreatePool(width)
+	return mr
 }
 
-func (m *MarkdownRenderer) Render(content string) string {
-	m.mu.RLock()
-	// Quick check if width changed (invalidate cache if so)
-	if cached, ok := m.cache[content]; ok {
-		m.mu.RUnlock()
-		return cached
+func (m *MarkdownRenderer) recreatePool(width int) {
+	m.pool = &sync.Pool{
+		New: func() interface{} {
+			r, _ := glamour.NewTermRenderer(
+				glamour.WithAutoStyle(),
+				glamour.WithWordWrap(width-4),
+			)
+			return r
+		},
 	}
-	m.mu.RUnlock()
-
-	m.mu.Lock()
-	defer m.mu.Unlock()
+}
 
-	// Double check cache after lock
-	if cached, ok := m.cache[content]; ok {
-		return cached
+func (m *MarkdownRenderer) Render(content string) string {
+	// 1. Concurrent-safe cache check
+	if cached, ok := m.cache.Load(content); ok {
+		return cached.(string)
 	}
 
-	rendered, err := m.renderer.Render(content)
+	// 2. Get a renderer from the pool
+	r := m.pool.Get().(*glamour.TermRenderer)
+	defer m.pool.Put(r)
+
+	rendered, err := r.Render(content)
 	if err != nil {
 		return content
 	}
 
-	m.cache[content] = rendered
+	m.cache.Store(content, rendered)
 	return rendered
 }
 
@@ -58,9 +59,8 @@ func (m *MarkdownRenderer) SetWidth(width int) {
 		return
 	}
 	m.width = width
-	m.cache = make(map[string]string)
-	m.renderer, _ = glamour.NewTermRenderer(
-		glamour.WithAutoStyle(),
-		glamour.WithWordWrap(width-4),
-	)
+	// Invalidate cache and pool for new width
+	m.cache = sync.Map{}
+	m.recreatePool(width)
 }
+