Collaboration diagram for com.mapd.tests.DumpRestoreConcurrencyTest:

Classes
class	ThreadDbDumpRestoreQueries

Public Member Functions
void	testConcurrency () throws Exception

Static Public Member Functions
static void	main (String[] args) throws Exception

Package Functions
List< String >	getDumpRestoreQueries (String table_identifier)

Static Package Attributes
static final Logger	logger = LoggerFactory.getLogger(DumpRestoreConcurrencyTest.class)

static String	tar_ball_path_prefix = "/tmp/_Orz__"

Private Member Functions
void	deleteFileIfExist (String file)

void	runTest () throws Exception

void	runSqlAsUser (String sql, HeavyDBTestClient user, int threadId) throws Exception

Detailed Description

Definition at line 28 of file DumpRestoreConcurrencyTest.java.

Member Function Documentation

void com.mapd.tests.DumpRestoreConcurrencyTest.deleteFileIfExist ( String file )

inlineprivate

Definition at line 42 of file DumpRestoreConcurrencyTest.java.

Referenced by com.mapd.tests.DumpRestoreConcurrencyTest.runTest().

                                               {
     try {
       Files.deleteIfExists((new File(file)).toPath());
     } catch (Exception e) {
       logger.error("While Deleting Archives Caught Exception: " + e.getMessage(), e);
     }
   }

Here is the caller graph for this function:

List<String> com.mapd.tests.DumpRestoreConcurrencyTest.getDumpRestoreQueries ( String table_identifier )

inlinepackage

Definition at line 53 of file DumpRestoreConcurrencyTest.java.

References com.mapd.tests.DumpRestoreConcurrencyTest.tar_ball_path_prefix.

Referenced by com.mapd.tests.DumpRestoreConcurrencyTest.runTest().

                                                               {
     return Arrays.asList("DROP TABLE IF EXISTS " + table_identifier + ";",
             "DROP TABLE IF EXISTS restored_" + table_identifier + ";",
             "CREATE TABLE " + table_identifier + " (v INT);",
             "INSERT INTO " + table_identifier + " VALUES (1),(2);",
             "DUMP TABLE " + table_identifier + " TO '" + tar_ball_path_prefix + "_"
                     + table_identifier + "';",
             "RESTORE TABLE restored_" + table_identifier + " FROM '"
                     + tar_ball_path_prefix + "_" + table_identifier + "';");
   }

Here is the caller graph for this function:

static void com.mapd.tests.DumpRestoreConcurrencyTest.main ( String[] args ) throws Exception

inlinestatic

Definition at line 31 of file DumpRestoreConcurrencyTest.java.

                                                           {
     DumpRestoreConcurrencyTest test = new DumpRestoreConcurrencyTest();
     test.testConcurrency();
   }

void com.mapd.tests.DumpRestoreConcurrencyTest.runSqlAsUser	(	String	sql,
		HeavyDBTestClient	user,
		int	threadId
	)		throws Exception

inlineprivate

Definition at line 128 of file DumpRestoreConcurrencyTest.java.

Referenced by com.mapd.tests.DumpRestoreConcurrencyTest.runTest().

                            {
     logger.info(threadId + " " + sql);
     user.runSql(sql);
   }

Here is the caller graph for this function:

void com.mapd.tests.DumpRestoreConcurrencyTest.runTest ( ) throws Exception

inlineprivate

Definition at line 64 of file DumpRestoreConcurrencyTest.java.

References com.mapd.tests.DumpRestoreConcurrencyTest.ThreadDbDumpRestoreQueries.archive, com.mapd.tests.DumpRestoreConcurrencyTest.deleteFileIfExist(), com.mapd.tests.DumpRestoreConcurrencyTest.getDumpRestoreQueries(), com.mapd.tests.DumpRestoreConcurrencyTest.ThreadDbDumpRestoreQueries.queries, com.mapd.tests.DumpRestoreConcurrencyTest.runSqlAsUser(), and com.mapd.tests.DumpRestoreConcurrencyTest.tar_ball_path_prefix.

Referenced by com.mapd.tests.DumpRestoreConcurrencyTest.testConcurrency().

                                           {
     List<ThreadDbDumpRestoreQueries> queriesPerThread =
             new ArrayList<ThreadDbDumpRestoreQueries>(Arrays.asList(
                     new ThreadDbDumpRestoreQueries(
                             tar_ball_path_prefix + "_aa", getDumpRestoreQueries("aa")),
                     new ThreadDbDumpRestoreQueries(
                             tar_ball_path_prefix + "_bb", getDumpRestoreQueries("bb")),
                     new ThreadDbDumpRestoreQueries(
                             tar_ball_path_prefix + "_cc", getDumpRestoreQueries("cc")),
                     new ThreadDbDumpRestoreQueries(
                             tar_ball_path_prefix + "_dd", getDumpRestoreQueries("dd")),
                     new ThreadDbDumpRestoreQueries(
                             tar_ball_path_prefix + "_ee", getDumpRestoreQueries("ee"))));
 
     final int num_threads = queriesPerThread.size();
     Exception[] exceptions = new Exception[num_threads];
     Thread[] threads = new Thread[num_threads];
 
     // Use a barrier here to synchronize the start of each competing thread and make
     // sure there is a race condition. The barrier ensures no thread will run until they
     // are all created and waiting for the barrier to complete.
     final CyclicBarrier barrier = new CyclicBarrier(
             num_threads, () -> { logger.info("Barrier acquired. Starting queries..."); });
 
     for (int i = 0; i < queriesPerThread.size(); i++) {
       final ThreadDbDumpRestoreQueries threadQueries = queriesPerThread.get(i);
       final int threadId = i;
       threads[threadId] = new Thread(() -> {
         try {
           logger.info("Starting thread[" + threadId + "]");
           HeavyDBTestClient user = HeavyDBTestClient.getClient(
                   "localhost", 6274, "heavyai", "admin", "HyperInteractive");
           deleteFileIfExist(threadQueries.archive);
           barrier.await();
           for (final String query : threadQueries.queries) {
             runSqlAsUser(query, user, threadId);
           }
           logger.info("Finished thread[" + threadId + "]");
         } catch (Exception e) {
           logger.error("Thread[" + threadId + "] Caught Exception: " + e.getMessage(), e);
           exceptions[threadId] = e;
         }
       });
       threads[threadId].start();
     }
 
     for (Thread t : threads) {
       t.join();
     }
 
     // clean up archive files
     for (int i = 0; i < queriesPerThread.size(); i++) {
       final ThreadDbDumpRestoreQueries threadQueries = queriesPerThread.get(i);
       deleteFileIfExist(threadQueries.archive);
     }
 
     for (Exception e : exceptions) {
       if (e != null) {
         logger.error("Exception: " + e.getMessage(), e);
         throw e;
       }
     }
   }

Here is the call graph for this function:

Here is the caller graph for this function:

void com.mapd.tests.DumpRestoreConcurrencyTest.testConcurrency ( ) throws Exception

inline

Definition at line 36 of file DumpRestoreConcurrencyTest.java.

References com.mapd.tests.DumpRestoreConcurrencyTest.runTest().

                                                  {
     logger.info("DumpRestoreConcurrencyTest()");
     runTest();
     logger.info("DumpRestoreConcurrencyTest() done");
   }