fizzbuzz in go

fizzbuzz.go

package main

import (
    "fmt"
    "os"
    "strconv"
)

func stream(ch chan int) {
    for i := 1; ; i++ {
        ch <- i
    }
}

func fizzBuzz(num int) {
    switch {
    case num % 15 == 0:
        fmt.Println("FizzBuzz")
    case num % 3 == 0:
        fmt.Println("Fizz")
    case num % 5 == 0:
        fmt.Println("Buzz")
    default:
        fmt.Println(num)
    }
}

func main() {
    ch := make(chan int)
    go stream(ch)
    if len(os.Args) > 1 {
        max, err := strconv.Atoi(os.Args[1])
        if err != nil {
            fmt.Println("invalid argument")
            os.Exit(1)
        }
        for i := 0; i < max; i++ {
            num := <- ch
            fizzBuzz(num)
        }
    }
}