Skip to content

Commit 9f51159

Browse files
chore(go): cleanup files generated by examples (#806)
1 parent fe49c73 commit 9f51159

File tree

2 files changed

+17
-0
lines changed

2 files changed

+17
-0
lines changed

AwsEncryptionSDK/runtimes/go/examples/main.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -165,4 +165,7 @@ func main() {
165165
utils.GenerateUUIDTestData(numOfString),
166166
utils.DefaultKMSKeyId(),
167167
utils.DefaultKmsKeyRegion())
168+
169+
// cleanup
170+
utils.CleanUpFiles()
168171
}

AwsEncryptionSDK/runtimes/go/examples/utils/exampleUtils.go

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,14 @@ twIDAQAB
5353
testKmsEcdhKeyIdP256RecipientKeyId = "arn:aws:kms:us-west-2:370957321024:key/0265c8e9-5b6a-4055-8f70-63719e09fda5"
5454
)
5555

56+
var filesCreatedByExamples = []string{
57+
eccPrivateKeyFileNameSender,
58+
eccPrivateKeyFileNameRecipient,
59+
eccPublicKeyFileNameRecipient,
60+
kmsEccPublicKeyFileNameRecipient,
61+
kmsEccPublicKeyFileNameSender,
62+
}
63+
5664
// Getter functions
5765

5866
func KmsEcdhKeyIdP256SenderKeyId() string {
@@ -157,6 +165,12 @@ func KeyStoreName() string {
157165

158166
// Utility functions
159167

168+
func CleanUpFiles() {
169+
for _, file := range filesCreatedByExamples {
170+
os.Remove(file)
171+
}
172+
}
173+
160174
func WriteRawEcdhEccKeys(ecdhCurveSpec awscryptographyprimitivessmithygeneratedtypes.ECDHCurveSpec) error {
161175
// Safety check: Validate neither file is present
162176
if FileExists(eccPrivateKeyFileNameSender) ||

0 commit comments

Comments
 (0)