Hawkin
/
arangodb
mirror of https://gitee.com/bigwinds/arangodb
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix SSSP

This commit is contained in:
Simon Grätzer 2017-03-02 12:47:50 +01:00
parent 27098e9e4f
commit c74fd5b2d8
2 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
arangod/Pregel/Algos/SSSP.cpp
View File

@ -47,17 +47,20 @@ struct SSSPComputation : public VertexComputation<int64_t, int64_t, int64_t> {
int64_t val = *edge->data() + tmp;
sendMessage(edge, val);
}
} else {
voteHalt();
}
voteHalt();
}
};
uint32_t SSSPAlgorithm::messageBatchSize(WorkerConfig const& config,
MessageStats const& stats) const {
if (config.localSuperstep() <= 2) {
return 1;
if (config.localSuperstep() <= 1) {
return 5;
} else {
return Algorithm::messageBatchSize(config, stats);
double msgsPerSec = stats.sendCount / stats.superstepRuntimeSecs;
msgsPerSec /= config.parallelism(); // per thread
return msgsPerSec > 250.0 ? (uint32_t)msgsPerSec : 250;
}
}

6
arangod/Pregel/Worker.cpp
View File

@ -561,7 +561,11 @@ void Worker<V, E, M>::_finishedProcessing() {
/// in async mode checks if there are messages to process
template <typename V, typename E, typename M>
void Worker<V, E, M>::_continueAsync() {
if (_state == WorkerState::IDLE && _writeCache->containedMessageCount() > 0) {
uint64_t cnt = _writeCache->containedMessageCount();
if (_state == WorkerState::IDLE && cnt > 0) {
if (cnt < 10) {
usleep(10000);
}
{ // swap these pointers atomically
MY_WRITE_LOCKER(guard, _cacheRWLock);
std::swap(_readCache, _writeCache);