LibmongocryptBindingBenchmark.cs

/* Copyright 2010-present MongoDB 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.
 */

usingSystem;
usingSystem.Collections.Generic;
usingSystem.Threading;
usingBenchmarkDotNet.Attributes;
usingMongoDB.Bson;
usingMongoDB.Bson.TestHelpers;
usingMongoDB.Driver;
usingMongoDB.Driver.Encryption;

namespaceMongoDB.Benchmarks
{
publicclassLibmongocryptBindingBenchmark
{
privateconstintRepeatCount=10;
privateconststringLocalMasterKey="Mng0NCt4ZHVUYUJCa1kxNkVyNUR1QURhZ2h2UzR2d2RrZzh0cFBwM3R6NmdWMDFBMUN3YkQ5aXRRMkhGRGdQV09wOGVNYUMxT2k3NjZKelhaQmRCZGJkTXVyZG9uSjFk";

privatebyte[]_encryptedValuesDocumentBytes;
privateIMongoClient_disposableKeyVaultClient;
privateIAutoEncryptionLibMongoCryptController_libMongoCryptController;

[Params(1)]
publicintThreadsCount{get;set;}

[GlobalSetup]
publicvoidSetup()
{
MongoClientSettings.Extensions.AddAutoEncryption();

varlocalMasterKey=Convert.FromBase64String(LocalMasterKey);

varkmsProviders=newDictionary<string,IReadOnlyDictionary<string,object>>();
varlocalKey=newDictionary<string,object>{{"key",localMasterKey}};
kmsProviders.Add("local",localKey);

varkeyVaultNamespace=CollectionNamespace.FromFullName("encryption.__keyVault");
varautoEncryptionOptions=newAutoEncryptionOptions(
keyVaultNamespace:keyVaultNamespace,
kmsProviders:kmsProviders,
bypassAutoEncryption:true);

varclientSettings=MongoClientSettings.FromConnectionString("mongodb://localhost");
clientSettings.AutoEncryptionOptions=autoEncryptionOptions;
clientSettings.ClusterSource=DisposingClusterSource.Instance;

_disposableKeyVaultClient=newMongoClient(clientSettings);

varkeyVaultDatabase=_disposableKeyVaultClient.GetDatabase(keyVaultNamespace.DatabaseNamespace.DatabaseName);
keyVaultDatabase.DropCollection(keyVaultNamespace.CollectionName);
_disposableKeyVaultClient.DropDatabase("crypt-test");

varclientEncryptionSettings=newClientEncryptionOptions(
_disposableKeyVaultClient,
keyVaultNamespace,
kmsProviders);

varencryptedValuesDocument=newBsonDocument();
using(varclientEncryption=newClientEncryption(clientEncryptionSettings))
{
vardataKeyId=clientEncryption.CreateDataKey(
"local",
newDataKeyOptions(),
CancellationToken.None);

varencryptOptions=newEncryptOptions(
EncryptionAlgorithm.AEAD_AES_256_CBC_HMAC_SHA_512_Deterministic.ToString(),
keyId:dataKeyId);

for(inti=0;i<1500;i++)
{
vartoEncryptString=$"value {(i+1):D4}";
varencryptedString=
clientEncryption.Encrypt(toEncryptString,encryptOptions,CancellationToken.None);
encryptedValuesDocument.Add(newBsonElement($"key{(i+1):D4}",encryptedString));
}
}
_encryptedValuesDocumentBytes=encryptedValuesDocument.ToBson();

// Create libmongocrypt binding that will be used for decryption
_libMongoCryptController=
MongoClientSettings.Extensions.AutoEncryptionProvider.CreateAutoCryptClientController(_disposableKeyVaultClient,autoEncryptionOptions);
}

[Benchmark]
publicvoidBulkDecryptionUsingBinding()
{
ThreadingUtilities.ExecuteOnNewThreads(ThreadsCount, _ =>
{
for(inti=0;i<RepeatCount;i++)
{
_libMongoCryptController.DecryptFields(_encryptedValuesDocumentBytes,CancellationToken.None);
}
},20000);
}

[GlobalCleanup]
publicvoidCleanup()
{
_libMongoCryptController.Dispose();
_disposableKeyVaultClient.Dispose();
}
}
}