Ophestra Umiker
c7b77d6e5e
All checks were successful
test / test (push) Successful in 23s
This fixes the hang on resume when no messages were ever printed. Signed-off-by: Ophestra Umiker <cat@ophivana.moe>
73 lines
994 B
Go
73 lines
994 B
Go
package fmsg
|
|
|
|
import (
|
|
"os"
|
|
"sync"
|
|
"sync/atomic"
|
|
)
|
|
|
|
var (
|
|
wstate atomic.Bool
|
|
withhold = make(chan struct{}, 1)
|
|
msgbuf = make(chan dOp, 64) // these ops are tiny so a large buffer is allocated for withholding output
|
|
|
|
dequeueOnce sync.Once
|
|
queueSync sync.WaitGroup
|
|
)
|
|
|
|
func dequeue() {
|
|
go func() {
|
|
for {
|
|
select {
|
|
case op := <-msgbuf:
|
|
op.Do()
|
|
queueSync.Done()
|
|
case <-withhold:
|
|
<-withhold
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
|
|
type dOp interface{ Do() }
|
|
|
|
func Exit(code int) {
|
|
queueSync.Wait()
|
|
os.Exit(code)
|
|
}
|
|
|
|
func Withhold() {
|
|
dequeueOnce.Do(dequeue)
|
|
if wstate.CompareAndSwap(false, true) {
|
|
withhold <- struct{}{}
|
|
}
|
|
}
|
|
|
|
func Resume() {
|
|
dequeueOnce.Do(dequeue)
|
|
if wstate.CompareAndSwap(true, false) {
|
|
withhold <- struct{}{}
|
|
}
|
|
}
|
|
|
|
type dPrint []any
|
|
|
|
func (v dPrint) Do() {
|
|
std.Print(v...)
|
|
}
|
|
|
|
type dPrintf struct {
|
|
format string
|
|
v []any
|
|
}
|
|
|
|
func (d *dPrintf) Do() {
|
|
std.Printf(d.format, d.v...)
|
|
}
|
|
|
|
type dPrintln []any
|
|
|
|
func (v dPrintln) Do() {
|
|
std.Println(v...)
|
|
}
|