Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
107 changes: 107 additions & 0 deletions sdk/keyvault/azcertificates/testdata/perf/get_certificate.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,107 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See License.txt in the project root for license information.

package main

import (
"context"
"fmt"
"os"
"time"

"github.com/Azure/azure-sdk-for-go/sdk/azcore"
"github.com/Azure/azure-sdk-for-go/sdk/azidentity"
"github.com/Azure/azure-sdk-for-go/sdk/internal/perf"
"github.com/Azure/azure-sdk-for-go/sdk/keyvault/azcertificates"
)

type getCertificatesTestOptions struct{}

var getCertTestOpts getCertificatesTestOptions = getCertificatesTestOptions{}

type getCertificateTest struct {
perf.PerfTestOptions
certificateName string
client *azcertificates.Client
}

// newGetCertificateTest is called once per process
func newGetCertificateTest(ctx context.Context, options perf.PerfTestOptions) (perf.GlobalPerfTest, error) {
certName := "livekvtestgetcertperfcert"
d := &getCertificateTest{
PerfTestOptions: options,
certificateName: certName,
}

vaultURL, ok := os.LookupEnv("AZURE_KEYVAULT_URL")
if !ok {
return nil, fmt.Errorf("the environment variable 'AZURE_KEYVAULT_URL' could not be found")
}

cred, err := azidentity.NewDefaultAzureCredential(nil)
if err != nil {
panic(err)
}

client, err := azcertificates.NewClient(vaultURL, cred, &azcertificates.ClientOptions{
ClientOptions: azcore.ClientOptions{
Transport: options.Transporter,
},
})
if err != nil {
return nil, err
}

poller, err := client.BeginCreateCertificate(ctx, d.certificateName, azcertificates.NewDefaultCertificatePolicy(), nil)
if err != nil {
return nil, err
}
_, err = poller.PollUntilDone(ctx, 500*time.Millisecond)
if err != nil {
return nil, err
}

d.client = client
return d, nil
}

func (gct *getCertificateTest) GlobalCleanup(ctx context.Context) error {
poller, err := gct.client.BeginDeleteCertificate(ctx, gct.certificateName, nil)
if err != nil {
return err
}

_, err = poller.PollUntilDone(ctx, 500*time.Millisecond)
if err != nil {
return err
}

_, err = gct.client.PurgeDeletedCertificate(ctx, gct.certificateName, nil)
return err
}

type getCertificatePerfTest struct {
*getCertificateTest
perf.PerfTestOptions
client *azcertificates.Client
certificateName string
}

// NewPerfTest is called once per goroutine
func (gct *getCertificateTest) NewPerfTest(ctx context.Context, options *perf.PerfTestOptions) (perf.PerfTest, error) {
return &getCertificatePerfTest{
getCertificateTest: gct,
PerfTestOptions: *options,
client: gct.client,
certificateName: gct.certificateName,
}, nil
}

func (gcpt *getCertificatePerfTest) Run(ctx context.Context) error {
_, err := gcpt.client.GetCertificate(ctx, gcpt.certificateName, nil)
return err
}

func (*getCertificatePerfTest) Cleanup(ctx context.Context) error {
return nil
}
23 changes: 23 additions & 0 deletions sdk/keyvault/azcertificates/testdata/perf/go.mod
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
module github.com/Azure/azure-sdk-for-go/sdk/keyvault/azcertificates/testdata/perf

go 1.18

require (
github.com/Azure/azure-sdk-for-go/sdk/azidentity v0.14.0
github.com/Azure/azure-sdk-for-go/sdk/internal v0.9.2
github.com/Azure/azure-sdk-for-go/sdk/keyvault/azcertificates v0.3.0
)

