mirror of
https://github.com/goharbor/harbor.git
synced 2024-12-21 08:07:59 +01:00
106 lines
3.0 KiB
Bash
106 lines
3.0 KiB
Bash
#!/bin/bash
|
|
# Copyright Project Harbor Authors
|
|
#
|
|
# 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.
|
|
#
|
|
|
|
set -e
|
|
|
|
DBCNF="-hlocalhost -u${DB_USR}"
|
|
|
|
function launch_mysql {
|
|
set +e
|
|
local usr="$1"
|
|
local pwd="$2"
|
|
if [ ! -z "$pwd" ]; then
|
|
export MYSQL_PWD="${DB_PWD}"
|
|
fi
|
|
echo 'Trying to start mysql server...'
|
|
chown -R 10000:10000 /var/lib/mysql
|
|
mysqld &
|
|
echo 'Waiting for MySQL start...'
|
|
for i in {60..0}; do
|
|
if [ -z "$pwd" ]; then
|
|
mysqladmin -u$usr processlist >/dev/null 2>&1
|
|
else
|
|
mysqladmin -u$usr -p$pwd processlist >/dev/null 2>&1
|
|
fi
|
|
if [ $? -eq 0 ]; then
|
|
break
|
|
fi
|
|
sleep 1
|
|
done
|
|
set -e
|
|
if [ "$i" -eq 0 ]; then
|
|
echo "timeout. Can't run mysql server."
|
|
return 1
|
|
fi
|
|
return 0
|
|
}
|
|
|
|
function test_mysql {
|
|
set +e
|
|
launch_mysql $DB_USR $DB_PWD
|
|
if [ $? -eq 0 ]; then
|
|
echo "DB test failed."
|
|
exit 0
|
|
else
|
|
echo "DB test success."
|
|
exit 1
|
|
fi
|
|
set -e
|
|
}
|
|
|
|
function stop_mysql {
|
|
if [ -z $2 ]; then
|
|
mysqladmin -u$1 shutdown
|
|
else
|
|
mysqladmin -u$1 -p$DB_PWD shutdown
|
|
fi
|
|
sleep 1
|
|
}
|
|
|
|
function get_version_mysql {
|
|
local cur_version=""
|
|
set +e
|
|
if [[ $(mysql $DBCNF -N -s -e "select count(*) from information_schema.tables \
|
|
where table_schema='registry' and table_name='alembic_version';") -eq 0 ]]; then
|
|
echo "table alembic_version does not exist. Trying to initial alembic_version."
|
|
mysql $DBCNF < /harbor-migration/db/alembic/mysql/alembic.sql
|
|
#compatible with version 0.1.0 and 0.1.1
|
|
if [[ $(mysql $DBCNF -N -s -e "select count(*) from information_schema.tables \
|
|
where table_schema='registry' and table_name='properties'") -eq 0 ]]; then
|
|
echo "table properties does not exist. The version of registry is 0.1.0"
|
|
cur_version='0.1.0'
|
|
else
|
|
echo "The version of registry is 0.1.1"
|
|
mysql $DBCNF -e "insert into registry.alembic_version values ('0.1.1')"
|
|
cur_version='0.1.1'
|
|
fi
|
|
else
|
|
cur_version=$(mysql $DBCNF -N -s -e "select * from registry.alembic_version;")
|
|
fi
|
|
set -e
|
|
echo $cur_version
|
|
}
|
|
|
|
# It's only for registry, leverage the code from 1.5.0
|
|
function backup_mysql {
|
|
mysqldump $DBCNF --add-drop-database --databases registry > /harbor-migration/backup/registry.sql
|
|
}
|
|
|
|
# It's only for registry, leverage the code from 1.5.0
|
|
function restore_mysql {
|
|
mysql $DBCNF < /harbor-migration/backup/registry.sql
|
|
}
|