squiid/rust
1
0
Fork 0
Code Issues Pull requests Projects Activity

rt: Untabify rust_port_selector.cpp

Browse source
This commit is contained in:
Brian Anderson 2012-02-15 19:29:18 -08:00
parent 96af6f05a1
commit 75503570c7
1 changed files with 32 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

64
src/rt/rust_port_selector.cpp
View file

@ -7,8 +7,8 @@ rust_port_selector::rust_port_selector()
void void
rust_port_selector::select(rust_task *task, rust_port **dptr, rust_port_selector::select(rust_task *task, rust_port **dptr,
rust_port **ports, rust_port **ports,
size_t n_ports, uintptr_t *yield) { size_t n_ports, uintptr_t *yield) {
I(task->thread, this->ports == NULL); I(task->thread, this->ports == NULL);
I(task->thread, this->n_ports == 0); I(task->thread, this->n_ports == 0);
@ -29,33 +29,33 @@ rust_port_selector::select(rust_task *task, rust_port **dptr,
// Start looking for ports from a different index each time. // Start looking for ports from a different index each time.
size_t j = isaac_rand(&task->thread->rctx); size_t j = isaac_rand(&task->thread->rctx);
for (size_t i = 0; i < n_ports; i++) { for (size_t i = 0; i < n_ports; i++) {
size_t k = (i + j) % n_ports; size_t k = (i + j) % n_ports;
rust_port *port = ports[k]; rust_port *port = ports[k];
I(task->thread, port != NULL); I(task->thread, port != NULL);
port->lock.lock(); port->lock.lock();
locks_taken++; locks_taken++;
if (port->buffer.size() > 0) { if (port->buffer.size() > 0) {
*dptr = port; *dptr = port;
found_msg = true; found_msg = true;
break; break;
} }
} }
if (!found_msg) { if (!found_msg) {
this->ports = ports; this->ports = ports;
this->n_ports = n_ports; this->n_ports = n_ports;
I(task->thread, task->rendezvous_ptr == NULL); I(task->thread, task->rendezvous_ptr == NULL);
task->rendezvous_ptr = (uintptr_t*)dptr; task->rendezvous_ptr = (uintptr_t*)dptr;
*yield = true; *yield = true;
task->block(this, "waiting for select rendezvous"); task->block(this, "waiting for select rendezvous");
} }
for (size_t i = 0; i < locks_taken; i++) { for (size_t i = 0; i < locks_taken; i++) {
size_t k = (i + j) % n_ports; size_t k = (i + j) % n_ports;
rust_port *port = ports[k]; rust_port *port = ports[k];
port->lock.unlock(); port->lock.unlock();
} }
} }
@ -72,16 +72,16 @@ rust_port_selector::msg_sent_on(rust_port *port) {
scoped_lock with(rendezvous_lock); scoped_lock with(rendezvous_lock);
if (task->blocked_on(this)) { if (task->blocked_on(this)) {
for (size_t i = 0; i < n_ports; i++) { for (size_t i = 0; i < n_ports; i++) {
if (port == ports[i]) { if (port == ports[i]) {
// This was one of the ports we were waiting on // This was one of the ports we were waiting on
ports = NULL; ports = NULL;
n_ports = 0; n_ports = 0;
*task->rendezvous_ptr = (uintptr_t) port; *task->rendezvous_ptr = (uintptr_t) port;
task->rendezvous_ptr = NULL; task->rendezvous_ptr = NULL;
task->wakeup(this); task->wakeup(this);
return; return;
} }
} }
} }
} }