@@ -24,8 +24,8 @@ var datakeyopts = map[string]primitive.M{
24
24
"key" : "arn:aws:kms:us-east-1:579766882180:key/89fcc2c4-08b0-4bd9-9f25-e30687b580d0" ,
25
25
},
26
26
"azure" : bson.M {
27
- "keyVaultEndpoint" : "https://keyvault-drivers-2411.vault.azure.net/keys/ " ,
28
- "keyName" : "KEY-NAME " ,
27
+ "keyVaultEndpoint" : "" ,
28
+ "keyName" : "" ,
29
29
},
30
30
"gcp" : bson.M {
31
31
"projectId" : "devprod-drivers" ,
@@ -53,6 +53,20 @@ func main() {
53
53
default :
54
54
ok = true
55
55
}
56
+ if provider == "azure" {
57
+ azureKmsKeyName := os .Getenv ("AZUREKMS_KEY_NAME" )
58
+ azureKmsKeyVaultEndpoint := os .Getenv ("AZUREKMS_KEY_VAULT_ENDPOINT" )
59
+ if azureKmsKeyName == "" {
60
+ fmt .Println ("ERROR: Please set required AZUREKMS_KEY_NAME environment variable." )
61
+ ok = false
62
+ }
63
+ if azureKmsKeyVaultEndpoint == "" {
64
+ fmt .Println ("ERROR: Please set required AZUREKMS_KEY_VAULT_ENDPOINT environment variable." )
65
+ ok = false
66
+ }
67
+ datakeyopts ["azure" ]["keyName" ] = azureKmsKeyName
68
+ datakeyopts ["azure" ]["keyVaultEndpoint" ] = azureKmsKeyVaultEndpoint
69
+ }
56
70
if ! ok {
57
71
providers := make ([]string , 0 , len (datakeyopts ))
58
72
for p := range datakeyopts {
@@ -63,6 +77,8 @@ func main() {
63
77
fmt .Println ("- MONGODB_URI as a MongoDB URI. Example: 'mongodb://localhost:27017'" )
64
78
fmt .Println ("- EXPECT_ERROR as an optional expected error substring." )
65
79
fmt .Println ("- PROVIDER as a KMS provider, which supports:" , strings .Join (providers , ", " ))
80
+ fmt .Println ("- AZUREKMS_KEY_NAME as the Azure key name. Required if PROVIDER=azure." )
81
+ fmt .Println ("- AZUREKMS_KEY_VAULT_ENDPOINT as the Azure key name. Required if PROVIDER=azure." )
66
82
os .Exit (1 )
67
83
}
68
84
0 commit comments