func ConstructRandomStateDelta(
t testing.TB,
chaincodeIDPrefix string,
numChaincodes int,
maxKeySuffix int,
numKeysToInsert int,
kvSize int) *StateDelta {
delta := NewStateDelta()
s2 := rand.NewSource(time.Now().UnixNano())
r2 := rand.New(s2)
for i := 0; i < numKeysToInsert; i++ {
chaincodeID := chaincodeIDPrefix + "_" + strconv.Itoa(r2.Intn(numChaincodes))
key := "key_" + strconv.Itoa(r2.Intn(maxKeySuffix))
valueSize := kvSize - len(key)
if valueSize < 1 {
panic(fmt.Errorf("valueSize cannot be less than one. ValueSize=%d", valueSize))
}
value := testutil.ConstructRandomBytes(t, valueSize)
delta.Set(chaincodeID, key, value, nil)
}
for _, chaincodeDelta := range delta.ChaincodeStateDeltas {
sortedKeys := chaincodeDelta.getSortedKeys()
smallestKey := sortedKeys[0]
largestKey := sortedKeys[len(sortedKeys)-1]
t.Logf("chaincode=%s, numKeys=%d, smallestKey=%s, largestKey=%s", chaincodeDelta.ChaincodeID, len(sortedKeys), smallestKey, largestKey)
}
return delta
}
func BenchmarkLedgerSingleKeyTransaction(b *testing.B) {
b.Logf("testParams:%q", testParams)
flags := flag.NewFlagSet("testParams", flag.ExitOnError)
key := flags.String("Key", "key", "key name")
kvSize := flags.Int("KVSize", 1000, "size of the key-value")
batchSize := flags.Int("BatchSize", 100, "size of the key-value")
numBatches := flags.Int("NumBatches", 100, "number of batches")
numWritesToLedger := flags.Int("NumWritesToLedger", 4, "size of the key-value")
flags.Parse(testParams)
b.Logf(`Running test with params: key=%s, kvSize=%d, batchSize=%d, numBatches=%d, NumWritesToLedger=%d`,
*key, *kvSize, *batchSize, *numBatches, *numWritesToLedger)
ledgerTestWrapper := createFreshDBAndTestLedgerWrapper(b)
ledger := ledgerTestWrapper.ledger
chaincode := "chaincodeId"
value := testutil.ConstructRandomBytes(b, *kvSize-(len(chaincode)+len(*key)))
tx := constructDummyTx(b)
serializedBytes, _ := tx.Bytes()
b.Logf("Size of serialized bytes for tx = %d", len(serializedBytes))
b.ResetTimer()
for n := 0; n < b.N; n++ {
for i := 0; i < *numBatches; i++ {
ledger.BeginTxBatch(1)
// execute one batch
var transactions []*protos.Transaction
for j := 0; j < *batchSize; j++ {
ledger.TxBegin("txUuid")
_, err := ledger.GetState(chaincode, *key, true)
if err != nil {
b.Fatalf("Error in getting state: %s", err)
}
for l := 0; l < *numWritesToLedger; l++ {
ledger.SetState(chaincode, *key, value)
}
ledger.TxFinished("txUuid", true)
transactions = append(transactions, tx)
}
ledger.CommitTxBatch(1, transactions, nil, []byte("proof"))
}
}
b.StopTimer()
//varify value persisted
value, _ = ledger.GetState(chaincode, *key, true)
size := ledger.GetBlockchainSize()
b.Logf("Value size=%d, Blockchain height=%d", len(value), size)
}
func populateDB(tb testing.TB, kvSize int, totalKeys int, keyPrefix string) {
dbWrapper := db.NewTestDBWrapper()
dbWrapper.CreateFreshDB(tb)
batch := gorocksdb.NewWriteBatch()
for i := 0; i < totalKeys; i++ {
key := []byte(keyPrefix + strconv.Itoa(i))
value := testutil.ConstructRandomBytes(tb, kvSize-len(key))
batch.Put(key, value)
if i%1000 == 0 {
dbWrapper.WriteToDB(tb, batch)
batch = gorocksdb.NewWriteBatch()
}
}
dbWrapper.CloseDB(tb)
}
func BenchmarkLedgerPopulate(b *testing.B) {
b.Logf("testParams:%q", testParams)
disableLogging()
flags := flag.NewFlagSet("testParams", flag.ExitOnError)
kvSize := flags.Int("KVSize", 1000, "size of the key-value")
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")
batchSize := flags.Int("BatchSize", 100, "size of the key-value")
flags.Parse(testParams)
b.Logf(`Running test with params: keyPrefix=%s, kvSize=%d, batchSize=%d, maxKeySuffix=%d`,
*keyPrefix, *kvSize, *batchSize, *maxKeySuffix)
ledgerTestWrapper := createFreshDBAndTestLedgerWrapper(b)
ledger := ledgerTestWrapper.ledger
chaincode := "chaincodeId"
numBatches := *maxKeySuffix / *batchSize
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++ {
ledger.TxBegin("txUuid")
b.StopTimer()
keyNumber := batchID*(*batchSize) + j
key := *keyPrefix + strconv.Itoa(keyNumber)
b.StartTimer()
ledger.SetState(chaincode, key, 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))
}
func BenchmarkCryptoHash(b *testing.B) {
flags := flag.NewFlagSet("testParams", flag.ExitOnError)
numBytesPointer := flags.Int("NumBytes", -1, "Number of Bytes")
flags.Parse(testParams)
numBytes := *numBytesPointer
if numBytes == -1 {
b.Fatal("Missing value for parameter NumBytes")
}
randomBytes := testutil.ConstructRandomBytes(b, numBytes)
//b.Logf("byte size=%d, b.N=%d", len(randomBytes), b.N)
b.ResetTimer()
for i := 0; i < b.N; i++ {
util.ComputeCryptoHash(randomBytes)
}
}
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))
}