gpointer asyncpriorityqueue_peek(AsyncPriorityQueue *q) {
g_assert(q);
g_mutex_lock(&(q->lock));
gpointer returnData = priorityqueue_peek(q->pq);
g_mutex_unlock(&(q->lock));
return returnData;
}
static void _schedulerpolicyhoststeal_findMinTime(Host* host, HostStealSearchState* state) {
g_rw_lock_reader_lock(&state->data->lock);
HostStealQueueData* qdata = g_hash_table_lookup(state->data->hostToQueueDataMap, host);
g_rw_lock_reader_unlock(&state->data->lock);
utility_assert(qdata);
g_mutex_lock(&(qdata->lock));
Event* event = priorityqueue_peek(qdata->pq);
g_mutex_unlock(&(qdata->lock));
if(event != NULL) {
state->nextEventTime = MIN(state->nextEventTime, event_getTime(event));
}
}
static Event* _schedulerpolicyhoststeal_popFromThread(SchedulerPolicy* policy, HostStealThreadData* tdata, GQueue* assignedHosts, SimulationTime barrier) {
/* if there is no tdata, that means this thread didn't get any hosts assigned to it */
if(!tdata) {
return NULL;
}
HostStealPolicyData* data = policy->data;
while(!g_queue_is_empty(assignedHosts) || tdata->runningHost) {
/* if there's no running host, we completed the last assignment and need a new one */
if(!tdata->runningHost) {
tdata->runningHost = g_queue_pop_head(assignedHosts);
}
Host* host = tdata->runningHost;
g_rw_lock_reader_lock(&data->lock);
HostStealQueueData* qdata = g_hash_table_lookup(data->hostToQueueDataMap, host);
g_rw_lock_reader_unlock(&data->lock);
utility_assert(qdata);
g_mutex_lock(&(qdata->lock));
Event* nextEvent = priorityqueue_peek(qdata->pq);
SimulationTime eventTime = (nextEvent != NULL) ? event_getTime(nextEvent) : SIMTIME_INVALID;
if(nextEvent != NULL && eventTime < barrier) {
utility_assert(eventTime >= qdata->lastEventTime);
qdata->lastEventTime = eventTime;
nextEvent = priorityqueue_pop(qdata->pq);
qdata->nPopped++;
/* migrate iff a migration is needed */
_schedulerpolicyhoststeal_migrateHost(policy, host, pthread_self());
} else {
nextEvent = NULL;
}
if(nextEvent == NULL) {
/* no more events on the runningHost, mark it as NULL so we get a new one */
g_queue_push_tail(tdata->processedHosts, host);
tdata->runningHost = NULL;
}
g_mutex_unlock(&(qdata->lock));
if(nextEvent != NULL) {
return nextEvent;
}
}
/* if we make it here, all hosts for this thread have no more events before barrier */
return NULL;
}
