38 lines
669 B
Go
38 lines
669 B
Go
|
package ice
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (a *Agent) context() context.Context {
|
||
|
|
return agentContext(a.done)
|
||
|
|
}
|
||
|
|
|
||
|
|
type agentContext chan struct{}
|
||
|
|
|
||
|
|
// Done implements context.Context
|
||
|
|
func (a agentContext) Done() <-chan struct{} {
|
||
|
|
return (chan struct{})(a)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Err implements context.Context
|
||
|
|
func (a agentContext) Err() error {
|
||
|
|
select {
|
||
|
|
case <-(chan struct{})(a):
|
||
|
|
return ErrRunCanceled
|
||
|
|
default:
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Deadline implements context.Context
|
||
|
|
func (a agentContext) Deadline() (deadline time.Time, ok bool) {
|
||
|
|
return time.Time{}, false
|
||
|
|
}
|
||
|
|
|
||
|
|
// Value implements context.Context
|
||
|
|
func (a agentContext) Value(key interface{}) interface{} {
|
||
|
|
return nil
|
||
|
|
}
|