mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Alex ShtinandGitHub
846d88c8cb
## What changed? Refactor: use new `serviceerror.New...f` constructors. ## Why? Most of service errors now supports `fmt.Sprintf` format and they should be used instead of `fmt.Sprintf`. ## How did you test it? - [ ] built - [ ] run locally and tested manually - [x] covered by existing tests - [ ] added new unit test(s) - [ ] added new functional test(s)
98 lines
2.5 KiB
Go
98 lines
2.5 KiB
Go
package nsmanager
|
|
|
|
import (
|
|
enumspb "go.temporal.io/api/enums/v1"
|
|
"go.temporal.io/api/serviceerror"
|
|
persistencespb "go.temporal.io/server/api/persistence/v1"
|
|
"go.temporal.io/server/common/cluster"
|
|
)
|
|
|
|
type (
|
|
Validator struct {
|
|
clusterMetadata cluster.Metadata
|
|
}
|
|
)
|
|
|
|
// NewValidator create a new namespace attribute Validator
|
|
func NewValidator(
|
|
clusterMetadata cluster.Metadata,
|
|
) *Validator {
|
|
return &Validator{
|
|
clusterMetadata: clusterMetadata,
|
|
}
|
|
}
|
|
|
|
func (d *Validator) ValidateNamespaceConfig(config *persistencespb.NamespaceConfig) error {
|
|
if config.HistoryArchivalState == enumspb.ARCHIVAL_STATE_ENABLED && len(config.HistoryArchivalUri) == 0 {
|
|
return errInvalidArchivalConfig
|
|
}
|
|
if config.VisibilityArchivalState == enumspb.ARCHIVAL_STATE_ENABLED && len(config.VisibilityArchivalUri) == 0 {
|
|
return errInvalidArchivalConfig
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (d *Validator) ValidateNamespaceReplicationConfigForLocalNamespace(
|
|
replicationConfig *persistencespb.NamespaceReplicationConfig,
|
|
) error {
|
|
activeCluster := replicationConfig.ActiveClusterName
|
|
clusters := replicationConfig.Clusters
|
|
|
|
if err := d.validateClusterName(activeCluster); err != nil {
|
|
return err
|
|
}
|
|
for _, clusterName := range clusters {
|
|
if err := d.validateClusterName(clusterName); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if activeCluster != d.clusterMetadata.GetCurrentClusterName() {
|
|
return serviceerror.NewInvalidArgument("Invalid local namespace active cluster")
|
|
}
|
|
|
|
if len(clusters) != 1 || clusters[0] != activeCluster {
|
|
return serviceerror.NewInvalidArgument("Invalid local namespace clusters")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *Validator) ValidateNamespaceReplicationConfigForGlobalNamespace(
|
|
replicationConfig *persistencespb.NamespaceReplicationConfig,
|
|
) error {
|
|
activeCluster := replicationConfig.ActiveClusterName
|
|
clusters := replicationConfig.Clusters
|
|
|
|
if err := d.validateClusterName(activeCluster); err != nil {
|
|
return err
|
|
}
|
|
for _, clusterName := range clusters {
|
|
if err := d.validateClusterName(clusterName); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
activeClusterInClusters := false
|
|
for _, clusterName := range clusters {
|
|
if clusterName == activeCluster {
|
|
activeClusterInClusters = true
|
|
break
|
|
}
|
|
}
|
|
if !activeClusterInClusters {
|
|
return errActiveClusterNotInClusters
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *Validator) validateClusterName(
|
|
clusterName string,
|
|
) error {
|
|
if info, ok := d.clusterMetadata.GetAllClusterInfo()[clusterName]; !ok || !info.Enabled {
|
|
return serviceerror.NewInvalidArgumentf("Invalid cluster name: %v", clusterName)
|
|
}
|
|
return nil
|
|
}
|