Verified feat: init base draft support window

1 files changed, 44 insertions, 0 deletions

diff --git a/state/memory_state.go b/state/memory_state.go
new file mode 100644
index 0000000..95c7318
--- /dev/null
+++ b/state/memory_state.go
@@ -0,0 +1,44 @@
+package state
+
+import (
+"context"
+)
+
+type memoryStateValue struct {
+        backend   *MemoryStateBackend
+        stateName string
+}
+
+func (m *memoryStateValue) Value(ctx context.Context) (any, error) {
+        kg := m.backend.currentGroupKey
+        key := m.backend.currentKey
+        ns := m.backend.currentNamespace
+        if kgMap, exist := m.backend.states[m.stateName][kg][key]; exist {
+                if v, hasKey := kgMap[ns]; hasKey {
+                        return v, nil
+                }
+        }
+
+        return nil, nil
+}
+
+func (m *memoryStateValue) Update(ctx context.Context, newVal any) (any, error) {
+        kg := m.backend.currentGroupKey
+        key := m.backend.currentKey
+        ns := m.backend.currentNamespace
+        if _, exists := m.backend.states[m.stateName][kg][key]; !exists {
+                m.backend.states[m.stateName][kg] = make(map[string]map[any]any)
+        }
+
+        m.backend.states[m.stateName][kg][key][ns] = newVal
+        return newVal, nil
+}
+
+func (m *memoryStateValue) Clear(ctx context.Context) error {
+        kg := m.backend.currentGroupKey
+        key := m.backend.currentKey
+        if kgMap, exists := m.backend.states[m.stateName][kg]; exists {
+                delete(kgMap, key)
+        }
+        return nil
+}