mysql80.py

# Copyright 2020 PerfKitBenchmarker Authors. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.


"""Module containing mysql installation and cleanup functions."""

import re

MYSQL_PSWD = 'perfkitbenchmarker'
PACKAGE_NAME = 'mysql'


def YumInstall(vm):
  """Installs the mysql package on the VM."""
  raise NotImplementedError


def AptInstall(vm):
  """Installs the mysql package on the VM."""
  vm.RemoteCommand('wget -c '
                   'https://repo.mysql.com//mysql-apt-config_0.8.17-1_all.deb')
  vm.RemoteCommand('echo mysql-apt-config mysql-apt-config/select-server'
                   ' select mysql-8.0 | sudo debconf-set-selections')
  vm.RemoteCommand('echo mysql-apt-config mysql-apt-config/select-product'
                   ' select Ok | sudo debconf-set-selections')
  vm.RemoteCommand('sudo -E DEBIAN_FRONTEND=noninteractive dpkg -i'
                   ' mysql-apt-config_0.8.17-1_all.deb')

  _, stderr = vm.RemoteCommand('sudo apt-get update', ignore_failure=True)

  if stderr:
    if 'public key is not available:' in stderr:
      # This error is due to mysql updated the repository and the public
      # key is not updated.
      # Import the updated public key
      match = re.match('.*NO_PUBKEY ([A-Z0-9]*)', stderr)
      if match:
        key = match.group(1)
        vm.RemoteCommand('sudo apt-key adv '
                         f'--keyserver keyserver.ubuntu.com --recv-keys {key}')
      else:
        raise RuntimeError('No public key found by regex.')
    else:
      raise RuntimeError(stderr)

  vm.RemoteCommand('echo "mysql-server-8.0 mysql-server/root_password password '
                   f'{MYSQL_PSWD}" | sudo debconf-set-selections')
  vm.RemoteCommand('echo "mysql-server-8.0 mysql-server/root_password_again '
                   f'password {MYSQL_PSWD}" | sudo debconf-set-selections')
  vm.InstallPackages('mysql-server')


def YumGetPathToConfig(vm):
  """Returns the path to the mysql config file."""
  raise NotImplementedError


def AptGetPathToConfig(vm):
  """Returns the path to the mysql config file."""
  del vm
  return '/etc/mysql/mysql.conf.d/mysqld.cnf'


def YumGetServiceName(vm):
  """Returns the name of the mysql service."""
  raise NotImplementedError


def AptGetServiceName(vm):
  """Returns the name of the mysql service."""
  del vm
  return 'mysql'