Monorepo for Tangled
Something went wrong. Try again.
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275package executor
import ( "context" "crypto/sha256" "encoding/hex" "fmt" "io" "os" "strings" "sync" "time"
"github.com/hpcloud/tail" "tangled.org/core/spindle/db" "tangled.org/core/spindle/engine" millproto "tangled.org/core/spindle/mill/proto" millv1 "tangled.org/core/spindle/mill/proto/gen" "tangled.org/core/spindle/models" "tangled.org/core/spindle/secrets")
func (e *Executor) observeLoop(ctx context.Context, sub <-chan struct{}, cursor int64) { ticker := time.NewTicker(5 * time.Second) defer ticker.Stop()
for { select { case <-ctx.Done(): return case <-sub: case <-ticker.C: } e.drainEvents(&cursor) }}
func (e *Executor) drainEvents(cursor *int64) { events, err := e.db.GetEvents(*cursor, 128) if err != nil { e.l.Error("drain status events failed", "err", err) return } for _, event := range events { if event.Id > *cursor { *cursor = event.Id } if err := e.onStatusRow(&event); err != nil { e.l.Error("process status row failed", "err", err) } }}
func (e *Executor) onStatusRow(st *db.StatusRow) error { res := e.reservationFor(st.Pipeline, st.Workflow) if res == nil { return nil }
if models.StatusKind(st.Status).IsFinish() { return e.finishJob(res, st) }
return e.appendStatus(res.leaseID, st)}
func (e *Executor) finishJob(res *reservation, st *db.StatusRow) error { e.mu.Lock() if e.active[res.leaseID] != res { e.mu.Unlock() return nil } cancelled := res.cancelled runDone := res.runDone e.mu.Unlock()
if runDone != nil { <-runDone } e.mu.Lock() failureClass := res.failureClass failureReason := res.failureReason millRecordsTerminalMetrics := res.millRecordsTerminalMetrics e.mu.Unlock()
// the log tail finalizes first so all log lines precede the terminal event if res.stopTail != nil { res.stopTail() }
terminalStatus := st.Status if cancelled { terminalStatus = string(models.StatusKindCancelled) } if cancelled { failureClass = string(engine.FailureClassUser) failureReason = string(engine.FailureReasonCancelled) } if failureClass == "" || failureReason == "" { failureClass, failureReason = defaultTerminalAttribution(terminalStatus) } var logDir string if e.cfg != nil { logDir = e.cfg.Server.LogDir } logPath := models.LogFilePath(logDir, res.wid)
var errStr string if st != nil && st.Error != nil { errStr = *st.Error } var exitCode int64 if st != nil && st.ExitCode != nil { exitCode = *st.ExitCode }
// sha256 of the log file hash := "" if f, err := os.Open(logPath); err == nil { h := sha256.New() if _, err := io.Copy(h, f); err == nil { hash = "sha256:" + hex.EncodeToString(h.Sum(nil)) } _ = f.Close() }
// refs are opaque keys interpreted by the configured artifact store ref := "logs/" + res.leaseID + ".log"
// persist pending artifact state so a restart can retry the upload if e.db != nil { _ = e.db.SavePendingArtifact( res.leaseID, res.wid, terminalStatus, errStr, exitCode, ref, hash, failureClass, failureReason, millRecordsTerminalMetrics, ) }
// upload with a context that survives job cancellation cleanupCtx, cancel := context.WithTimeout(context.WithoutCancel(context.Background()), 2*time.Minute) defer cancel()
if e.writer != nil { if f, err := os.Open(logPath); err == nil { defer f.Close() if uploadErr := e.writer.Put(cleanupCtx, ref, f); uploadErr != nil { e.l.Error("artifact upload failed", "lease", res.leaseID, "err", uploadErr) return fmt.Errorf("artifact upload: %w", uploadErr) } } }
// append the terminal event with the LogArtifact ref if err := e.appendTerminalWithArtifact( res.leaseID, terminalStatus, st, ref, hash, failureClass, failureReason, millRecordsTerminalMetrics, ); err != nil { return err }
// only drop the pending state once the terminal event is in the outbox if e.db != nil { _ = e.db.RemovePendingArtifact(res.leaseID) } e.mu.Lock() cleanup := e.removeReservationLocked(res, false) e.mu.Unlock()
cleanup()
e.pushSnapshot() return nil}
func (e *Executor) reservationFor(pipelineID, workflow string) *reservation { e.mu.Lock() defer e.mu.Unlock() for _, res := range e.active { if res.wid.PipelineId == models.PipelineId(pipelineID) && res.wid.Name == workflow { return res } } return nil}
func (e *Executor) maskSecrets(res *reservation, text string) string { if res == nil || res.vault == nil { return text } for _, s := range res.vault.secrets { if s.Value != "" { text = strings.ReplaceAll(text, s.Value, "***") } } return text}
func (e *Executor) SendLiveLog(leaseID string, raw []byte) error { e.connMu.Lock() enc := e.enc e.connMu.Unlock() if enc == nil { return nil } return enc.Encode(&millproto.Message{ LiveLog: &millv1.LiveLog{ LeaseId: leaseID, RawJson: raw, }, })}
func (e *Executor) startTail(res *reservation) { path := models.LogFilePath(e.cfg.Server.LogDir, res.wid) t, err := tail.TailFile(path, tail.Config{ Follow: true, ReOpen: true, MustExist: false, Location: &tail.SeekInfo{Offset: 0, Whence: io.SeekStart}, Logger: tail.DiscardingLogger, }) if err != nil { e.l.Error("tail log file failed", "wid", res.wid, "err", err) return }
done := make(chan struct{}) go func() { defer close(done) for line := range t.Lines { if line == nil || line.Err != nil { continue } masked := e.maskSecrets(res, line.Text) _ = e.SendLiveLog(res.leaseID, []byte(masked+"\n")) } }()
var once sync.Once res.stopTail = func() { once.Do(func() { _ = t.StopAtEOF() <-done }) }}
type memVault struct { secrets []secrets.UnlockedSecret}
func newMemVault(pb []*millv1.Secret) *memVault { v := &memVault{secrets: make([]secrets.UnlockedSecret, 0, len(pb))} for _, s := range pb { v.secrets = append(v.secrets, secrets.UnlockedSecret{ Key: s.Key, Value: s.Value, }) } return v}
func (v *memVault) GetSecretsUnlocked(ctx context.Context, repo secrets.RepoIdentifier) ([]secrets.UnlockedSecret, error) { return v.secrets, nil}func (v *memVault) GetSecretsLocked(ctx context.Context, repo secrets.RepoIdentifier) ([]secrets.LockedSecret, error) { return nil, nil}func (v *memVault) AddSecret(ctx context.Context, s secrets.UnlockedSecret) error { return nil }func (v *memVault) RemoveSecret(ctx context.Context, s secrets.Secret[any]) error { return nil }func (v *memVault) RemoveAllSecrets(ctx context.Context, repo secrets.RepoIdentifier) error { return nil}
var _ secrets.Manager = (*memVault)(nil)