ed/Odin
1
0
Fork 0
mirror of https://github.com/Ed94/Odin.git synced 2026-06-18 20:02:22 -07:00
Code Issues Packages Projects Releases Wiki Activity

Update sync.Channel

Browse Source
This commit is contained in:
gingerBill
2020-06-22 17:45:01 +01:00
parent 910ab7b3d4
commit 2a598aa061
1 changed files with 18 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/sync/channel.odin
+18 -9
View File
@@ -8,7 +8,6 @@ _, _ :: time, rand;
Channel :: struct(T: typeid) {
using internal: ^_Channel_Internal(T),
_: bool,
}
_Channel_Internal :: struct(T: typeid) {
@@ -27,6 +26,10 @@ _Channel_Internal :: struct(T: typeid) {
w_waiting: int,
}
channel_init :: proc(c: ^$C/Channel($T), cap: int = 0, allocator := context.allocator) {
c^ = cast(C)channel_make(T, cap, allocator);
}
channel_make :: proc($T: typeid, cap: int = 0, allocator := context.allocator) -> (ch: Channel(T)) {
ch.internal = new(_Channel_Internal(T), allocator);
if ch.internal == nil {
@@ -166,6 +169,12 @@ channel_can_read :: proc(ch: $C/Channel($T)) -> bool {
return len(ch.queue) > 0;
}
channel_can_read_write :: proc(ch: $C/Channel($T)) -> bool {
mutex_lock(&ch.mutex);
defer mutex_unlock(&ch.mutex);
return 0 < len(ch.queue) && len(ch.queue) < cap(ch.queue);
}
channel_iterator :: proc(ch: $C/Channel($T)) -> (elem: T, ok: bool) {
mutex_lock(&ch.mutex);
defer mutex_unlock(&ch.mutex);
@@ -179,7 +188,7 @@ channel_iterator :: proc(ch: $C/Channel($T)) -> (elem: T, ok: bool) {
channel_select :: proc(read_channels, write_channels: []$C/Channel($T), write_msgs: []T) -> (read_msg: T, index: int) {
channel_select :: proc(readers, writers: []$C/Channel($T), write_msgs: []T) -> (read_msg: T, index: int) {
Candidate :: struct {
ch: C,
msg: T,
@@ -188,10 +197,10 @@ channel_select :: proc(read_channels, write_channels: []$C/Channel($T), write_ms
};
count := 0;
candidates := make([]Candidate, len(read_channels) + len(write_channels));
candidates := make([]Candidate, len(readers) + len(writers));
defer delete(candidates);
for c, i in read_channels {
for c, i in readers {
if channel_can_read(c) {
candidates[count] = {
ch = c,
@@ -202,7 +211,7 @@ channel_select :: proc(read_channels, write_channels: []$C/Channel($T), write_ms
}
}
for c, i in write_channels {
for c, i in writers {
if channel_can_write(c) {
candidates[count] = {
ch = c,
@@ -239,10 +248,10 @@ channel_select :: proc(read_channels, write_channels: []$C/Channel($T), write_ms
}
channel_select_write :: proc(write_channels: []$C/Channel($T), write_msgs: []T) -> (read_msg: T, index: int) {
return channel_select([]C{}, write_channels, msg);
channel_select_write :: proc(writers: []$C/Channel($T), write_msgs: []T) -> (read_msg: T, index: int) {
return channel_select([]C{}, writers, msg);
}
channel_select_read :: proc(read_channels: []$C/Channel($T)) -> (index: int) {
_, index = channel_select(read_channels, []C{}, nil);
channel_select_read :: proc(readers: []$C/Channel($T)) -> (index: int) {
_, index = channel_select(readers, []C{}, nil);
return;
}