chore: replace sync.Map with HashTrieMap

Use in for `inmem` and `namespaced` packages. Closes #455 Signed-off-by: Dmitriy Matrenichev <[email protected]>
cosi-project · Jun 3, 2024 · 2b7f173 · 2b7f173
1 parent b2c32a5
commit 2b7f173
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/pkg/state/impl/inmem/inmem.go b/pkg/state/impl/inmem/inmem.go
@@ -10,7 +10,7 @@ import (
 	"sync"
 	"sync/atomic"
 
-	"github.com/siderolabs/gen/containers"
+	"github.com/siderolabs/gen/concurrent"
 
 	"github.com/cosi-project/runtime/pkg/resource"
 	"github.com/cosi-project/runtime/pkg/state"
@@ -20,7 +20,7 @@ var _ state.CoreState = &State{}
 
 // State implements state.CoreState.
 type State struct {
-	collections containers.SyncMap[resource.Type, *ResourceCollection]
+	collections *concurrent.HashTrieMap[resource.Type, *ResourceCollection]
 	store       BackingStore
 
 	ns resource.Namespace
@@ -44,11 +44,12 @@ func NewStateWithOptions(opts ...StateOption) func(ns resource.Namespace) *State
 
 	return func(ns resource.Namespace) *State {
 		return &State{
+			collections:     concurrent.NewHashTrieMap[resource.Type, *ResourceCollection](),
+			store:           options.BackingStore,
 			ns:              ns,
 			initialCapacity: options.HistoryInitialCapacity,
 			maxCapacity:     options.HistoryMaxCapacity,
 			gap:             options.HistoryGap,
-			store:           options.BackingStore,
 		}
 	}
 }

diff --git a/pkg/state/impl/namespaced/namespaced.go b/pkg/state/impl/namespaced/namespaced.go
@@ -8,7 +8,7 @@ package namespaced
 import (
 	"context"
 
-	"github.com/siderolabs/gen/containers"
+	"github.com/siderolabs/gen/concurrent"
 
 	"github.com/cosi-project/runtime/pkg/resource"
 	"github.com/cosi-project/runtime/pkg/state"
@@ -23,13 +23,14 @@ var _ state.CoreState = (*State)(nil)
 type State struct {
 	builder StateBuilder
 
-	namespaces containers.SyncMap[resource.Namespace, state.CoreState]
+	namespaces *concurrent.HashTrieMap[resource.Namespace, state.CoreState]
 }
 
 // NewState initializes new namespaced State.
 func NewState(builder StateBuilder) *State {
 	return &State{
-		builder: builder,
+		builder:    builder,
+		namespaces: concurrent.NewHashTrieMap[resource.Namespace, state.CoreState](),
 	}
 }