Take home coding test

farm-animals.csv

main.go

package main

import (
	"encoding/csv"
	"fmt"
	"math/rand"
	"os"
	"time"
)

var species = map[string]animal{
	"Horse": {20, []string{"Black", "Brown"}},
	"Duck":  {5, []string{"Green", "Brown"}},
	"Sheep": {15, []string{"White"}},
	"Goat":  {12, []string{"White", "Black", "Brown"}},
	"Cat":   {20, []string{"Black", "White", "Ginger"}},
	"Pig":   {15, []string{"Pink"}},
}
var speciesNames = func() (r []string) {
	for k := range species {
		r = append(r, k)
	}
	return
}()

type animal struct {
	maxAge  int
	colours []string
}

func main() {
	csvw := csv.NewWriter(os.Stdout)
	rand.Seed(time.Now().UnixNano())

	csvw.Write([]string{"Species", "Age", "Colour"})

	animalCount := 10000
	for i := 0; i < animalCount; i++ {
		t := speciesNames[rand.Intn(len(speciesNames))]
		var age = rand.Intn(species[t].maxAge)
		var colour = species[t].colours[rand.Intn(len(species[t].colours))]
		csvw.Write([]string{t, fmt.Sprintf("%d", age), colour})
	}

	csvw.Flush()
}