mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-30 14:56:19 +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`.
40 lines
1.4 KiB
Go
40 lines
1.4 KiB
Go
//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination stream_receiver_flow_controller_mock.go
|
|
package replication
|
|
|
|
import (
|
|
enumsspb "go.temporal.io/server/api/enums/v1"
|
|
"go.temporal.io/server/service/history/configs"
|
|
)
|
|
|
|
type (
|
|
FlowControlSignalProvider func() *FlowControlSignal
|
|
|
|
// FlowControlSignal holds signals to make flow control decision, more signalsProvider can be added here i.e. total persistence rps, cpu usage etc.
|
|
FlowControlSignal struct {
|
|
taskTrackingCount int
|
|
}
|
|
ReceiverFlowController interface {
|
|
GetFlowControlInfo(priority enumsspb.TaskPriority) enumsspb.ReplicationFlowControlCommand
|
|
}
|
|
streamReceiverFlowControllerImpl struct {
|
|
signalsProvider map[enumsspb.TaskPriority]FlowControlSignalProvider
|
|
config *configs.Config
|
|
}
|
|
)
|
|
|
|
func NewReceiverFlowControl(signals map[enumsspb.TaskPriority]FlowControlSignalProvider, config *configs.Config) *streamReceiverFlowControllerImpl {
|
|
return &streamReceiverFlowControllerImpl{
|
|
signalsProvider: signals,
|
|
config: config,
|
|
}
|
|
}
|
|
|
|
func (s *streamReceiverFlowControllerImpl) GetFlowControlInfo(priority enumsspb.TaskPriority) enumsspb.ReplicationFlowControlCommand {
|
|
if signal, ok := s.signalsProvider[priority]; ok {
|
|
if signal().taskTrackingCount > s.config.ReplicationReceiverMaxOutstandingTaskCount() {
|
|
return enumsspb.REPLICATION_FLOW_CONTROL_COMMAND_PAUSE
|
|
}
|
|
}
|
|
return enumsspb.REPLICATION_FLOW_CONTROL_COMMAND_RESUME
|
|
}
|