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; }