khaf · December 16, 2014 16:27 · Dec 16, 2014
diff --git a/gistfile1.go b/gistfile1.go
@@ -0,0 +1,75 @@
+package main
+
+import (
+	"log"
+	"math"
+	"runtime"
+	"time"
+
+	as "github.com/aerospike/aerospike-client-go"
+)
+
+const (
+	HOSTNAME  = "127.0.0.1"
+	PORT      = 3000
+	NAMESPACE = "test"
+	SET       = ""
+)
+
+func panicOnError(err error) {
+	if err != nil {
+		log.Fatalln(err)
+	}
+}
+
+func main() {
+	runtime.GOMAXPROCS(runtime.NumCPU())
+
+	client, err := as.NewClient(HOSTNAME, PORT)
+	panicOnError(err)
+
+	count := 0
+	for i := 1; i < math.MaxInt16; i *= 2 {
+		for j := count; j < i; j++ {
+			key, _ := as.NewKey(NAMESPACE, SET, j)
+			err := client.PutBins(nil, key, as.NewBin("bin", j))
+			panicOnError(err)
+			count++
+		}
+
+		log.Printf("There are now %d records in the database...\n", i)
+
+		for j := 0; j < 3; j++ {
+			t := time.Now()
+			scan(client, i)
+			log.Printf("Scanning %d records took: %v\n", i, time.Now().Sub(t))
+		}
+
+	}
+}
+
+// scans the database and checks if the number of returned records
+// is the same as the expected number
+func scan(client *as.Client, expectedRecordsCount int) {
+	recordset, err := client.ScanAll(nil, NAMESPACE, SET)
+	panicOnError(err)
+	count := 0
+
+L:
+	for {
+		select {
+		case rec := <-recordset.Records:
+			if rec == nil {
+				break L
+			}
+			count++
+		case err := <-recordset.Errors:
+			panicOnError(err)
+		}
+	}
+
+	log.Println("Number of records returned by ScanAll:", count)
+	if expectedRecordsCount != count {
+		log.Fatalf("Expected ScanAll to return %d records, but it returned %d.\n", expectedRecordsCount, count)
+	}
+}