Improved word count

file-wordcount/main.go

@@ -2,19 +2,18 @@ package main
 
 import "fmt"
 import "io"
-import "io/ioutil"
 import "log"
 import "os"
+import "bufio"
 import "strings"
 
 func wc(r io.Reader) map[string]uint {
 	words := make(map[string]uint)
-	if text, err := ioutil.ReadAll(r); err != nil {
+	s := bufio.NewScanner(r)
-		log.Fatalln(err)
+	s.Split(bufio.ScanWords)
-	} else {
+	for s.Scan() {
-		for _, word := range strings.Fields(string(text)) {
+		word := strings.ToLower(s.Text())
-			words[word]++
+		words[word]++
-		}
 	}
 	return words
 }