Support CRUD operations for DR (#142)

yugabyte · Dec 10, 2024 · 79634ed · 79634ed
1 parent 4086375
commit 79634ed
Show file tree

Hide file tree

Showing 7 changed files with 507 additions and 3 deletions.
diff --git a/go.mod b/go.mod
@@ -8,7 +8,7 @@ require (
 	github.com/hashicorp/terraform-plugin-framework-validators v0.4.0
 	github.com/hashicorp/terraform-plugin-log v0.9.0
 	github.com/sethvargo/go-retry v0.2.3
-	github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20241111184456-64fb00419bec
+	github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20241121131757-c39c1074998a
 )
 
 require (

diff --git a/go.sum b/go.sum
@@ -244,8 +244,8 @@ github.com/vmihailenco/msgpack/v5 v5.3.5 h1:5gO0H1iULLWGhs2H5tbAHIZTV8/cYafcFOr9
 github.com/vmihailenco/msgpack/v5 v5.3.5/go.mod h1:7xyJ9e+0+9SaZT0Wt1RGleJXzli6Q/V5KbhBonMG9jc=
 github.com/vmihailenco/tagparser/v2 v2.0.0 h1:y09buUbR+b5aycVFQs/g70pqKVZNBmxwAhO7/IwNM9g=
 github.com/vmihailenco/tagparser/v2 v2.0.0/go.mod h1:Wri+At7QHww0WTrCBeu4J6bNtoV6mEfg5OIWRZA9qds=
-github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20241111184456-64fb00419bec h1:VLtUGFreqNxI54GgSuwypYHl+MA8aqxl22bLWpxAP0w=
-github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20241111184456-64fb00419bec/go.mod h1:5vW0xIzIZw+1djkiWKx0qqNmqbRBSf4mjc4qw8lIMik=
+github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20241121131757-c39c1074998a h1:eXXFUXaaoOuuRKz++a5ndNsqFAscPC0fgQzEwiZ4C1o=
+github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20241121131757-c39c1074998a/go.mod h1:5vW0xIzIZw+1djkiWKx0qqNmqbRBSf4mjc4qw8lIMik=
 github.com/yuin/goldmark v1.1.25/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 github.com/yuin/goldmark v1.1.32/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=

diff --git a/managed/fflags/feature_flags.go b/managed/fflags/feature_flags.go
@@ -16,12 +16,14 @@ const (
 	GOOGLECLOUD_INTEGRATION_ENABLED FeatureFlag = "GOOGLECLOUD_INTEGRATION_ENABLED"
 	DB_AUDIT_LOGGING                FeatureFlag = "DB_AUDIT_LOGGING"
 	CONNECTION_POOLING              FeatureFlag = "CONNECTION_POOLING"
+	DR                              FeatureFlag = "DR"
 )
 
 var flagEnabled = map[FeatureFlag]bool{
 	GOOGLECLOUD_INTEGRATION_ENABLED: false,
 	DB_AUDIT_LOGGING:                false,
 	CONNECTION_POOLING:              false,
+	DR:                              false,
 }
 
 func (f FeatureFlag) String() string {

diff --git a/managed/models.go b/managed/models.go
@@ -391,3 +391,13 @@ type DbQueryLoggingConfig struct {
 	ConfigID        types.String `tfsdk:"config_id"`
 	LogConfig       *LogConfig   `tfsdk:"log_config"`
 }
+
+type DrConfig struct {
+	AccountId       types.String   `tfsdk:"account_id"`
+	ProjectId       types.String   `tfsdk:"project_id"`
+	DrConfigId      types.String   `tfsdk:"dr_config_id"`
+	Name            types.String   `tfsdk:"name"`
+	SourceClusterId types.String   `tfsdk:"source_cluster_id"`
+	TargetClusterId types.String   `tfsdk:"target_cluster_id"`
+	Databases       []types.String `tfsdk:"databases"`
+}
diff --git a/managed/provider.go b/managed/provider.go
@@ -165,6 +165,11 @@ func (p *provider) GetResources(_ context.Context) (map[string]tfsdk.ResourceTyp
 		resources["ybm_db_audit_logging"] = resourceDbAuditLoggingType{}
 	}
 
+	// Add DR config resource only if the feature flag is enabled
+	if fflags.IsFeatureFlagEnabled(fflags.DR) {
+		resources["ybm_dr_config"] = resourceDrConfigType{}
+	}
+
 	return resources, nil
 }