@Override
public void prepare(GraphDatabaseService database, final Map<String, Object> params) {
generatorConfiguration =
new BasicGeneratorConfig(
new BarabasiAlbertRelationshipGenerator(
new BarabasiAlbertConfig((Integer) params.get(NETWORK_SIZE), 3)),
SocialNetworkNodeCreator.getInstance(),
SocialNetworkRelationshipCreator.getInstance()) {
@Override
public int getBatchSize() {
return (int) params.get(BATCH_SIZE);
}
};
}
private GeneratorConfiguration getGeneratorConfiguration(int numberOfNodes, int numberOfEdges) {
return new BasicGeneratorConfig(
new ErdosRenyiRelationshipGenerator(new ErdosRenyiConfig(numberOfNodes, numberOfEdges)),
SocialNetworkNodeCreator.getInstance(),
SocialNetworkRelationshipCreator.getInstance());
}