From 339a6e5a518b7c9828eefe5552f30d29c6b0b82a Mon Sep 17 00:00:00 2001
From: bengreenberg5 <bengreenberg5@gmail.com>
Date: Wed, 30 Jul 2025 11:42:45 -0400
Subject: [PATCH] Add enable_data_source_optimizations variable

---
 main.tf      | 4 ++--
 variables.tf | 6 ++++++
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/main.tf b/main.tf
index 48207dc238..a4bb4e5a4d 100644
--- a/main.tf
+++ b/main.tf
@@ -18,8 +18,8 @@ data "aws_iam_session_context" "current" {
 locals {
   create = var.create && var.putin_khuylo
 
-  account_id = try(data.aws_caller_identity.current[0].account_id, "")
-  partition  = try(data.aws_partition.current[0].partition, "")
+  account_id = var.enable_data_source_optimizations ? try(data.aws_caller_identity.current[0].account_id, "") : ""
+  partition  = var.enable_data_source_optimizations ? try(data.aws_partition.current[0].partition, "") : ""
 
   role_arn = try(aws_iam_role.this[0].arn, var.iam_role_arn)
 
diff --git a/variables.tf b/variables.tf
index 5e23df6c98..c6db8b3b38 100644
--- a/variables.tf
+++ b/variables.tf
@@ -22,6 +22,12 @@ variable "tags" {
   default     = {}
 }
 
+variable "enable_data_source_optimizations" {
+  description = "Enable data source optimizations to reduce API calls."
+  type        = bool
+  default     = true
+}
+
 ################################################################################
 # Cluster
 ################################################################################