func BenchmarkDB(b *testing.B) {
b.Logf("testParams:%q", testParams)
flags := flag.NewFlagSet("testParams", flag.ExitOnError)
kvSize := flags.Int("KVSize", 1000, "size of the key-value")
toPopulateDB := flags.Bool("PopulateDB", false, "Run in populate DB mode")
maxKeySuffix := flags.Int("MaxKeySuffix", 1, "the keys are appended with _1, _2,.. upto MaxKeySuffix")
keyPrefix := flags.String("KeyPrefix", "Key_", "The generated workload will have keys such as KeyPrefix_1, KeyPrefix_2, and so on")
flags.Parse(testParams)
if *toPopulateDB {
b.ResetTimer()
populateDB(b, *kvSize, *maxKeySuffix, *keyPrefix)
return
}
dbWrapper := db.NewTestDBWrapper()
randNumGen := testutil.NewTestRandomNumberGenerator(*maxKeySuffix)
b.ResetTimer()
for i := 0; i < b.N; i++ {
key := []byte(*keyPrefix + strconv.Itoa(randNumGen.Next()))
value := dbWrapper.GetFromDB(b, key)
b.SetBytes(int64(len(value)))
}
}
func BenchmarkLedgerRandomTransactions(b *testing.B) {
disableLogging()
b.Logf("testParams:%q", testParams)
flags := flag.NewFlagSet("testParams", flag.ExitOnError)
keyPrefix := flags.String("KeyPrefix", "Key_", "The generated workload will have keys such as KeyPrefix_1, KeyPrefix_2, and so on")
kvSize := flags.Int("KVSize", 1000, "size of the key-value")
maxKeySuffix := flags.Int("MaxKeySuffix", 1, "the keys are appended with _1, _2,.. upto MaxKeySuffix")
batchSize := flags.Int("BatchSize", 100, "size of the key-value")
numBatches := flags.Int("NumBatches", 100, "number of batches")
numReadsFromLedger := flags.Int("NumReadsFromLedger", 4, "Number of Key-Values to read")
numWritesToLedger := flags.Int("NumWritesToLedger", 4, "Number of Key-Values to write")
flags.Parse(testParams)
b.Logf(`Running test with params: keyPrefix=%s, kvSize=%d, batchSize=%d, maxKeySuffix=%d, numBatches=%d, numReadsFromLedger=%d, numWritesToLedger=%d`,
*keyPrefix, *kvSize, *batchSize, *maxKeySuffix, *numBatches, *numReadsFromLedger, *numWritesToLedger)
ledger, err := newLedger()
testutil.AssertNoError(b, err, "Error while constructing ledger")
chaincode := "chaincodeId"
tx := constructDummyTx(b)
value := testutil.ConstructRandomBytes(b, *kvSize-(len(chaincode)+len(*keyPrefix)))
b.ResetTimer()
for i := 0; i < b.N; i++ {
for batchID := 0; batchID < *numBatches; batchID++ {
ledger.BeginTxBatch(1)
// execute one batch
var transactions []*protos.Transaction
for j := 0; j < *batchSize; j++ {
b.StopTimer()
randomKeySuffixGen := testutil.NewTestRandomNumberGenerator(*maxKeySuffix)
b.StartTimer()
ledger.TxBegin("txUuid")
for k := 0; k < *numReadsFromLedger; k++ {
b.StopTimer()
randomKey := *keyPrefix + strconv.Itoa(randomKeySuffixGen.Next())
b.StartTimer()
ledger.GetState(chaincode, randomKey, true)
}
for k := 0; k < *numWritesToLedger; k++ {
b.StopTimer()
randomKey := *keyPrefix + strconv.Itoa(randomKeySuffixGen.Next())
b.StartTimer()
ledger.SetState(chaincode, randomKey, value)
}
ledger.TxFinished("txUuid", true)
b.StopTimer()
transactions = append(transactions, tx)
b.StartTimer()
}
ledger.CommitTxBatch(1, transactions, nil, []byte("proof"))
}
}
b.StopTimer()
b.Logf("DB stats afters populating: %s", testDBWrapper.GetEstimatedNumKeys(b))
}