chore: replace State with CoreState in StateWatch and safe.StateWatch…

…Kind Somehow I missed those during the last update. Signed-off-by: Dmitriy Matrenichev <[email protected]>
cosi-project · Oct 30, 2023 · c4ad980 · c4ad980
1 parent c6658ca
commit c4ad980
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/safe/state.go b/pkg/safe/state.go
@@ -160,7 +160,7 @@ func watch[T resource.Resource](ctx context.Context, eventCh chan<- WrappedState
 }
 
 // StateWatch is a type safe wrapper around State.Watch.
-func StateWatch[T resource.Resource](ctx context.Context, st state.State, ptr resource.Pointer, eventCh chan<- WrappedStateEvent[T], opts ...state.WatchOption) error {
+func StateWatch[T resource.Resource](ctx context.Context, st state.CoreState, ptr resource.Pointer, eventCh chan<- WrappedStateEvent[T], opts ...state.WatchOption) error {
 	untypedEventCh := make(chan state.Event)
 
 	err := st.Watch(ctx, ptr, untypedEventCh, opts...)
@@ -181,7 +181,7 @@ func StateWatchFor[T resource.Resource](ctx context.Context, st state.State, ptr
 }
 
 // StateWatchKind is a type safe wrapper around State.WatchKind.
-func StateWatchKind[T resource.Resource](ctx context.Context, st state.State, kind resource.Kind, eventCh chan<- WrappedStateEvent[T], opts ...state.WatchKindOption) error {
+func StateWatchKind[T resource.Resource](ctx context.Context, st state.CoreState, kind resource.Kind, eventCh chan<- WrappedStateEvent[T], opts ...state.WatchKindOption) error {
 	untypedEventCh := make(chan state.Event)
 
 	err := st.WatchKind(ctx, kind, untypedEventCh, opts...)