mirror of
https://github.com/langgenius/dify-plugin-daemon.git
synced 2026-07-22 09:45:27 -04:00
Yeuoly
28 lines
447 B
Go
28 lines
447 B
Go
package debugging
|
|
|
|
import (
|
|
"time"
|
|
)
|
|
|
|
// PossibleBlocking runs the function f in a goroutine and returns the result.
|
|
// If the function f is blocking, the test will fail.
|
|
func PossibleBlocking[T any](f func() T, timeout time.Duration, trigger func()) T {
|
|
d := make(chan T)
|
|
|
|
go func() {
|
|
d <- f()
|
|
}()
|
|
|
|
timer := time.NewTimer(timeout)
|
|
defer timer.Stop()
|
|
|
|
for {
|
|
select {
|
|
case <-timer.C:
|
|
trigger()
|
|
case v := <-d:
|
|
return v
|
|
}
|
|
}
|
|
}
|