trying_out_go/playground/countVotes.go

// from Chapter 7 of Head First Go
package main

import (
	"fmt"
	"git.linuxhg.com/Go_Training/datafile"
	"git.linuxhg.com/Go_Training/keyboard"
	"sort"
	"strings"
)

func countVotes() {
	fmt.Printf("What is the name of the file: ")
	fileString, err := keyboard.GetString()
	Error(err)
	fileString = strings.TrimSpace(fileString)
	lines, err := datafile.GetStrings(fileString)
	Error(err)

	// Count the hard way with slices
	var names []string
	var counts []int
	for _, line := range lines {
		matched := false
		for i, name := range names {
			if name == line {
				counts[i]++
				matched = true
			}
		}
		if matched == false {
			names = append(names, line)
			counts = append(counts, 1)
		}
	}

	for i, name := range names {
		fmt.Printf("%s: %d\n", name, counts[i])
	}

	fmt.Println("")

	// Easy way
	votes := make(map[string]int)
	var namesMap []string
	for _, line := range lines {
		votes[line]++
	}
	for name := range votes {
		namesMap = append(namesMap, name)
	}
	sort.Strings(namesMap)
	for _, name := range namesMap {
		fmt.Printf("%v: %v\n", name, votes[name])
	}
}