Add --prepare option to migrate command

mgradm/cmd/install/kubernetes/utils.go

@@ -65,7 +65,7 @@ func installForKubernetes(globalFlags *types.GlobalFlags,
  helmArgs = append(helmArgs, sslArgs...)
 
  // Deploy Uyuni and wait for it to be up
- if err := kubernetes.Deploy(cnx, &flags.Image, &flags.Helm, &flags.Ssl, clusterInfos, fqdn, flags.Debug.Java, helmArgs...); err != nil {
+ if err := kubernetes.Deploy(cnx, &flags.Image, &flags.Helm, &flags.Ssl, clusterInfos, fqdn, flags.Debug.Java, false, helmArgs...); err != nil {
  return shared_utils.Errorf(err, L("cannot deploy uyuni"))
  }
 

mgradm/cmd/migrate/kubernetes/utils.go

@@ -52,7 +52,7 @@ func migrateToKubernetes(
  sshConfigPath, sshKnownhostsPath := migration_shared.GetSshPaths()
 
  // Prepare the migration script and folder
- scriptDir, err := adm_utils.GenerateMigrationScript(fqdn, flags.User, true)
+ scriptDir, err := adm_utils.GenerateMigrationScript(fqdn, flags.User, true, flags.Prepare)
  if err != nil {
  return utils.Errorf(err, L("failed to generate migration script"))
  }
@@ -71,7 +71,7 @@ func migrateToKubernetes(
  var sslFlags adm_utils.SslCertFlags
 
  // Deploy for running migration command
- if err := kubernetes.Deploy(cnx, &flags.Image, &flags.Helm, &sslFlags, clusterInfos, fqdn, false,
+ if err := kubernetes.Deploy(cnx, &flags.Image, &flags.Helm, &sslFlags, clusterInfos, fqdn, false, flags.Prepare,
  "--set", "migration.ssh.agentSocket="+sshAuthSocket,
  "--set", "migration.ssh.configPath="+sshConfigPath,
  "--set", "migration.ssh.knownHostsPath="+sshKnownhostsPath,
@@ -101,6 +101,11 @@ func migrateToKubernetes(
  return utils.Errorf(err, L("cannot set replicas to 0"))
  }
 
+ if flags.Prepare {
+ log.Info().Msg(L("Migration prepared. Run the 'migrate' command without '--prepare' to finish the migration."))
+ return nil
+ }
+
  defer func() {
  // if something is running, we don't need to set replicas to 1
  if _, err = shared_kubernetes.GetNode("uyuni"); err != nil {

mgradm/cmd/migrate/podman/utils.go

@@ -35,10 +35,14 @@ func migrateToPodman(globalFlags *types.GlobalFlags, flags *podmanMigrateFlags,
  sshAuthSocket := migration_shared.GetSshAuthSocket()
  sshConfigPath, sshKnownhostsPath := migration_shared.GetSshPaths()
 
- tz, oldPgVersion, newPgVersion, err := podman.RunMigration(serverImage, flags.Image.PullPolicy, sshAuthSocket, sshConfigPath, sshKnownhostsPath, sourceFqdn, flags.User)
+ tz, oldPgVersion, newPgVersion, err := podman.RunMigration(serverImage, flags.Image.PullPolicy, sshAuthSocket, sshConfigPath, sshKnownhostsPath, sourceFqdn, flags.User, flags.Prepare)
  if err != nil {
  return utils.Errorf(err, L("cannot run migration script"))
  }
+ if flags.Prepare {
+ log.Info().Msg(L("Migration prepared. Run the 'migrate' command without '--prepare' to finish the migration."))
+ return nil
+ }
 
  if oldPgVersion != newPgVersion {
  if err := podman.RunPgsqlVersionUpgrade(flags.Image, flags.DbUpgradeImage, oldPgVersion, newPgVersion); err != nil {

mgradm/cmd/migrate/shared/flags.go

@@ -13,6 +13,7 @@ import (
 
 // MigrateFlags represents flag required by migration command.
 type MigrateFlags struct {
+ Prepare bool
  Image types.ImageFlags `mapstructure:",squash"`
  DbUpgradeImage types.ImageFlags `mapstructure:"dbupgrade"`
  User string
@@ -21,6 +22,7 @@ type MigrateFlags struct {
 
 // AddMigrateFlags add migration flags to a command.
 func AddMigrateFlags(cmd *cobra.Command) {
+ cmd.Flags().Bool("prepare", false, L("Prepare the mgration - copy the data without stopping the source server."))
  utils.AddMirrorFlag(cmd)
  utils.AddImageFlag(cmd)
  utils.AddDbUpgradeImageFlag(cmd)

mgradm/shared/kubernetes/install.go

@@ -26,14 +26,16 @@ const HELM_APP_NAME = "uyuni"
 // Deploy execute a deploy of a given image and helm to a cluster.
 func Deploy(cnx *shared.Connection, imageFlags *types.ImageFlags,
  helmFlags *cmd_utils.HelmFlags, sslFlags *cmd_utils.SslCertFlags, clusterInfos *kubernetes.ClusterInfos,
- fqdn string, debug bool, helmArgs ...string) error {
+ fqdn string, debug bool, prepare bool, helmArgs ...string) error {
  // If installing on k3s, install the traefik helm config in manifests
  isK3s := clusterInfos.IsK3s()
  IsRke2 := clusterInfos.IsRke2()
- if isK3s {
- InstallK3sTraefikConfig(debug)
- } else if IsRke2 {
- kubernetes.InstallRke2NginxConfig(utils.TCP_PORTS, utils.UDP_PORTS, helmFlags.Uyuni.Namespace)
+ if !prepare {
+ if isK3s {
+ InstallK3sTraefikConfig(debug)
+ } else if IsRke2 {
+ kubernetes.InstallRke2NginxConfig(utils.TCP_PORTS, utils.UDP_PORTS, helmFlags.Uyuni.Namespace)
+ }
  }
 
  serverImage, err := utils.ComputeImage(*imageFlags)

mgradm/shared/podman/podman.go

@@ -177,8 +177,8 @@ func UpdateSslCertificate(cnx *shared.Connection, chain *ssl.CaChain, serverPair
 }
 
 // RunMigration migrate an existing remote server to a container.
-func RunMigration(serverImage string, pullPolicy string, sshAuthSocket string, sshConfigPath string, sshKnownhostsPath string, sourceFqdn string, user string) (string, string, string, error) {
- scriptDir, err := adm_utils.GenerateMigrationScript(sourceFqdn, user, false)
+func RunMigration(serverImage string, pullPolicy string, sshAuthSocket string, sshConfigPath string, sshKnownhostsPath string, sourceFqdn string, user string, prepare bool) (string, string, string, error) {
+ scriptDir, err := adm_utils.GenerateMigrationScript(sourceFqdn, user, false, prepare)
  if err != nil {
  return "", "", "", utils.Errorf(err, L("cannot generate migration script"))
  }

mgradm/shared/templates/migrateScriptTemplate.go

@@ -19,15 +19,22 @@ if test -e /tmp/ssh_config; then
 fi
 SSH="ssh -o User={{ .User }} -A $SSH_CONFIG "
 
+{{ if .Prepare }}
+echo "Preparing migration..."
+$SSH {{ .SourceFqdn }} "sudo systemctl start postgresql.service"
+{{ else }}
 echo "Stopping spacewalk service..."
 $SSH {{ .SourceFqdn }} "sudo spacewalk-service stop ; sudo systemctl start postgresql.service"
+{{ end }}
 
 $SSH {{ .SourceFqdn }} \
  "echo \"COPY (SELECT MIN(CONCAT(org_id, '-', label)) AS target, base_path FROM rhnKickstartableTree GROUP BY base_path) TO STDOUT WITH CSV;\" \
  |sudo spacewalk-sql --select-mode - " > distros
 
+{{ if not .Prepare }}
 echo "Stopping posgresql service..."
 $SSH {{ .SourceFqdn }} "sudo systemctl stop postgresql.service"
+{{ end }}
 
 while IFS="," read -r target path ; do
  echo "-/ $path"
@@ -144,6 +151,7 @@ type MigrateScriptTemplateData struct {
  SourceFqdn string
  User string
  Kubernetes bool
+ Prepare bool
 }
 
 // Render will create migration script.

mgradm/shared/utils/exec.go

@@ -131,7 +131,7 @@ func RunMigration(cnx *shared.Connection, tmpPath string, scriptName string) err
 }
 
 // GenerateMigrationScript generates the script that perform migration.
-func GenerateMigrationScript(sourceFqdn string, user string, kubernetes bool) (string, error) {
+func GenerateMigrationScript(sourceFqdn string, user string, kubernetes bool, prepare bool) (string, error) {
  scriptDir, err := os.MkdirTemp("", "mgradm-*")
  if err != nil {
  return "", utils.Errorf(err, L("failed to create temporary directory"))
@@ -142,6 +142,7 @@ func GenerateMigrationScript(sourceFqdn string, user string, kubernetes bool) (s
  SourceFqdn: sourceFqdn,
  User: user,
  Kubernetes: kubernetes,
+ Prepare: prepare,
  }
 
  scriptPath := filepath.Join(scriptDir, "migrate.sh")

uyuni-tools.changes.nadvornik.prepare

@@ -0,0 +1 @@
+- Add --prepare option to migrate command