shithub: mc

ref: 9892228031c0ffcfa95a1443fc565b031525bb3a
dir: /lib/thread/future.myr/

use std

use "mutex"
use "condvar"
use "atomic"

pkg thread =
	type future(@a) = struct
		mtx	: mutex
		cv	: cond
		set	: bool
		val	: @a
	;;

	generic mkfut	: (-> future(@a)#)
	generic futput	: (fut : future(@a)#, val : @a -> void)
	generic futget	: (fut : future(@a)# -> @a)
	generic futpeek	: (fut : future(@a)# -> @a)
;;

generic mkfut = {
	var fut

	fut = std.alloc()
	fut.mtx = mkmtx()
	fut.cv = mkcond(&fut.mtx)
	fut.set = false
	-> fut
}

generic futput = {fut, val
	fut.val = val
	xset(&fut.set, true)
	condsignal(&fut.cv)
}

generic futget = {fut
	var val
	
	val = futpeek(fut)
	std.free(fut)
	-> val
}

generic futpeek = {fut
	if !xget(&fut.set)
		mtxlock(&fut.mtx)
		if !xget(&fut.set)
			condwait(&fut.cv)
		;;
		mtxunlock(&fut.mtx)
	;;
	-> fut.val
}