[shard-distributor] Added logic to return the shard owner based on the namesplace

common/types/mapper/proto/errors.go

@@ -27,6 +27,7 @@ import (
 	"go.uber.org/yarpc/encoding/protobuf"
 	"go.uber.org/yarpc/yarpcerrors"
 
+	sharddistributorv1 "github.com/uber/cadence/.gen/proto/sharddistributor/v1"
 	sharedv1 "github.com/uber/cadence/.gen/proto/shared/v1"
 	cadence_errors "github.com/uber/cadence/common/errors"
 	"github.com/uber/cadence/common/types"
@@ -86,6 +87,8 @@ func FromError(err error) error {
 		return typedErr
 	} else if ok, typedErr = errorutils.ConvertError(err, fromStickyWorkerUnavailableErr); ok {
 		return typedErr
+	} else if ok, typedErr = errorutils.ConvertError(err, fromNamespaceNotFoundErr); ok {
+		return typedErr
 	}
 
 	return protobuf.NewError(yarpcerrors.CodeUnknown, err.Error())
@@ -118,6 +121,10 @@ func ToError(err error) error {
 			return &types.WorkflowExecutionAlreadyCompletedError{
 				Message: status.Message(),
 			}
+		case *sharddistributorv1.NamespaceNotFoundError:
+			return &types.NamespaceNotFoundError{
+				Namespace: details.Namespace,
+			}
 		}
 	case yarpcerrors.CodeInvalidArgument:
 		switch getErrorDetails(err).(type) {
@@ -363,3 +370,9 @@ func fromRemoteSyncMatchedErr(e *types.RemoteSyncMatchedError) error {
 func fromStickyWorkerUnavailableErr(e *types.StickyWorkerUnavailableError) error {
 	return protobuf.NewError(yarpcerrors.CodeUnavailable, e.Message, protobuf.WithErrorDetails(&apiv1.StickyWorkerUnavailableError{}))
 }
+
+func fromNamespaceNotFoundErr(e *types.NamespaceNotFoundError) error {
+	return protobuf.NewError(yarpcerrors.CodeNotFound, e.Error(), protobuf.WithErrorDetails(&sharddistributorv1.NamespaceNotFoundError{
+		Namespace: e.Namespace,
+	}))
+}

common/types/sharddistributor.go

@@ -22,6 +22,8 @@
 
 package types
 
+import "fmt"
+
 type GetShardOwnerRequest struct {
 	ShardKey  string
 	Namespace string
@@ -59,3 +61,14 @@ func (v *GetShardOwnerResponse) GetNamespace() (o string) {
 	}
 	return
 }
+
+type NamespaceNotFoundError struct {
+	Namespace string
+}
+
+func (n *NamespaceNotFoundError) Error() (o string) {
+	if n != nil {
+		return fmt.Sprintf("namespace not found %v", n.Namespace)
+	}
+	return
+}

common/types/testdata/common.go

@@ -58,6 +58,7 @@ const (
 	ClientLibraryVersion = "ClientLibraryVersion"
 	SupportedVersions    = "SupportedVersions"
 	FeatureFlag          = "FeatureFlag"
+	Namespace            = "Namespace"
 
 	Attempt            = 2
 	ScheduleID         = 5

common/types/testdata/error.go

@@ -118,6 +118,9 @@ var (
 		MyIdentity:      HostName2,
 		TasklistName:    TaskListName,
 	}
+	NamespaceNotFoundError = types.NamespaceNotFoundError{
+		Namespace: Namespace,
+	}
 )
 
 var Errors = []error{
@@ -143,4 +146,5 @@ var Errors = []error{
 	&WorkflowExecutionAlreadyStartedError,
 	&StickyWorkerUnavailableError,
 	&TaskListNotOwnedByHostError,
+	&NamespaceNotFoundError,
 }

proto/internal/uber/cadence/sharddistributor/v1/service.proto

@@ -40,3 +40,7 @@ message GetShardOwnerResponse {
   string owner = 1;
   string namespace = 2;
 }
+
+message NamespaceNotFoundError {
+  string namespace = 1;
+}