From d95bd702aaf1f4aed834b50f071a4022ab6b7e76 Mon Sep 17 00:00:00 2001
From: Hanwen <hanwenli@amazon.com>
Date: Thu, 25 Jan 2024 07:07:15 -0800
Subject: [PATCH] [Temporary] Download MySQL

Signed-off-by: Hanwen <hanwenli@amazon.com>
---
 .../mysql_client/partial/_setup_rhel_based.rb | 32 +++++++++++--------
 1 file changed, 19 insertions(+), 13 deletions(-)

diff --git a/cookbooks/aws-parallelcluster-slurm/resources/mysql_client/partial/_setup_rhel_based.rb b/cookbooks/aws-parallelcluster-slurm/resources/mysql_client/partial/_setup_rhel_based.rb
index 4cd5d3aa14..4fb731f141 100644
--- a/cookbooks/aws-parallelcluster-slurm/resources/mysql_client/partial/_setup_rhel_based.rb
+++ b/cookbooks/aws-parallelcluster-slurm/resources/mysql_client/partial/_setup_rhel_based.rb
@@ -15,31 +15,37 @@
 
 action :setup do
   mysql_archive_url = package_archive(node['cluster']['artifacts_s3_url'])
-  mysql_tar_file = "/tmp/#{package_filename}"
+  mysql_tar_file = "/tmp/" # TODO: upload mysql to S3 and revert the code
 
   log "Downloading MySQL packages archive from #{mysql_archive_url}"
 
   # Add MySQL source file
   action_create_source_link
-
-  remote_file mysql_tar_file do
-    source mysql_archive_url
-    mode '0644'
-    retries 3
-    retry_delay 5
-    action :create_if_missing
+  platform_version = node['platform_version'].to_i
+  if platform_version == 2
+    platform_version = 7
+  end
+  packages = %w(common client-plugins libs devel)
+  if platform_version == 7
+    packages = %w(common client-plugins libs-compat libs devel)
+  end
+  packages.each do |package|
+    file_name = "mysql-community-#{package}-#{package_version}.el#{platform_version}.#{arm_instance? ? 'aarch64' : 'x86_64'}.rpm"
+    remote_file "/tmp/#{file_name}" do
+      source "https://dev.mysql.com/get/Downloads/MySQL-8.0/#{file_name}"
+      mode '0644'
+      retries 3
+      retry_delay 5
+      action :create_if_missing
+    end
   end
-
   bash 'Install MySQL packages' do
     user 'root'
     group 'root'
     cwd '/tmp'
     code <<-MYSQL
         set -e
-
-        EXTRACT_DIR=$(mktemp -d --tmpdir mysql.XXXXXXX)
-        tar xf "#{mysql_tar_file}" --directory "${EXTRACT_DIR}"
-        yum install -y ${EXTRACT_DIR}/*
+        yum install -y mysql-community-*
     MYSQL
   end
 end