build/google-cloud-workload-agent-migration.ps1

$ErrorActionPreference = 'Stop'
if ($env:ProgramData -eq $null -or $env:ProgramData -eq '') {
  $DATA_DIR = 'C:\Program Files\Google\google-cloud-workload-agent'
}
else {
  $DATA_DIR = $env:ProgramData + '\Google\google-cloud-workload-agent'
}
$INSTALL_DIR = 'C:\Program Files\Google\google-cloud-workload-agent'
$BIN_NAME_EXE = 'google-cloud-workload-agent.exe'
$LOGS_DIR = "$DATA_DIR\logs"
$LOG_FILE ="$LOGS_DIR\google-cloud-workload-agent-migration.log"

function Log-Write {
  #.DESCRIPTION
  #  Writes to log file.
  param (
    [string] $log_message
  )
  Write-Host $log_message
  if (-not (Test-Path $LOGS_DIR)) {
    return
  }
  $time_stamp = Get-Date -Format 'yyyy-MM-dd HH:mm:ss'
  $logFileSize = $(Get-Item $LOG_FILE -ErrorAction Ignore).Length/1kb
  if ($logFileSize -ge 1024) {
    Write-Host "Logfilesize: $logFileSize kb, rotating"
    Move-Item -Force $LOG_FILE "$LOG_FILE.1"
  }
  Add-Content -Value ("$time_stamp - $log_message") -path $LOG_FILE
}

function Log-Migrated {
  #.DESCRIPTION
  #  Invokes the service with usage logging enabled to log an migrated action.
  Start-Process $INSTALL_DIR\$BIN_NAME_EXE -ArgumentList 'logusage','-s','ACTION', '-a', 1 | Wait-Process -Timeout 30
}

function Migrate-Configuration {
  #.DESCRIPTION
  #  Migrates the configuration file from google-cloud-sql-server-agent to google-cloud-workload-agent.
  Start-Process $INSTALL_DIR\$BIN_NAME_EXE -ArgumentList 'migrate' | Wait-Process -Timeout 30
}

try {
  Log-Write 'Updating google-cloud-sql-server-agent to the latest version'
  # update google-cloud-sql-server-agent to the latest version
  googet -noconfirm install google-cloud-sql-server-agent
  Log-Write 'google-cloud-sql-server-agent updated'

  Log-Write 'Backing up the old configuration file'
  # copy and rename the old configuration file
  Copy-Item -Path 'C:\Program Files\Google\google-cloud-sql-server-agent\configuration.json' -Destination 'C:\Program Files\Google\google-cloud-workload-agent\conf\cfg_sqlserver_backup.json'
  Log-Write 'Backup completed'

  Log-Write 'Migrating the old configuration file'
  # run the migration script
  Migrate-Configuration
  Log-Write 'Migration completed'

  Log-Write 'Removing google-cloud-sql-server-agent'
  # uninstall google-cloud-sql-server-agent
  googet -noconfirm remove google-cloud-sql-server-agent
  Log-Write 'google-cloud-sql-server-agent removed'

  Log-Write 'Restarting google-cloud-workload-agent service'
  Restart-Service -Force 'google-cloud-workload-agent'
  Log-Write 'Agent for Compute Workloads service restarted'

  Log-Migrated
}
catch {
  Log-Write $_.Exception|Format-List -force | Out-String
  break
}
finally {
}