stats.go

package main

import (
	"context"
	"fmt"
	"sync"
	"sync/atomic"
	"time"
)

type stats struct {
	checked atomic.Uint64
	success atomic.Uint64

	errMu  sync.RWMutex
	errors map[string]uint64

	start time.Time
}

func newStats() *stats {
	return &stats{
		errors: make(map[string]uint64),
		start:  time.Now(),
	}
}

func (s *stats) incChecked() { s.checked.Add(1) }
func (s *stats) incSuccess() { s.success.Add(1) }

func (s *stats) incErr(cat string) {
	s.errMu.Lock()
	s.errors[cat]++
	s.errMu.Unlock()
}

func (s *stats) totalErrors() uint64 {
	s.errMu.RLock()
	defer s.errMu.RUnlock()
	var t uint64
	for _, v := range s.errors {
		t += v
	}
	return t
}

func (s *stats) run(ctx context.Context) {
	tick := time.NewTicker(time.Second)
	defer tick.Stop()

	for {
		select {
		case <-ctx.Done():
			return
		case <-tick.C:
			s.print()
		}
	}
}

func (s *stats) print() {
	checked := s.checked.Load()
	secs := time.Since(s.start).Seconds()
	if secs < 1 {
		secs = 1
	}

	fmt.Printf("\r\033[K\033[90m[\033[0m checked \033[36m%d\033[0m \033[90m/\033[0m valid \033[32m%d\033[0m \033[90m/\033[0m bad \033[31m%d\033[0m \033[90m/\033[0m \033[33m%.0f/s\033[0m \033[90m]\033[0m",
		checked,
		s.success.Load(),
		s.totalErrors(),
		float64(checked)/secs,
	)
}

func (s *stats) printFinal() {
	checked := s.checked.Load()
	success := s.success.Load()
	failed := s.totalErrors()
	dur := time.Since(s.start).Round(time.Second)

	var rate float64
	if checked > 0 {
		rate = float64(success) / float64(checked) * 100
	}

	fmt.Printf("\n\n\033[90m─────────────────────────────────────\033[0m\n")
	fmt.Printf("  time     \033[36m%s\033[0m\n", dur)
	fmt.Printf("  checked  \033[36m%d\033[0m\n", checked)
	fmt.Printf("  valid    \033[32m%d\033[0m \033[90m(%.1f%%)\033[0m\n", success, rate)
	fmt.Printf("  bad      \033[31m%d\033[0m\n", failed)
	fmt.Printf("\033[90m─────────────────────────────────────\033[0m\n")
}