mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Lina JodoinandGitHub
4c140933cf
## What changed? - Merge PR combining the previous PRs: #8164, #8165, #8166, #8167, #8168, #8169 ## Why? - All of those PRs are interdependent and cannot be merged individually cleanly. Sending a single merge commit PR to be merged to main. ## How did you test it? - [ ] built - [ ] run locally and tested manually - [ ] covered by existing tests - [ ] added new unit test(s) - [ ] added new functional test(s) ## Potential risks - It's still disabled.
69 lines
1.6 KiB
Go
69 lines
1.6 KiB
Go
package scheduler
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/pborman/uuid"
|
|
schedulespb "go.temporal.io/server/api/schedule/v1"
|
|
"go.temporal.io/server/chasm"
|
|
"go.temporal.io/server/chasm/lib/scheduler/gen/schedulerpb/v1"
|
|
"google.golang.org/protobuf/types/known/timestamppb"
|
|
)
|
|
|
|
// The Backfiller component is responsible for buffering manually
|
|
// requested actions. Each backfill request has its own Backfiller node.
|
|
type Backfiller struct {
|
|
chasm.UnimplementedComponent
|
|
|
|
*schedulerpb.BackfillerState
|
|
|
|
Scheduler chasm.Field[*Scheduler]
|
|
}
|
|
|
|
type BackfillRequestType int
|
|
|
|
const (
|
|
RequestTypeTrigger BackfillRequestType = iota
|
|
RequestTypeBackfill
|
|
)
|
|
|
|
// newBackfiller returns an initialized backfiller without a request set or tasks
|
|
// created.
|
|
func newBackfiller(
|
|
ctx chasm.MutableContext,
|
|
scheduler *Scheduler,
|
|
) *Backfiller {
|
|
id := uuid.New()
|
|
backfiller := &Backfiller{
|
|
BackfillerState: &schedulerpb.BackfillerState{
|
|
BackfillId: id,
|
|
LastProcessedTime: timestamppb.New(ctx.Now(scheduler)),
|
|
},
|
|
Scheduler: chasm.ComponentPointerTo(ctx, scheduler),
|
|
}
|
|
return backfiller
|
|
}
|
|
|
|
func (b *Backfiller) LifecycleState(ctx chasm.Context) chasm.LifecycleState {
|
|
return chasm.LifecycleStateRunning
|
|
}
|
|
|
|
func (b *Backfiller) RequestType() BackfillRequestType {
|
|
if b.GetTriggerRequest() != nil {
|
|
return RequestTypeTrigger
|
|
}
|
|
|
|
return RequestTypeBackfill
|
|
}
|
|
|
|
type backfillProgressResult struct {
|
|
// BufferedStarts that should be enqueued to the Invoker.
|
|
BufferedStarts []*schedulespb.BufferedStart
|
|
|
|
// High water mark for when state was last updated.
|
|
LastProcessedTime time.Time
|
|
|
|
// When true, the backfill has completed and the node can be deleted.
|
|
Complete bool
|
|
}
|