3797091d37
- Updated BuildIdentifier to include assistant ID in session identifiers for better uniqueness. - Implemented automatic starting of stopped boxes in resolveBox, improving recovery processes. - Introduced status management for boxes, allowing for accurate tracking of their state (running, exited, stopped). - Added idle timeout defaults based on lifecycle policy in BuildCreateOptions, enhancing configuration flexibility. - Refactored tests to validate new box status behavior and lifecycle management improvements. Made-with: Cursor
87 lines
2 KiB
Go
87 lines
2 KiB
Go
package sandbox
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/yaoapp/yao/monitor"
|
|
)
|
|
|
|
func init() {
|
|
monitor.Register(&sandboxWatcher{})
|
|
}
|
|
|
|
type sandboxWatcher struct{}
|
|
|
|
func (w *sandboxWatcher) Name() string { return "sandbox" }
|
|
func (w *sandboxWatcher) Interval() time.Duration { return 30 * time.Second }
|
|
|
|
func (w *sandboxWatcher) Check(ctx context.Context) []monitor.Alert {
|
|
if mgr == nil {
|
|
return nil
|
|
}
|
|
|
|
var alerts []monitor.Alert
|
|
mgr.boxes.Range(func(_, v any) bool {
|
|
b := v.(*Box)
|
|
|
|
status := b.inspectStatus(ctx)
|
|
old, _ := b.status.Swap(status).(string)
|
|
if old != "" && old != status {
|
|
alerts = append(alerts, monitor.Alert{
|
|
Level: monitor.Info,
|
|
Target: "box:" + b.id,
|
|
Message: fmt.Sprintf("status %s → %s", old, status),
|
|
})
|
|
}
|
|
|
|
if status != "running" {
|
|
return true
|
|
}
|
|
|
|
idle := time.Since(b.idleSince())
|
|
timeout := b.idleTimeout()
|
|
if timeout <= 0 || idle <= timeout {
|
|
return true
|
|
}
|
|
|
|
switch b.policy {
|
|
case Session:
|
|
alerts = append(alerts, monitor.Alert{
|
|
Level: monitor.Warn,
|
|
Target: "box:" + b.id,
|
|
Message: fmt.Sprintf("session idle expired (idle=%s, timeout=%s), removing", idle.Round(time.Second), timeout),
|
|
Action: func(ctx context.Context) {
|
|
mgr.Remove(ctx, b.id)
|
|
},
|
|
})
|
|
|
|
case LongRunning:
|
|
alerts = append(alerts, monitor.Alert{
|
|
Level: monitor.Warn,
|
|
Target: "box:" + b.id,
|
|
Message: fmt.Sprintf("longrunning idle expired (idle=%s, timeout=%s), stopping", idle.Round(time.Second), timeout),
|
|
Action: func(ctx context.Context) {
|
|
b.Stop(ctx)
|
|
},
|
|
})
|
|
}
|
|
|
|
if b.policy == LongRunning {
|
|
if lifetime := b.maxLifetime(); lifetime > 0 && time.Since(b.createdAt) > lifetime {
|
|
alerts = append(alerts, monitor.Alert{
|
|
Level: monitor.Warn,
|
|
Target: "box:" + b.id,
|
|
Message: fmt.Sprintf("lifetime expired (%s), removing", lifetime),
|
|
Action: func(ctx context.Context) {
|
|
mgr.Remove(ctx, b.id)
|
|
},
|
|
})
|
|
}
|
|
}
|
|
|
|
return true
|
|
})
|
|
return alerts
|
|
}
|