30 lines
388 B
Go
30 lines
388 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
// context.AfterFunc
|
||
|
|
|
||
|
|
func WithCtxAfterFunc(ctx context.Context, action func()) {
|
||
|
|
if action != nil {
|
||
|
|
go func() {
|
||
|
|
<-ctx.Done()
|
||
|
|
action()
|
||
|
|
}()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
||
|
|
WithCtxAfterFunc(ctx, func() {
|
||
|
|
fmt.Println("after")
|
||
|
|
})
|
||
|
|
|
||
|
|
cancel()
|
||
|
|
|
||
|
|
time.Sleep(100 * time.Millisecond)
|
||
|
|
}
|