From c81ef55cc1743d60b6cf6f4fcad849f6a1559fc4 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Milo=C5=A1=20Mula=C4=8D?= <mulac@civ.zcu.cz>
Date: Fri, 21 Mar 2008 13:57:14 +0000
Subject: [PATCH] further enhancements

---
 .../glite-lb-migrate_server_db2proxy_server_db     | 56 +++++++++++++++++++++-
 1 file changed, 54 insertions(+), 2 deletions(-)

diff --git a/org.glite.lb.server/config/glite-lb-migrate_server_db2proxy_server_db b/org.glite.lb.server/config/glite-lb-migrate_server_db2proxy_server_db
index 77382d1..c299334 100644
--- a/org.glite.lb.server/config/glite-lb-migrate_server_db2proxy_server_db
+++ b/org.glite.lb.server/config/glite-lb-migrate_server_db2proxy_server_db
@@ -5,6 +5,53 @@
 
 #!/bin/bash
 
+usage() {
+cat <<EOF
+NAME:
+	migrate - script for migrating older LB server or proxy databases
+		to new schema required for transaction locking and USP 
+		(United Server and Proxy)
+USAGE:
+	migrate -s|-p [-d db_name]
+
+		-s		migrate old server DB to new version
+		-p		migrate old proxy DB to new version
+		-d db_name	non-default database name
+EOF
+
+}
+
+SERVER=""
+PROXY=""
+
+while getopts "spd:h" OPTION 
+do
+    case "$OPTION" in 
+    "s") SERVER=1
+    ;;
+
+    "p") PROXY=1
+    ;;
+
+    "d") DB_NAME=$OPTARG
+    ;;
+
+    "h") usage;
+    ;;
+
+    esac
+done
+
+if [ -z "$SERVER" -a -z "$PROXY" ]; then
+	usage;
+fi
+
+if [ "$SERVER" -a "$PROXY" ]; then
+	usage;
+fi
+
+exit 1;
+
 
 # add new columns
 mysql -u lbserver lbserver20 -e "ALTER TABLE jobs ADD proxy bool not null"
@@ -15,8 +62,13 @@ mysql -u lbserver lbserver20 -e "ALTER TABLE jobs ADD nevents bool not null"
 
 
 # flag all jobs as server jobs
-mysql -u lbserver lbserver20 -e "UPDATE jobs SET proxy='0'"
-mysql -u lbserver lbserver20 -e "UPDATE jobs SET server='1'"
+if [ "$SERVER" ]; then
+	mysql -u lbserver lbserver20 -e "UPDATE jobs SET proxy='0'"
+	mysql -u lbserver lbserver20 -e "UPDATE jobs SET server='1'"
+else
+	mysql -u lbserver lbserver20 -e "UPDATE jobs SET proxy='1'"
+	mysql -u lbserver lbserver20 -e "UPDATE jobs SET server='0'"
+fi
 
 
 # for all greyjobs set a flag
-- 
1.8.2.3