aerospike_bench_test.go

// Copyright 2013-2015 Aerospike, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package aerospike_test

import (
	"bytes"
	"runtime"

	. "github.com/aerospike/aerospike-client-go"

	"testing"
)

var r *Record
var err error

type OBJECT struct {
	Price  int
	DBName string
	Blob   []byte
}

func benchGet(times int, client *Client, key *Key, obj interface{}) {
	for i := 0; i < times; i++ {
		if obj == nil {
			if r, err = client.Get(nil, key); err != nil {
				panic(err)
			}
		} else {
			if err = client.GetObject(nil, key, obj); err != nil {
				panic(err)
			}
		}
	}
}

func benchPut(times int, client *Client, key *Key, wp *WritePolicy, obj interface{}) {
	for i := 0; i < times; i++ {
		if obj == nil {
			dbName := NewBin("dbname", "CouchDB")
			price := NewBin("price", 0)
			keywords := NewBin("keywords", []string{"concurrent", "fast"})
			if err = client.PutBins(wp, key, dbName, price, keywords); err != nil {
				panic(err)
			}
		} else {
			if err = client.PutObject(wp, key, obj); err != nil {
				panic(err)
			}
		}
	}
}

func Benchmark_Get(b *testing.B) {
	client, err := NewClientWithPolicy(clientPolicy, *host, *port)
	if err != nil {
		b.Fail()
	}

	key, _ := NewKey("test", "databases", "Aerospike")
	obj := &OBJECT{198, "Jack Shaftoe and Company", []byte(bytes.Repeat([]byte{32}, 1000))}
	client.PutObject(nil, key, obj)

	b.N = 100000
	runtime.GC()
	b.ResetTimer()
	benchGet(b.N, client, key, nil)
}

func Benchmark_GetObject(b *testing.B) {
	client, err := NewClientWithPolicy(clientPolicy, *host, *port)
	if err != nil {
		b.Fail()
	}

	key, _ := NewKey("test", "databases", "Aerospike")

	obj := &OBJECT{198, "Jack Shaftoe and Company", []byte(bytes.Repeat([]byte{32}, 1000))}
	client.PutObject(nil, key, obj)

	b.N = 100000
	runtime.GC()
	b.ResetTimer()
	benchGet(b.N, client, key, obj)
}

func Benchmark_Put(b *testing.B) {
	client, err := NewClient(*host, *port)
	if err != nil {
		b.Fail()
	}

	key, _ := NewKey("test", "databases", "Aerospike")
	writepolicy := NewWritePolicy(0, 0)

	b.N = 100000
	runtime.GC()
	b.ResetTimer()
	benchPut(b.N, client, key, writepolicy, nil)
}

func Benchmark_PutObject(b *testing.B) {
	client, err := NewClient(*host, *port)
	if err != nil {
		b.Fail()
	}

	obj := &OBJECT{198, "Jack Shaftoe and Company", []byte(bytes.Repeat([]byte{32}, 1000))}
	key, _ := NewKey("test", "databases", "Aerospike")
	writepolicy := NewWritePolicy(0, 0)

	b.N = 100000
	runtime.GC()
	b.ResetTimer()
	benchPut(b.N, client, key, writepolicy, obj)
}