Added tmp code

Author: akhil-sumologic

go.mod

@@ -10,6 +10,7 @@ require (
 	github.com/aws/aws-sdk-go-v2/feature/s3/manager v1.19.8
 	github.com/aws/aws-sdk-go-v2/service/kms v1.45.4
 	github.com/aws/aws-sdk-go-v2/service/s3 v1.88.2
+	github.com/aws/aws-sdk-go-v2/service/sts v1.38.5
 	github.com/google/uuid v1.6.0
 	github.com/sirupsen/logrus v1.9.3
 )
@@ -28,7 +29,6 @@ require (
 	github.com/aws/aws-sdk-go-v2/service/internal/s3shared v1.19.8 // indirect
 	github.com/aws/aws-sdk-go-v2/service/sso v1.29.4 // indirect
 	github.com/aws/aws-sdk-go-v2/service/ssooidc v1.35.0 // indirect
-	github.com/aws/aws-sdk-go-v2/service/sts v1.38.5 // indirect
 	github.com/aws/smithy-go v1.23.0 // indirect
 	golang.org/x/sys v0.36.0 // indirect
 )

lambda-extensions/utils/awsutils.go

@@ -4,10 +4,11 @@ import (
 	"context"
 	"io"
 	"os"
-
+    "fmt"
 	"github.com/aws/aws-sdk-go-v2/config"
 	"github.com/aws/aws-sdk-go-v2/feature/s3/manager"
 	"github.com/aws/aws-sdk-go-v2/service/s3"
+	"github.com/aws/aws-sdk-go-v2/service/sts"
 )
 
 var uploader *manager.Uploader
@@ -31,6 +32,20 @@ func init() {
 
 	// Create uploader
 	uploader = manager.NewUploader(s3Client)
+
+	// Create STS client to get account ID
+    stsClient := sts.NewFromConfig(cfg)
+    // Get caller identity
+    identity, err := stsClient.GetCallerIdentity(context.TODO(), &sts.GetCallerIdentityInput{})
+    if err != nil {
+        panic("unable to get AWS caller identity: " + err.Error())
+    }
+
+    // Print complete identity
+    fmt.Println("AWS Caller Identity:")
+    fmt.Println("Account ID:", *identity.Account)
+    fmt.Println("ARN       :", *identity.Arn)
+    fmt.Println("UserID    :", *identity.UserId)
 }
 
 // UploadToS3 sends data to S3