mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
zeekay
5a601d081b
Full module path rebrand from Temporal to Hanzo Tasks: - go.mod module: github.com/hanzoai/tasks - All internal Go imports updated (1906 files) - cmd/server renamed to cmd/tasksd - Binary name: tasksd (was temporal-server) - Docker images: hanzoai/tasks (was temporalio/server) - README rewritten with Hanzo Tasks branding - LLM.md created with integration notes - Makefile, goreleaser, CI workflows updated - go.temporal.io/sdk and go.temporal.io/api imports preserved (separate repos) Fork of Temporal.io (MIT license).
73 lines
2.3 KiB
Go
73 lines
2.3 KiB
Go
package replication
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/hanzoai/tasks/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
|
|
}
|