From 4ca27aec303f54dc9ee670c67c2fefa80dd004a1 Mon Sep 17 00:00:00 2001 From: SangTran-127 Date: Sun, 9 Aug 2026 18:32:44 +0700 Subject: feat: add KeyByOperator and wordCounter for stateful processing --- core/state_backend.go | 13 +++++++++++++ 1 file changed, 13 insertions(+) (limited to 'core/state_backend.go') diff --git a/core/state_backend.go b/core/state_backend.go index 219bb63..f8b83fb 100644 --- a/core/state_backend.go +++ b/core/state_backend.go @@ -11,3 +11,16 @@ type StateBackend interface { // GetValueState will perform in disk GetValueState(ctx context.Context, key string) (ValueState[any], error) } + +type stateBackendKeyType struct{} + +var stateBackendKey stateBackendKeyType + +func InjectStateBackend(ctx context.Context, state StateBackend) context.Context { + return context.WithValue(ctx, stateBackendKey, state) +} + +func ExtractStateBackend(ctx context.Context) (StateBackend, bool) { + backend, ok := ctx.Value(stateBackendKey).(StateBackend) + return backend, ok +} -- cgit v1.2.3