Michael Chus
a393dcb731
- jobState now has optional cancel func; abort() calls it if job is running - handleAPISATRun passes cancellable context to RunNvidiaAcceptancePackWithOptions - POST /api/sat/abort?job_id=... cancels the running SAT job - bible-local/runtime-flows.md: replace TUI SAT flow with Web UI flow Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
103 lines
2.0 KiB
Go
103 lines
2.0 KiB
Go
package webui
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// jobState holds the output lines and completion status of an async job.
|
|
type jobState struct {
|
|
lines []string
|
|
done bool
|
|
err string
|
|
mu sync.Mutex
|
|
subs []chan string
|
|
cancel func() // optional cancel function; nil if job is not cancellable
|
|
}
|
|
|
|
// abort cancels the job if it has a cancel function and is not yet done.
|
|
func (j *jobState) abort() bool {
|
|
j.mu.Lock()
|
|
defer j.mu.Unlock()
|
|
if j.done || j.cancel == nil {
|
|
return false
|
|
}
|
|
j.cancel()
|
|
return true
|
|
}
|
|
|
|
func (j *jobState) append(line string) {
|
|
j.mu.Lock()
|
|
defer j.mu.Unlock()
|
|
j.lines = append(j.lines, line)
|
|
for _, ch := range j.subs {
|
|
select {
|
|
case ch <- line:
|
|
default:
|
|
}
|
|
}
|
|
}
|
|
|
|
func (j *jobState) finish(errMsg string) {
|
|
j.mu.Lock()
|
|
defer j.mu.Unlock()
|
|
j.done = true
|
|
j.err = errMsg
|
|
for _, ch := range j.subs {
|
|
close(ch)
|
|
}
|
|
j.subs = nil
|
|
}
|
|
|
|
// subscribe returns a channel that receives all future lines.
|
|
// Existing lines are returned first, then the channel streams new ones.
|
|
func (j *jobState) subscribe() ([]string, <-chan string) {
|
|
j.mu.Lock()
|
|
defer j.mu.Unlock()
|
|
existing := make([]string, len(j.lines))
|
|
copy(existing, j.lines)
|
|
if j.done {
|
|
return existing, nil
|
|
}
|
|
ch := make(chan string, 256)
|
|
j.subs = append(j.subs, ch)
|
|
return existing, ch
|
|
}
|
|
|
|
// jobManager manages async jobs identified by string IDs.
|
|
type jobManager struct {
|
|
mu sync.Mutex
|
|
jobs map[string]*jobState
|
|
}
|
|
|
|
var globalJobs = &jobManager{jobs: make(map[string]*jobState)}
|
|
|
|
func (m *jobManager) create(id string) *jobState {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
j := &jobState{}
|
|
m.jobs[id] = j
|
|
// Schedule cleanup after 30 minutes
|
|
go func() {
|
|
time.Sleep(30 * time.Minute)
|
|
m.mu.Lock()
|
|
delete(m.jobs, id)
|
|
m.mu.Unlock()
|
|
}()
|
|
return j
|
|
}
|
|
|
|
// isDone returns true if the job has finished (either successfully or with error).
|
|
func (j *jobState) isDone() bool {
|
|
j.mu.Lock()
|
|
defer j.mu.Unlock()
|
|
return j.done
|
|
}
|
|
|
|
func (m *jobManager) get(id string) (*jobState, bool) {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
j, ok := m.jobs[id]
|
|
return j, ok
|
|
}
|