word_count.go

package main

import (
	"golang.org/x/tour/wc"
	"strings"
)

func WordCount(s string) map[string]int {
	countForWord := make(map[string]int)

	for _, word := range strings.Fields(s) {
		var currentCount int

		if count, set := countForWord[word]; set {
			currentCount = count
		} else {
			currentCount = 0
		}

		countForWord[word] = currentCount + 1
	}

	return countForWord
}

func main() {
	wc.Test(WordCount)
}