[dbnode] GetNamespaceFn returns public Namespace interface (#2696)

m3db · Oct 6, 2020 · 47c219a · 47c219a
1 parent ac2ef9b
commit 47c219a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/dbnode/storage/database.go b/src/dbnode/storage/database.go
@@ -412,7 +412,7 @@ func (d *db) addNamespacesWithLock(namespaces []namespace.Metadata) error {
 
 	hooks := d.Options().NamespaceHooks()
 	for _, ns := range createdNamespaces {
-		err := hooks.OnCreatedNamespace(ns, d.namespaces.Get)
+		err := hooks.OnCreatedNamespace(ns, d.getNamespaceWithLock)
 		if err != nil {
 			return err
 		}
@@ -421,6 +421,10 @@ func (d *db) addNamespacesWithLock(namespaces []namespace.Metadata) error {
 	return nil
 }
 
+func (d *db) getNamespaceWithLock(id ident.ID) (Namespace, bool) {
+	return d.namespaces.Get(id)
+}
+
 func (d *db) newDatabaseNamespaceWithLock(
 	md namespace.Metadata,
 ) (databaseNamespace, error) {

diff --git a/src/dbnode/storage/types.go b/src/dbnode/storage/types.go
@@ -1316,4 +1316,4 @@ type NamespaceHooks interface {
 	OnCreatedNamespace(Namespace, GetNamespaceFn) error
 }
 
-type GetNamespaceFn func (k ident.ID) (databaseNamespace, bool)
+type GetNamespaceFn func (id ident.ID) (Namespace, bool)