ed/Odin
1
0
Fork 0
mirror of https://github.com/Ed94/Odin.git synced 2026-08-06 07:38:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add sync.Wait_Group

Browse Source
This commit is contained in:
gingerBill
2020-07-15 00:25:37 +01:00
parent ca818fb857
commit 3a1492fc99
2 changed files with 87 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/sync/channel.odin
+29 -5
View File
@@ -1,6 +1,5 @@
package sync
// import "core:fmt"
import "core:mem"
import "core:time"
import "core:intrinsics"
@@ -113,17 +112,32 @@ channel_close :: proc(ch: $C/Channel($T), loc := #caller_location) {
}
channel_iterator :: proc(ch: $C/Channel($T)) -> (val: T, ok: bool) {
channel_iterator :: proc(ch: $C/Channel($T)) -> (msg: T, ok: bool) {
c := ch._internal;
if c == nil {
return;
}
if !c.closed || c.len > 0 {
val, ok = channel_recv(ch), true;
msg, ok = channel_recv(ch), true;
}
return;
}
channel_drain :: proc(ch: $C/Channel($T)) {
raw_channel_drain(ch._internal);
}
channel_move :: proc(dst, src: $C/Channel($T)) {
// for channel_len(src) > 0 {
// msg := channel_recv(src);
// channel_send(dst, msg);
// }
for msg in channel_iterator(src) {
channel_send(dst, msg);
}
}
channel_select_recv :: proc(channels: ..^Raw_Channel) -> (index: int) {
@@ -247,8 +261,6 @@ channel_select_send_msg :: proc(msg: $T, channels: ..$C/Channel(T)) -> (index: i
Raw_Channel :: struct {
data: rawptr,
elem_size: int,
@@ -393,3 +405,15 @@ raw_channel_can_recv :: proc(c: ^Raw_Channel) -> (ok: bool) {
mutex_unlock(&c.mutex);
return;
}
raw_channel_drain :: proc(c: ^Raw_Channel) {
if c == nil {
return;
}
mutex_lock(&c.mutex);
c.len = 0;
c.read = 0;
c.write = 0;
mutex_unlock(&c.mutex);
}