mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Alex ShtinandGitHub
8e8f1b0f7c
## What changed?
<!-- Describe what has changed in this PR -->
Remove license header from every file. Because it is really hard to
follow in this PR here is the summary:
1. License header is removed from all `.go` and `.proto` files
:fireworks::fireworks:🎆.
2. `LICENSE` file in the root directory has only Temporal and Uber
copyrights.
3. 5 other `LICENSE` files added to the packages which have copyrights
different from Temporal and Uber: Datadog, Xargin, "Mat Ryer, Tyler
Bunnell and contributors".
4. `license_file` flag is removed from all code generation tools.
5. `copyright_file` flag is removed from `go:generate mockgen`
directive.
6. All copyright related targets are removed from `Makefile`.
7. Updated Temporal copyright year to 2025 everywhere.
## Why?
<!-- Tell your future self why have you made these changes -->
I double checked with legal department that it is not needed to have
license header in every file. One file per repo is enough. I put all
copyrights to the root `LICENSE` file and removed header from all other
files. Also updated tools and `Makefile`.
125 lines
3.9 KiB
Go
125 lines
3.9 KiB
Go
package scheduler
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
"time"
|
|
|
|
enumspb "go.temporal.io/api/enums/v1"
|
|
"go.temporal.io/api/workflowservice/v1"
|
|
sdkworker "go.temporal.io/sdk/worker"
|
|
"go.temporal.io/sdk/workflow"
|
|
schedulespb "go.temporal.io/server/api/schedule/v1"
|
|
"go.temporal.io/server/common/dynamicconfig"
|
|
"go.temporal.io/server/common/log"
|
|
"go.temporal.io/server/common/metrics"
|
|
"go.temporal.io/server/common/namespace"
|
|
"go.temporal.io/server/common/quotas"
|
|
"go.temporal.io/server/common/resource"
|
|
"go.temporal.io/server/common/searchattribute"
|
|
workercommon "go.temporal.io/server/service/worker/common"
|
|
"go.uber.org/fx"
|
|
)
|
|
|
|
const (
|
|
WorkflowType = "temporal-sys-scheduler-workflow"
|
|
NamespaceDivision = "TemporalScheduler"
|
|
)
|
|
|
|
var (
|
|
VisibilityBaseListQuery = fmt.Sprintf(
|
|
"%s = '%s' AND %s = '%s' AND %s = '%s'",
|
|
searchattribute.WorkflowType,
|
|
WorkflowType,
|
|
searchattribute.TemporalNamespaceDivision,
|
|
NamespaceDivision,
|
|
searchattribute.ExecutionStatus,
|
|
enumspb.WORKFLOW_EXECUTION_STATUS_RUNNING.String(),
|
|
)
|
|
)
|
|
|
|
type (
|
|
workerComponent struct {
|
|
specBuilder *SpecBuilder // workflow dep
|
|
activityDeps activityDeps
|
|
enabledForNs dynamicconfig.BoolPropertyFnWithNamespaceFilter
|
|
globalNSStartWorkflowRPS dynamicconfig.TypedSubscribableWithNamespaceFilter[float64]
|
|
maxBlobSize dynamicconfig.IntPropertyFnWithNamespaceFilter
|
|
localActivitySleepLimit dynamicconfig.DurationPropertyFnWithNamespaceFilter
|
|
}
|
|
|
|
activityDeps struct {
|
|
fx.In
|
|
MetricsHandler metrics.Handler
|
|
Logger log.Logger
|
|
HistoryClient resource.HistoryClient
|
|
FrontendClient workflowservice.WorkflowServiceClient
|
|
}
|
|
|
|
fxResult struct {
|
|
fx.Out
|
|
Component workercommon.PerNSWorkerComponent `group:"perNamespaceWorkerComponent"`
|
|
}
|
|
)
|
|
|
|
var Module = fx.Options(
|
|
fx.Provide(NewResult),
|
|
fx.Provide(NewSpecBuilder),
|
|
)
|
|
|
|
func NewResult(
|
|
dc *dynamicconfig.Collection,
|
|
specBuilder *SpecBuilder,
|
|
params activityDeps,
|
|
) fxResult {
|
|
return fxResult{
|
|
Component: &workerComponent{
|
|
specBuilder: specBuilder,
|
|
activityDeps: params,
|
|
enabledForNs: dynamicconfig.WorkerEnableScheduler.Get(dc),
|
|
globalNSStartWorkflowRPS: dynamicconfig.SchedulerNamespaceStartWorkflowRPS.Subscribe(dc),
|
|
maxBlobSize: dynamicconfig.BlobSizeLimitError.Get(dc),
|
|
localActivitySleepLimit: dynamicconfig.SchedulerLocalActivitySleepLimit.Get(dc),
|
|
},
|
|
}
|
|
}
|
|
|
|
func (s *workerComponent) DedicatedWorkerOptions(ns *namespace.Namespace) *workercommon.PerNSDedicatedWorkerOptions {
|
|
return &workercommon.PerNSDedicatedWorkerOptions{
|
|
Enabled: s.enabledForNs(ns.Name().String()),
|
|
}
|
|
}
|
|
|
|
func (s *workerComponent) Register(registry sdkworker.Registry, ns *namespace.Namespace, details workercommon.RegistrationDetails) func() {
|
|
wfFunc := func(ctx workflow.Context, args *schedulespb.StartScheduleArgs) error {
|
|
return schedulerWorkflowWithSpecBuilder(ctx, args, s.specBuilder)
|
|
}
|
|
registry.RegisterWorkflowWithOptions(wfFunc, workflow.RegisterOptions{Name: WorkflowType})
|
|
|
|
activities, cleanup := s.newActivities(ns.Name(), ns.ID(), details)
|
|
registry.RegisterActivity(activities)
|
|
return cleanup
|
|
}
|
|
|
|
func (s *workerComponent) newActivities(name namespace.Name, id namespace.ID, details workercommon.RegistrationDetails) (*activities, func()) {
|
|
const burstRatio = 1.0
|
|
|
|
lim := quotas.NewRateLimiter(1, 1)
|
|
cb := func(rps float64) {
|
|
localRPS := rps * float64(details.Multiplicity) / float64(details.TotalWorkers)
|
|
burst := max(1, int(math.Ceil(localRPS*burstRatio)))
|
|
lim.SetRateBurst(localRPS, burst)
|
|
}
|
|
initialRPS, cancel := s.globalNSStartWorkflowRPS(name.String(), cb)
|
|
cb(initialRPS)
|
|
|
|
return &activities{
|
|
activityDeps: s.activityDeps,
|
|
namespace: name,
|
|
namespaceID: id,
|
|
startWorkflowRateLimiter: lim,
|
|
maxBlobSize: func() int { return s.maxBlobSize(name.String()) },
|
|
localActivitySleepLimit: func() time.Duration { return s.localActivitySleepLimit(name.String()) },
|
|
}, cancel
|
|
}
|