Added totals feature that shows log size and record count.

-t or --totals

Author: Timthreetwelve

cmd/printlog.go

@@ -22,8 +22,11 @@ THE SOFTWARE.
 package cmd
 
 import (
+	"log"
+
 	"github.com/fatih/color"
 	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
 	"github.com/timthreetwelve/bootlogger/utils"
 )
 
@@ -44,7 +47,12 @@ var printlogCmd = &cobra.Command{
 }
 
 func init() {
-	rootCmd.AddCommand(printlogCmd)
-	// No flags are needed for this command, it just prints the log file
 	// Output may be large, but can be piped through 'more'
+	rootCmd.AddCommand(printlogCmd)
+
+	// Boolean flag to print totals after the log entries
+	printlogCmd.Flags().BoolP("totals", "t", false, "Show log file size and total lines after printing log")
+	if err := viper.BindPFlag("totals", printlogCmd.Flags().Lookup("totals")); err != nil {
+		log.Printf("Error binding flag 'totals': %v", err)
+	}
 }

utils/printlog.go

@@ -61,9 +61,16 @@ func PrintLog() error {
 	defer file.Close()
 
 	// Using bufio scanner here for memory-efficient line-by-line reading.
+	lineCounter := 0
 	scanner := bufio.NewScanner(file)
 	for scanner.Scan() {
 		fmt.Println(scanner.Text())
+		lineCounter++
+	}
+
+	// Print totals if flag is set
+	if viper.GetBool("totals") {
+		fmt.Printf("\n%s is %s and contains %d records.", logFile, formatFileSize(stat.Size()), lineCounter)
 	}
 
 	// Check for errors during scanning
@@ -73,3 +80,26 @@ func PrintLog() error {
 	}
 	return nil
 }
+
+// Formats a size given in bytes to kb, mb, etc.
+func formatFileSize(size int64) string {
+	const (
+		KB = 1024
+		MB = KB * 1024
+		GB = MB * 1024
+		TB = GB * 1024
+	)
+
+	switch {
+	case size >= TB:
+		return fmt.Sprintf("%.2f TB", float64(size)/float64(TB))
+	case size >= GB:
+		return fmt.Sprintf("%.2f GB", float64(size)/float64(GB))
+	case size >= MB:
+		return fmt.Sprintf("%.2f MB", float64(size)/float64(MB))
+	case size >= KB:
+		return fmt.Sprintf("%.2f KB", float64(size)/float64(KB))
+	default:
+		return fmt.Sprintf("%d bytes", size)
+	}
+}