add flag.parse for assume role (awsdocs#2356)

cubiccompass · Nov 9, 2021 · 1fcf7ab · 1fcf7ab
1 parent 6236ad2
commit 1fcf7ab
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gov2/sts/AssumeRole/AssumeRolev2.go b/gov2/sts/AssumeRole/AssumeRolev2.go
@@ -35,6 +35,7 @@ func TakeRole(c context.Context, api STSAssumeRoleAPI, input *sts.AssumeRoleInpu
 func main() {
 	roleARN := flag.String("r", "", "The Amazon Resource Name (ARN) of the role to assume")
 	sessionName := flag.String("s", "", "The name of the session")
+	flag.Parse()
 
 	if *roleARN == "" || *sessionName == "" {
 		fmt.Println("You must supply a role ARN and session name")