Fix MemProfileAllocs and MemProfileHeap options

Fixes #55

Author: davecheney

example_test.go

@@ -29,6 +29,16 @@ func ExampleMemProfileRate() {
 	defer profile.Start(profile.MemProfileRate(2048)).Stop()
 }
 
+func ExampleMemProfileHeap() {
+	// use heap memory profiling.
+	defer profile.Start(profile.MemProfileHeap).Stop()
+}
+
+func ExampleMemProfileAllocs() {
+	// use allocs memory profiling.
+	defer profile.Start(profile.MemProfileAllocs).Stop()
+}
+
 func ExampleProfilePath() {
 	// set the location that the profile will be written to
 	defer profile.Start(profile.ProfilePath(os.Getenv("HOME"))).Stop()

profile.go

@@ -90,20 +90,16 @@ func MemProfileRate(rate int) func(*Profile) {
 
 // MemProfileHeap changes which type of memory profiling to profile
 // the heap.
-func MemProfileHeap() func(*Profile) {
-	return func(p *Profile) {
-		p.memProfileType = "heap"
-		p.mode = memMode
-	}
+func MemProfileHeap(p *Profile) {
+	p.memProfileType = "heap"
+	p.mode = memMode
 }
 
 // MemProfileAllocs changes which type of memory to profile
 // allocations.
-func MemProfileAllocs() func(*Profile) {
-	return func(p *Profile) {
-		p.memProfileType = "allocs"
-		p.mode = memMode
-	}
+func MemProfileAllocs(p *Profile) {
+	p.memProfileType = "allocs"
+	p.mode = memMode
 }
 
 // MutexProfile enables mutex profiling.