Skip to content

Commit

Permalink
[#15,!11] Addressed review comments
Browse files Browse the repository at this point in the history
    Mostly added commentary.
  • Loading branch information
tmarkwalder committed Sep 5, 2018
1 parent 8d04ad0 commit ca63bff
Show file tree
Hide file tree
Showing 2 changed files with 59 additions and 10 deletions.
6 changes: 3 additions & 3 deletions src/bin/admin/tests/cql_tests.sh.in
Original file line number Diff line number Diff line change
Expand Up @@ -289,7 +289,7 @@ cql_upgrade_schema_to_version() {

# Verifies that you can upgrade from an earlier version and
# that unused subnet ID values in hosts and options tables are
# converted to -1
# converted to -1
cql_unused_subnet_id_test() {
test_start "cql.unused_subnet_id_test"

Expand All @@ -299,7 +299,7 @@ cql_unused_subnet_id_test() {
# We need to create an older database with lease data so we can
# verify the upgrade mechanisms which convert subnet id values
#
# Initialize database to scheme 1.0.
# Initialize database to schema 1.0.
cql_execute_script @abs_top_srcdir@/src/bin/admin/tests/dhcpdb_create_1.0.cql

# Now upgrade to schema 2.0, the version just before global HRs
Expand Down Expand Up @@ -342,7 +342,7 @@ cql_unused_subnet_id_test() {
host_reservations where id in(1,2,3,4,5);"

cql_execute "$qry" >$export_file
assert_eq 0 $? "insert hosts failed, expected exit code: %d, actual: %d"
assert_eq 0 $? "insert hosts failed, expected exit code: %d, actual: %d"
# Compare the dump output to reference file, they should be identical.
cmp -s $export_file $ref_file
assert_eq 0 $? "export file does not match reference file, expected exit code %d, actual %d"
Expand Down
63 changes: 56 additions & 7 deletions src/share/database/scripts/cql/upgrade_2.0_to_3.0.sh.in
Original file line number Diff line number Diff line change
Expand Up @@ -20,15 +20,17 @@ fi

cqlargs=$@

# Ensures the current schema version is 2.0. If not it exits.
check_version() {
version=$(cql_version $cqlargs)

if [ "${version}" != "2.0" ]; then
printf "This script upgrades 2.0 to 3.0. Reported version is %s. Skipping upgrade.\n" "${version}"
exit 0
exit 0
fi
}

# Peforms the schema changes from 2.0 to 3.0
update_schema() {
cqlsh $cqlargs <<EOF
-- This line starts database upgrade to version 3.0
Expand Down Expand Up @@ -63,10 +65,11 @@ EOF
if [ $? -ne 0 ]
then
echo Schema udpate FAILED!
exit -1
fi
exit -1
fi
}

# Function to delete temporary migration files
clean_up() {
# clean up the files
if [ -e $export_file ]
Expand All @@ -80,6 +83,13 @@ clean_up() {
fi
}

# Function to clean up and exit the script gracefull
#
# Called by migrate_host_data()
#
# Parameters:
# status - integer value to pass to sh:exit
# explanation - "quoted" text message to emit to stdout
exit_now() {
status=$1;shift
explanation=$1
Expand All @@ -95,24 +105,63 @@ exit_now() {
exit $status
}

# Function adds a column to the global, $update_cols if needed
#
# Called by migrate_host_data() to determine if the given column
# value needs to be updated, and if so appends CQL update text
# to a global string accumulator, $update_cols.
#
# The accumlator text is of the form:
#
# "<column_name>=<column value>{,<column_name>=<column_value>,..}"
#
# Parameters:
# val - current numerical value of the subnet ID column in question
# col - column name of the column in question
#
check_column() {
local val="$1";shift
local col="$1"
local old_id="0"
local new_id="-1"
local comma=""

# If the current value equals the value to be replaced
# add it to the accumulator
if [ "$val" = "$old_id" ]
then
# If the accumulator isn't empty, we need a comma
if [ ! -z "$update_cols" ]
then
comma=","
fi

update_cols="$update_cols$comma $col = $new_id"
fi
fi
}

# This function converts subnet ID columns in of existing host_reservations
# from 0 to 0xFFFFFFFF (SUBNET_ID_UNUSED). To do this it first creates an
# excerpt from host_reservations containing one row for each reservation
# with the following columns:
#
# id (primary key)
# host_ipv4_subnet_id
# host_ipv6_subnet_id
# option_subnet_id
#
# The excerpt is captured in a temporary file, the "export" file.
#
# Next, it iterates over the export file emitting a CQL update statement
# for each row that needs at least one of the columns updated. In other
# words, if at least one of the subnet ID columns in a row is 0, an
# update statement for that row will be emitted. The update statements
# are captured in a second temprory file, the "update" file.
#
# After exhausting the export file, the update file is submitted to
# cqlsh for execution.
#
# No parameters.
migrate_host_data() {
export_file="$temp_file_dir/cql_export.csv"
update_file="$temp_file_dir/cql_update.cql"
Expand Down Expand Up @@ -142,7 +191,7 @@ migrate_host_data() {
let line_cnt++;
update_cols=""
xIFS="$IFS"
IFS=$'\r,'
IFS=$'\r,'
i=1
# Parse the column values
for val in $line
Expand All @@ -168,7 +217,7 @@ migrate_host_data() {
let i++
done

if [ $i -ne 5 ]
if [ $i -ne 5 ]
then
# We're going to assume that since any error is fatal
exit_now -1 "Line# $line_cnt, too few values, wrong or corrupt file"
Expand All @@ -187,7 +236,7 @@ migrate_host_data() {
# If we didn't record any updates, then hey, we're good to go!
if [ $update_cnt == 0 ]
then
exit_now 0 "Completedly successfully: No updates were needed"
exit_now 0 "Completed successfully: No updates were needed"
fi

# We have at least one update in the update file, so submit it # to cqlsh.
Expand Down

0 comments on commit ca63bff

Please sign in to comment.