Skip to content

Commit 2473bb9

Browse files
committed
[CLD-9031] Extend clusterStore interface to update pgbouncer configmap
1 parent a20ff12 commit 2473bb9

2 files changed

Lines changed: 5 additions & 9 deletions

File tree

internal/provisioner/eks_provisioner.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,8 @@ const EKSProvisionerType = "eks"
2929

3030
type clusterUpdateStore interface {
3131
UpdateCluster(cluster *model.Cluster) error
32+
GetMultitenantDatabases(filter *model.MultitenantDatabaseFilter) ([]*model.MultitenantDatabase, error)
33+
GetLogicalDatabases(filter *model.LogicalDatabaseFilter) ([]*model.LogicalDatabase, error)
3234
}
3335

3436
// EKSProvisioner provisions clusters using AWS EKS.

internal/provisioner/external_provisioner.go

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -121,24 +121,18 @@ func (provisioner *ExternalProvisioner) ProvisionCluster(cluster *model.Cluster)
121121
if cluster.HasAWSInfrastructure() {
122122
logger.Info("Provisioning resources for AWS infrastructure")
123123

124+
logger.Info("Deploying PgBouncer manifests")
124125
err = utility.DeployPgbouncerManifests(k8sClient, logger)
125126
if err != nil {
126127
return errors.Wrap(err, "failed to deploy pgbouncer manifests")
127128
}
128129

129-
vpc := cluster.ProvisionerMetadataExternal.VPC
130-
// Update PgBouncer ConfigMap
130+
vpc := cluster.VpcID()
131131
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
132132
defer cancel()
133133
logger.Info("Updating PgBouncer ConfigMap")
134134

135-
sqlStore, ok := provisioner.clusterUpdateStore.(*store.SQLStore)
136-
if !ok {
137-
logger.Warn("Store is not a SQLStore, skipping PgBouncer ConfigMap update")
138-
return nil
139-
}
140-
141-
err = pgbouncer.UpdatePGBouncerConfigMap(ctx, vpc, sqlStore, cluster.PgBouncerConfig, k8sClient, logger)
135+
err = pgbouncer.UpdatePGBouncerConfigMap(ctx, vpc, provisioner.clusterUpdateStore, cluster.PgBouncerConfig, k8sClient, logger)
142136
if err != nil {
143137
return errors.Wrap(err, "failed to update configmap for pgbouncer-configmap")
144138
}

0 commit comments

Comments
 (0)