Data streaming demo to a csv file in go

main.go

package main

import (
	"encoding/csv"
	"fmt"
	"log"
	"os"
)

type userInfo struct {
	Name	string
	Age		float64
}

func getHeaders() []string {
	return []string{"name", "age"}
}

func (u *userInfo) getValues() []string {
	return []string{u.Name, fmt.Sprintf("%f", u.Age)}
}

func main() {
	// create dummy date
	users := []userInfo{
		{
			"sudaraka",
			24.4,
		},
		{
			"swee",
			24.4,
		},
	}

	// setup writer
	csvOut, err := os.Create("demo.csv")
	if err != nil {
		log.Fatal("Unable to create file")
	}
	defer csvOut.Close()

	writer := csv.NewWriter(csvOut)
	defer writer.Flush()

	// get headers and write headers
	headers := getHeaders()

	// write headers
	err = writer.Write(headers)
	if err != nil {
		log.Fatal("Error:", err)
		return
	}

	for _, user := range users {
		err = writer.Write(user.getValues())
		if err != nil {
			log.Fatal("Error:", err)
			return
		}
	}
}