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`.
73 lines
2.3 KiB
Go
73 lines
2.3 KiB
Go
package replication
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"go.temporal.io/server/common/cluster"
|
|
)
|
|
|
|
type (
|
|
pollerManager interface {
|
|
getSourceClusterShardIDs(sourceClusterName string) ([]int32, error)
|
|
}
|
|
|
|
pollerManagerImpl struct {
|
|
currentShardId int32
|
|
clusterMetadata cluster.Metadata
|
|
}
|
|
)
|
|
|
|
var _ pollerManager = (*pollerManagerImpl)(nil)
|
|
|
|
func newPollerManager(
|
|
currentShardId int32,
|
|
clusterMetadata cluster.Metadata,
|
|
) *pollerManagerImpl {
|
|
return &pollerManagerImpl{
|
|
currentShardId: currentShardId,
|
|
clusterMetadata: clusterMetadata,
|
|
}
|
|
}
|
|
|
|
func (p pollerManagerImpl) getSourceClusterShardIDs(sourceClusterName string) ([]int32, error) {
|
|
currentCluster := p.clusterMetadata.GetCurrentClusterName()
|
|
allClusters := p.clusterMetadata.GetAllClusterInfo()
|
|
currentClusterInfo, ok := allClusters[currentCluster]
|
|
if !ok {
|
|
return nil, errors.New("cannot get current cluster info from cluster metadata cache")
|
|
}
|
|
remoteClusterInfo, ok := allClusters[sourceClusterName]
|
|
if !ok {
|
|
return nil, errors.New(fmt.Sprintf("cannot get source cluster %s info from cluster metadata cache", sourceClusterName))
|
|
}
|
|
|
|
// The remote shard count and local shard count must be multiples.
|
|
large, small := remoteClusterInfo.ShardCount, currentClusterInfo.ShardCount
|
|
if small > large {
|
|
large, small = small, large
|
|
}
|
|
if large%small != 0 {
|
|
return nil, errors.New(fmt.Sprintf("remote shard count %d and local shard count %d are not multiples.", remoteClusterInfo.ShardCount, currentClusterInfo.ShardCount))
|
|
}
|
|
return generateShardIDs(p.currentShardId, currentClusterInfo.ShardCount, remoteClusterInfo.ShardCount), nil
|
|
}
|
|
|
|
// NOTE generateShardIDs is different than common.MapShardID
|
|
// common.MapShardID guarantee to return the corresponding shard IDs for give shard ID
|
|
// this function however is only a helper function for polling & redirecting replication task,
|
|
func generateShardIDs(localShardId int32, localShardCount int32, remoteShardCount int32) []int32 {
|
|
var pollingShards []int32
|
|
if remoteShardCount <= localShardCount {
|
|
if localShardId <= remoteShardCount {
|
|
pollingShards = append(pollingShards, localShardId)
|
|
}
|
|
return pollingShards
|
|
}
|
|
// remoteShardCount > localShardCount, replication poller will poll from multiple remote shard.
|
|
for i := localShardId; i <= remoteShardCount; i += localShardCount {
|
|
pollingShards = append(pollingShards, i)
|
|
}
|
|
return pollingShards
|
|
}
|