support Oracle flash back; auto-fix metric type; compile ojdbc

client/driver/driver_config.py

@@ -68,6 +68,10 @@ PG_DATADIR = '/var/lib/postgresql/9.6/main'
 
 # ORACLE-SPECIFIC OPTIONS >>>
 ORACLE_AWR_ENABLED = False
+ORACLE_FLASH_BACK = True
+RESTORE_POINT = 'tpcc_point'
+RECOVERY_FILE_DEST = '/opt/oracle/oradata/ORCL'
+RECOVERY_FILE_DEST_SIZE = '15G'
 
 
 #==========================================================

client/driver/fabfile.py

@@ -415,15 +415,23 @@ def upload_batch(result_dir=None, sort=True, upload_code=None):
 @task
 def dump_database():
     dumpfile = os.path.join(dconf.DB_DUMP_DIR, dconf.DB_NAME + '.dump')
-    if file_exists(dumpfile):
+    if not dconf.ORACLE_FLASH_BACK and file_exists(dumpfile):
         LOG.info('%s already exists ! ', dumpfile)
         return False
 
-    LOG.info('Dump database %s to %s', dconf.DB_NAME, dumpfile)
+    if dconf.ORACLE_FLASH_BACK:
+        LOG.info('create restore point %s for database %s in %s', dconf.RESTORE_POINT,
+                 dconf.DB_NAME, dconf.RECOVERY_FILE_DEST)
+    else:
+        LOG.info('Dump database %s to %s', dconf.DB_NAME, dumpfile)
 
     if dconf.DB_TYPE == 'oracle':
-        run_sql_script('dumpOracle.sh', dconf.DB_USER, dconf.DB_PASSWORD,
-                       dconf.DB_NAME, dconf.DB_DUMP_DIR)
+        if dconf.ORACLE_FLASH_BACK:
+            run_sql_script('createRestore.sh', dconf.RESTORE_POINT,
+                           dconf.RECOVERY_FILE_DEST_SIZE, dconf.RECOVERY_FILE_DEST)
+        else:
+            run_sql_script('dumpOracle.sh', dconf.DB_USER, dconf.DB_PASSWORD,
+                           dconf.DB_NAME, dconf.DB_DUMP_DIR)
 
     elif dconf.DB_TYPE == 'postgres':
         run('PGPASSWORD={} pg_dump -U {} -h {} -F c -d {} > {}'.format(
@@ -434,19 +442,28 @@ def dump_database():
     return True
 
 
+@task
+def clean_recovery():
+    run_sql_script('removeRestore.sh', dconf.RESTORE_POINT)
+    cmds = ("""rman TARGET / <<EOF\nDELETE ARCHIVELOG ALL;\nexit\nEOF""")
+    run(cmds)
+
+
 @task
 def restore_database():
     dumpfile = os.path.join(dconf.DB_DUMP_DIR, dconf.DB_NAME + '.dump')
-    if not file_exists(dumpfile):
+    if not dconf.ORACLE_FLASH_BACK and not file_exists(dumpfile):
         raise FileNotFoundError("Database dumpfile '{}' does not exist!".format(dumpfile))
 
     LOG.info('Start restoring database')
     if dconf.DB_TYPE == 'oracle':
-        # You must create a directory named dpdata through sqlplus in your Oracle database
-        # The following script assumes such directory exists.
-        drop_user()
-        create_user()
-        run_sql_script('restoreOracle.sh', dconf.DB_USER, dconf.DB_NAME)
+        if dconf.ORACLE_FLASH_BACK:
+            run_sql_script('flashBack.sh', dconf.RESTORE_POINT)
+            clean_recovery()
+        else:
+            drop_user()
+            create_user()
+            run_sql_script('restoreOracle.sh', dconf.DB_USER, dconf.DB_NAME)
     elif dconf.DB_TYPE == 'postgres':
         drop_database()
         create_database()

client/driver/oracleScripts/archiveLog.sh

@@ -0,0 +1,11 @@
+#!/bin/sh
+
+# Change log mode
+sqlplus / as sysdba <<EOF
+    shutdown immediate
+    startup mount
+    ALTER DATABASE ARCHIVELOG;
+    ALTER DATABASE OPEN;
+quit
+EOF
+

client/driver/oracleScripts/createRestore.sh

@@ -0,0 +1,33 @@
+#!/bin/sh
+
+RESTORE_POINT="$1"
+SIZE="$2"
+RECOVERY_FILE="$3"
+
+# Make sure the physical directory exists
+mkdir -p "$DP_PATH"
+
+# Set recovery file and restore point
+sqlplus / as sysdba <<EOF
+    DECLARE
+      rfdexists INTEGER;
+    BEGIN
+      SELECT COUNT(*) INTO rfdexists FROM V\$RECOVERY_FILE_DEST WHERE name='$RECOVERY_FILE';
+      IF (rfdexists = 0) then
+        EXECUTE IMMEDIATE 'ALTER SYSTEM SET DB_RECOVERY_FILE_DEST_SIZE = $SIZE';
+        EXECUTE IMMEDIATE 'ALTER SYSTEM SET DB_RECOVERY_FILE_DEST = $RECOVERY_FILE';
+      END IF;
+      DBMS_OUTPUT.PUT_LINE(rfdexists);
+    END;
+    DECLARE
+      rpexists INTEGER;
+    BEGIN
+      SELECT COUNT(*) INTO rpexists FROM v\\\\\$restore_point WHERE name='$RESTORE_POINT';
+      IF (rpexists = 0) then
+        EXECUTE IMMEDIATE 'CREATE RESTORE POINT $RESTORE_POINT GUARANTEE FLASHBACK DATABASE';
+      END IF;
+      DBMS_OUTPUT.PUT_LINE(rpexists);
+    END;
+quit
+EOF
+

client/driver/oracleScripts/flashBack.sh

@@ -0,0 +1,13 @@
+#!/bin/sh
+
+RESTORE_POINT="$1"
+
+# Flash back
+sqlplus / as sysdba <<EOF
+    SHUTDOWN IMMEDIATE
+    STARTUP MOUNT
+    FLASHBACK DATABASE TO RESTORE POINT $RESTORE_POINT;
+    ALTER DATABASE OPEN RESETLOGS;
+quit
+EOF
+

client/driver/oracleScripts/removeRestore.sh

@@ -0,0 +1,10 @@
+#!/bin/sh
+
+RESTORE_POINT="$1"
+
+# Remove RESTORE Point
+sqlplus / as sysdba <<EOF
+    'DROP RESTORE POINT $RESTORE_POINT
+quit
+EOF
+