controllers/migration: add revisions_schema (#51)

ariga · Jun 15, 2023 · 611e328 · 611e328
1 parent 3625866
commit 611e328
Show file tree

Hide file tree

Showing 5 changed files with 19 additions and 4 deletions.
diff --git a/api/v1alpha1/atlasmigration_types.go b/api/v1alpha1/atlasmigration_types.go
@@ -39,6 +39,8 @@ type AtlasMigrationSpec struct {
 	Cloud Cloud `json:"cloud,omitempty"`
 	// Dir defines the directory to use for migrations as a configmap key reference.
 	Dir Dir `json:"dir"`
+	// RevisionsSchema defines the schema that revisions table resides in
+	RevisionsSchema string `json:"revisionsSchema,omitempty"`
 }
 
 // Cloud defines the Atlas Cloud configuration.

diff --git a/charts/atlas-operator/templates/crds/crd.yaml b/charts/atlas-operator/templates/crds/crd.yaml
@@ -93,6 +93,10 @@ spec:
                 description: EnvName sets the environment name used for reporting
                   runs to Atlas Cloud.
                 type: string
+              revisionsSchema:
+                description: RevisionsSchema defines the schema that revisions table
+                  resides in
+                type: string
               url:
                 description: URL of the target database schema.
                 type: string

diff --git a/config/crd/bases/db.atlasgo.io_atlasmigrations.yaml b/config/crd/bases/db.atlasgo.io_atlasmigrations.yaml
@@ -94,6 +94,10 @@ spec:
                 description: EnvName sets the environment name used for reporting
                   runs to Atlas Cloud.
                 type: string
+              revisionsSchema:
+                description: RevisionsSchema defines the schema that revisions table
+                  resides in
+                type: string
               url:
                 description: URL of the target database schema.
                 type: string

diff --git a/controllers/atlasmigration_controller.go b/controllers/atlasmigration_controller.go
@@ -74,10 +74,11 @@ func NewAtlasMigrationReconciler(mgr manager.Manager, cli MigrateCLI) *AtlasMigr
 // that will be used for Atlas CLI
 type (
 	atlasMigrationData struct {
-		EnvName   string
-		URL       string
-		Migration *migration
-		Cloud     *cloud
+		EnvName         string
+		URL             string
+		Migration       *migration
+		Cloud           *cloud
+		RevisionsSchema string
 	}
 
 	migration struct {
@@ -271,6 +272,7 @@ func (r *AtlasMigrationReconciler) extractMigrationData(
 		tmplData.EnvName = "kubernetes"
 	}
 
+	tmplData.RevisionsSchema = am.Spec.RevisionsSchema
 	return tmplData, cleanUpDir, nil
 }
 

diff --git a/controllers/templates/atlas_migration.tmpl b/controllers/templates/atlas_migration.tmpl
@@ -28,6 +28,9 @@ env {
 {{- end }}
 {{- with .Migration }}
     dir = "{{ .Dir }}"
+{{- end }}
+{{- if .RevisionsSchema }}
+    revisions_schema = "{{ .RevisionsSchema }}" 
 {{- end }}
   }
 }