Expose namespace data (#2804)

* Expose namespace data
temporalio · May 5, 2022 · ddb29dc · ddb29dc
1 parent ae84776
commit ddb29dc
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/common/namespace/namespace.go b/common/namespace/namespace.go
@@ -224,6 +224,13 @@ func (ns *Namespace) ReplicationPolicy() ReplicationPolicy {
 	return ReplicationPolicyOneCluster
 }
 
+func (ns *Namespace) GetCustomData(key string) string {
+	if ns.info.Data == nil {
+		return ""
+	}
+	return ns.info.Data[key]
+}
+
 // Len return length
 func (t Namespaces) Len() int {
 	return len(t)

diff --git a/common/namespace/namespace_test.go b/common/namespace/namespace_test.go
@@ -28,6 +28,8 @@ import (
 	"testing"
 	"time"
 
+	"github.com/stretchr/testify/assert"
+
 	"github.com/google/uuid"
 	"github.com/stretchr/testify/require"
 	namespacepb "go.temporal.io/api/namespace/v1"
@@ -131,3 +133,12 @@ func Test_GetRetentionDays(t *testing.T) {
 		})
 	}
 }
+
+func TestNamespace_GetCustomData(t *testing.T) {
+	base := base(t)
+	ns := base.Clone(namespace.WithData("foo", "bar"))
+	data := ns.GetCustomData("foo")
+	assert.Equal(t, "bar", data)
+	data2 := ns.GetCustomData("fake")
+	assert.Equal(t, "", data2)
+}