Merge pull request #4 from kishaningithub/add-profiling

Add profiling and update documentation
kishaningithub · Oct 30, 2021 · f38f417 · f38f417
2 parents d7bfc03 + eb9e455
commit f38f417
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -26,4 +26,7 @@ tools/
 assets.go
 
 # Deps
-vendor/*
+vendor/*
+
+# Pprof files
+*.pprof
diff --git a/README.md b/README.md
@@ -13,10 +13,12 @@ Command line random text generator
 ```bash
 $ randomtext -h
 Usage of randomtext:
+  -profile string
+    	Type of profile - cpu,mem
   -size string
-        Size of generated random text in KB, MB, GB, TB (default "1MB")
+    	Size of generated random text in KB, MB, GB, TB (default "1MB")
   -type string
-        Type of text to be generated - chars, words, zeros (default "chars")
+    	Type of text to be generated - chars, words, zeros (default "chars")
 ```
 
 ### Examples

diff --git a/cmd/randomtext/main.go b/cmd/randomtext/main.go
@@ -5,6 +5,7 @@ import (
 	"flag"
 	"fmt"
 	"github.com/kishaningithub/randomtext"
+	"github.com/pkg/profile"
 	"os"
 	"regexp"
 	"strconv"
@@ -14,7 +15,15 @@ import (
 func main() {
 	sizePtr := flag.String("size", "1MB", "Size of generated random text in KB, MB, GB, TB")
 	typePtr := flag.String("type", "chars", "Type of text to be generated - chars, words, zeros")
+	profileTypePtr := flag.String("profile", "", "Type of profile - cpu,mem")
 	flag.Parse()
+	profileType := *profileTypePtr
+	switch profileType {
+	case "cpu":
+		defer profile.Start(profile.ProfilePath(".")).Stop()
+	case "mem":
+		defer profile.Start(profile.ProfilePath("."), profile.MemProfile).Stop()
+	}
 	sizeInBytes, err := parseSize(strings.ToLower(*sizePtr))
 	handleErr(err)
 	generator, err := parseType(strings.ToLower(*typePtr))

diff --git a/go.mod b/go.mod
@@ -2,7 +2,10 @@ module github.com/kishaningithub/randomtext
 
 go 1.17
 
-require github.com/stretchr/testify v1.7.0
+require (
+	github.com/pkg/profile v1.6.0
+	github.com/stretchr/testify v1.7.0
+)
 
 require (
 	github.com/davecgh/go-spew v1.1.0 // indirect

diff --git a/go.sum b/go.sum
@@ -1,5 +1,7 @@
 github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8=
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/pkg/profile v1.6.0 h1:hUDfIISABYI59DyeB3OTay/HxSRwTQ8rB/H83k6r5dM=
+github.com/pkg/profile v1.6.0/go.mod h1:qBsxPvzyUincmltOk6iyRVxHYg4adc0OFOv72ZdLa18=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=