Measure function time to execute in Go

measure_time.go

package main

import (
	"fmt"
	"time"
)

func measureTime(funcName string) func() {
	start := time.Now()
	return func() {
		fmt.Printf("Time taken by %s function is %v \n", funcName, time.Since(start))
	}
}

func expensivePrint() {
	defer measureTime("expensivePrint")()
	for i := 1; i <= 5; i++ {
		fmt.Printf("Current number is %d \n", i)
		time.Sleep(100 * time.Millisecond)
	}
}

func main() {
	expensivePrint()
	fmt.Println("Finished executing main")
}