require (
github.com/Azure/azure-sdk-for-go/sdk/azcore v0.23.0 // indirect
github.com/Azure/azure-sdk-for-go/sdk/keyvault/internal v0.3.0 // indirect
github.com/AzureAD/microsoft-authentication-library-for-go v0.4.0 // indirect
github.com/golang-jwt/jwt v3.2.1+incompatible // indirect
github.com/google/uuid v1.1.1 // indirect
github.com/kylelemons/godebug v1.1.0 // indirect
github.com/pkg/browser v0.0.0-20210115035449-ce105d075bb4 // indirect
golang.org/x/crypto v0.0.0-20201016220609-9e8e0b390897 // indirect
golang.org/x/net v0.0.0-20211015210444-4f30a5c0130f // indirect
golang.org/x/sys v0.0.0-20211019181941-9d821ace8654 // indirect
golang.org/x/text v0.3.7 // indirect
)
57 changes: 57 additions & 0 deletions sdk/keyvault/azcertificates/testdata/perf/go.sum
Original file line number Diff line number Diff line change
@@ -0,0 +1,57 @@
github.com/Azure/azure-sdk-for-go/sdk/azcore v0.23.0 h1:D7l5jspkc4kwBYRWoZE4DQnu6LVpLwDsMZjBKS4wZLQ=
github.com/Azure/azure-sdk-for-go/sdk/azcore v0.23.0/go.mod h1:w5pDIZuawUmY3Bj4tVx3Xb8KS96ToB0j315w9rqpAg0=
github.com/Azure/azure-sdk-for-go/sdk/azidentity v0.14.0 h1:NVS/4LOQfkBpk+B1VopIzv1ptmYeEskA8w/3K/w7vjo=
github.com/Azure/azure-sdk-for-go/sdk/azidentity v0.14.0/go.mod h1:RG0cZndeZM17StwohYclmcXSr4oOJ8b1I5hB8llIc6Y=
github.com/Azure/azure-sdk-for-go/sdk/internal v0.9.2 h1:Px2KVERcYEg2Lv25AqC2hVr0xUWaq94wuEObLIkYzmA=
github.com/Azure/azure-sdk-for-go/sdk/internal v0.9.2/go.mod h1:CdSJQNNzZhCkwDaV27XV1w48ZBPtxe7mlrZAsPNxD5g=
github.com/Azure/azure-sdk-for-go/sdk/keyvault/azcertificates v0.3.0 h1:XA+SA4ftUu4YuI+WKhSuGC5IubhBVjv2uat2zYlfF+0=
github.com/Azure/azure-sdk-for-go/sdk/keyvault/azcertificates v0.3.0/go.mod h1:wsBD1Qnm3+LVvHC4JOLGNwosCz4veZ9mmClsBOctQSA=
github.com/Azure/azure-sdk-for-go/sdk/keyvault/internal v0.3.0 h1:0gy84rslo34rGGBe2cDxfs4iDMwbKc0/4yDna1S7j8Q=
github.com/Azure/azure-sdk-for-go/sdk/keyvault/internal v0.3.0/go.mod h1:mu846WjGmdK5vWqWv25J416znWpnFjZp4+O34KW8H7U=
github.com/AzureAD/microsoft-authentication-library-for-go v0.4.0 h1:WVsrXCnHlDDX8ls+tootqRE87/hL9S/g4ewig9RsD/c=
github.com/AzureAD/microsoft-authentication-library-for-go v0.4.0/go.mod h1:Vt9sXTKwMyGcOxSmLDMnGPgqsUg7m8pe215qMLrDXw4=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
github.com/dnaeon/go-vcr v1.1.0 h1:ReYa/UBrRyQdant9B4fNHGoCNKw6qh6P0fsdGmZpR7c=
github.com/dnaeon/go-vcr v1.1.0/go.mod h1:M7tiix8f0r6mKKJ3Yq/kqU1OYf3MnfmBWVbPx/yU9ko=
github.com/golang-jwt/jwt v3.2.1+incompatible h1:73Z+4BJcrTC+KczS6WvTPvRGOp1WmfEP4Q1lOd9Z/+c=
github.com/golang-jwt/jwt v3.2.1+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I=
github.com/golang-jwt/jwt/v4 v4.2.0 h1:besgBTC8w8HjP6NzQdxwKH9Z5oQMZ24ThTrHp3cZ8eU=
github.com/google/uuid v1.1.1 h1:Gkbcsh/GbpXz7lPftLA3P6TYMwjCLYm83jiFQZF/3gY=
github.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
github.com/kylelemons/godebug v1.1.0 h1:RPNrshWIDI6G2gRW9EHilWtl7Z6Sb1BR0xunSBf0SNc=
github.com/kylelemons/godebug v1.1.0/go.mod h1:9/0rRGxNHcop5bhtWyNeEfOS8JIWk580+fNqagV/RAw=
github.com/modocache/gover v0.0.0-20171022184752-b58185e213c5/go.mod h1:caMODM3PzxT8aQXRPkAt8xlV/e7d7w8GM5g0fa5F0D8=
github.com/montanaflynn/stats v0.6.6/go.mod h1:etXPPgVO6n31NxCd9KQUMvCM+ve0ruNzt6R8Bnaayow=
github.com/pkg/browser v0.0.0-20210115035449-ce105d075bb4 h1:Qj1ukM4GlMWXNdMBuXcXfz/Kw9s1qm0CLY32QxuSImI=
github.com/pkg/browser v0.0.0-20210115035449-ce105d075bb4/go.mod h1:N6UoU20jOqggOuDwUaBQpluzLNDqif3kq9z2wpdYEfQ=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY=
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
golang.org/x/crypto v0.0.0-20201016220609-9e8e0b390897 h1:pLI5jrR7OSLijeIDcmRxNmw2api+jEfxLoykJVice/E=
golang.org/x/crypto v0.0.0-20201016220609-9e8e0b390897/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
golang.org/x/net v0.0.0-20201010224723-4f7140c49acb/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
golang.org/x/net v0.0.0-20211015210444-4f30a5c0130f h1:OfiFi4JbukWwe3lzw+xunroH1mnC1e2Gy5cxNJApiSY=
golang.org/x/net v0.0.0-20211015210444-4f30a5c0130f/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20210124154548-22da62e12c0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20211019181941-9d821ace8654 h1:id054HUawV2/6IGm2IV8KZQjqtwAOo2CYlOToYqa0d0=
golang.org/x/sys v0.0.0-20211019181941-9d821ace8654/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
golang.org/x/text v0.3.7 h1:olpwvP2KacW1ZWvsR7uQhoyTYvKAupfQrRGBFM352Gk=
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b h1:h8qDotaEPuJATrMmW04NCwg7v22aHH28wwpauUhK9Oo=
12 changes: 12 additions & 0 deletions sdk/keyvault/azcertificates/testdata/perf/main.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See License.txt in the project root for license information.

