Method Write

cmd/client/main.go:29–72

(p []byte)

}

func (w *clientLogWriter) Write(p []byte) (int, error) {
	raw := strings.TrimRight(string(p), "\r\n")
	if raw == "" {
		_, err := w.out.Write(p)
		return len(p), err
	}

	module := "client"
	msg := raw
	if strings.HasPrefix(raw, "[") {
		if idx := strings.Index(raw, "]"); idx > 1 {
			module = strings.ToUpper(strings.TrimSpace(raw[1:idx]))
			msg = strings.TrimSpace(raw[idx+1:])
		}
	}
	module = strings.ToUpper(module)

	level := "INFO"
	lower := strings.ToLower(msg)
	if strings.Contains(lower, "fatal") || strings.Contains(lower, "invalid") || strings.Contains(lower, "required") {
		level = "ERROR"
	} else if strings.Contains(lower, "timeout") || strings.Contains(lower, "non-ok") || strings.Contains(lower, "failed") || strings.Contains(lower, "shutting down") {
		level = "WARN"
	}

	ts := time.Now().Format("15:04:05")
	line := fmt.Sprintf("%s  %-7s %-7s %s\n", ts, module, level, msg)

	if !w.useColor {
		_, err := io.WriteString(w.out, line)
		return len(p), err
	}

	levelColor := "\x1b[36m" // cyan
	if level == "WARN" {
		levelColor = "\x1b[33m" // yellow
	}
	if level == "ERROR" {
		levelColor = "\x1b[31m" // red
	}
	colored := fmt.Sprintf("%s  \x1b[35m%-7s\x1b[0m %s%-7s\x1b[0m %s\n", ts, module, levelColor, level, msg)
	_, err := io.WriteString(w.out, colored)
	return len(p), err
}

func setupClientLogging() {
	log.SetFlags(0)