瀏覽代碼

Trying to speed up table truncate

pull/62/head
Martin Stockhammer 2 年之前
父節點
當前提交
d0aa5dd637

+ 2
- 1
archiva-modules/metadata/metadata-store-provider/metadata-store-cassandra/pom.xml 查看文件

@@ -396,7 +396,7 @@
<addTestClasspath>false</addTestClasspath>
<startWaitSeconds>500</startWaitSeconds>
<startNativeTransport>true</startNativeTransport>
<logLevel>DEBUG</logLevel>
<logLevel>INFO</logLevel>
<loadAfterFirstStart>false</loadAfterFirstStart>
<yaml>
broadcast_rpc_address: 127.0.0.1
@@ -404,6 +404,7 @@
<systemPropertyVariables>
<java.net.preferIPv4Stack>true</java.net.preferIPv4Stack>
</systemPropertyVariables>

</configuration>
</execution>
<execution>

+ 18
- 11
archiva-modules/metadata/metadata-store-provider/metadata-store-cassandra/src/test/java/org/apache/archiva/metadata/repository/cassandra/CassandraMetadataRepositoryTest.java 查看文件

@@ -44,7 +44,10 @@ import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionStage;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.stream.Collectors;

import static com.datastax.oss.driver.api.querybuilder.QueryBuilder.truncate;
import static com.datastax.oss.driver.api.querybuilder.SchemaBuilder.dropTable;
@@ -72,7 +75,7 @@ public class CassandraMetadataRepositoryTest

long cTime;
int testNum = 0;
AtomicBoolean clearedTables = new AtomicBoolean( false );
final AtomicBoolean clearedTables = new AtomicBoolean( false );


@Override
@@ -121,8 +124,7 @@ public class CassandraMetadataRepositoryTest

if (!clearedTables.get())
{
clearReposAndNamespace( cassandraArchivaManager );
clearedTables.set( true );
clearReposAndNamespace( cassandraArchivaManager, clearedTables );
}
System.err.println( "Finished setting up - "+testInfo.getDisplayName() + " - " + (System.currentTimeMillis( ) - cTime) +"ms");
}
@@ -164,13 +166,12 @@ public class CassandraMetadataRepositoryTest
throws Exception
{
System.err.println( "Shutting down - " + (testNum-1) + " - " + testInfo.getDisplayName( ) + " - " + ( System.currentTimeMillis( ) - cTime ) +"ms");
clearReposAndNamespace( cassandraArchivaManager );
clearedTables.set( true );
clearReposAndNamespace( cassandraArchivaManager, clearedTables );
super.tearDown();
System.err.println( "Shutting down finished - " + testInfo.getDisplayName( ) + " - " + ( System.currentTimeMillis( ) - cTime ) +"ms");
}

static void clearReposAndNamespace( CassandraArchivaManager cassandraArchivaManager )
static void clearReposAndNamespace( final CassandraArchivaManager cassandraArchivaManager, final AtomicBoolean clearedFlag )
throws Exception
{
if (cassandraArchivaManager!=null)
@@ -188,15 +189,21 @@ public class CassandraMetadataRepositoryTest
cassandraArchivaManager.getLicenseFamilyName( ),
cassandraArchivaManager.getDependencyFamilyName( )
);
for ( String table : tables )
{
session.execute( truncate( table ).build( ) );
}

CompletableFuture.allOf( tables.stream( ).map( table -> session.executeAsync( truncate( table ).build( ) ) )
.map( CompletionStage::toCompletableFuture ).collect( Collectors.toList( ) ).toArray( new CompletableFuture[0] ) )
.thenAccept( ( c ) -> {
if ( clearedFlag != null ) clearedFlag.set( true );
} ).get( )
;
}
} else {
System.err.println( "cassandraArchivaManager is null" );
}
}

static void clearReposAndNamespace( final CassandraArchivaManager cassandraArchivaManager)
throws Exception {
clearReposAndNamespace( cassandraArchivaManager, null );
}

}

Loading…
取消
儲存