package main

import "github.com/Azure/azure-sdk-for-go/sdk/internal/perf"

func main() {
perf.Run(map[string]perf.PerfMethods{
"GetCertificateTest": {Register: nil, New: newGetCertificateTest},
})
}
132 changes: 132 additions & 0 deletions sdk/keyvault/azkeys/testdata/perf/decrypt.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,132 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See License.txt in the project root for license information.

package main

import (
"context"
"fmt"
"math/rand"
"os"
"time"

"github.com/Azure/azure-sdk-for-go/sdk/azcore"
"github.com/Azure/azure-sdk-for-go/sdk/azcore/to"
"github.com/Azure/azure-sdk-for-go/sdk/azidentity"
"github.com/Azure/azure-sdk-for-go/sdk/internal/perf"
"github.com/Azure/azure-sdk-for-go/sdk/keyvault/azkeys"
"github.com/Azure/azure-sdk-for-go/sdk/keyvault/azkeys/crypto"
)

type decryptTestOptions struct{}

var decryptTestOpts decryptTestOptions = decryptTestOptions{}

type decryptTest struct {
perf.PerfTestOptions
keyName string
client *azkeys.Client
cryptoClient *crypto.Client
encrypAlg crypto.EncryptionAlg
cipherText []byte
}

// newDecryptTest is called once per process
func newDecryptTest(ctx context.Context, options perf.PerfTestOptions) (perf.GlobalPerfTest, error) {
d := &decryptTest{
PerfTestOptions: options,
keyName: "livekvtestdecryptperfkey",
encrypAlg: crypto.EncryptionAlgRSAOAEP256,
}

vaultURL, ok := os.LookupEnv("AZURE_KEYVAULT_URL")
if !ok {
return nil, fmt.Errorf("the environment variable 'AZURE_KEYVAULT_URL' could not be found")
}

cred, err := azidentity.NewDefaultAzureCredential(nil)
if err != nil {
panic(err)
}

client, err := azkeys.NewClient(vaultURL, cred, &azkeys.ClientOptions{
ClientOptions: azcore.ClientOptions{
Transport: options.Transporter,
},
})
if err != nil {
return nil, err
}

resp, err := client.CreateRSAKey(ctx, d.keyName, &azkeys.CreateRSAKeyOptions{Size: to.Ptr(int32(2048))})
if err != nil {
return nil, err
}

cryptoClient, err := crypto.NewClient(*resp.ID, cred, &crypto.ClientOptions{
ClientOptions: azcore.ClientOptions{
Transport: options.Transporter,
},
})
if err != nil {
return nil, err
}

d.cryptoClient = cryptoClient
d.client = client

b := make([]byte, 32)
_, err = rand.Read(b)
if err != nil {
return nil, err
}

result, err := d.cryptoClient.Encrypt(ctx, d.encrypAlg, b, nil)
if err != nil {
return nil, err
}
d.cipherText = result.Ciphertext
Comment thread
seankane-msft marked this conversation as resolved.
return d, nil
}

func (gct *decryptTest) GlobalCleanup(ctx context.Context) error {
poller, err := gct.client.BeginDeleteKey(ctx, gct.keyName, nil)
if err != nil {
return err
}

_, err = poller.PollUntilDone(ctx, 500*time.Millisecond)
if err != nil {
return err
}

_, err = gct.client.PurgeDeletedKey(ctx, gct.keyName, nil)
return err
}

type decryptPerfTest struct {
*decryptTest
perf.PerfTestOptions
cryptoClient *crypto.Client
alg crypto.EncryptionAlg
cipher []byte
}

// NewPerfTest is called once per goroutine
func (gct *decryptTest) NewPerfTest(ctx context.Context, options *perf.PerfTestOptions) (perf.PerfTest, error) {
return &decryptPerfTest{
decryptTest: gct,
PerfTestOptions: *options,
alg: gct.encrypAlg,
cipher: gct.cipherText,
}, nil
}

func (gcpt *decryptPerfTest) Run(ctx context.Context) error {
_, err := gcpt.cryptoClient.Decrypt(ctx, gcpt.alg, gcpt.cipher, nil)
return err
}

func (*decryptPerfTest) Cleanup(ctx context.Context) error {
return nil
}
Loading