Verified feat: add on pebble state backend and using sync Pool to reuse buffer []byte

1 files changed, 17 insertions, 13 deletions

diff --git a/state/pebble_backend.go b/state/pebble_backend.go
index e428988..fbb8d68 100644
--- a/state/pebble_backend.go
+++ b/state/pebble_backend.go
@@ -33,10 +33,11 @@ func NewPebbleStateBackend(dir string, maxParalleism int) (*PebbleStateBackend,
                 return nil, fmt.Errorf("failed to open pebbledb %w", err)
         }
         return &PebbleStateBackend{
-                maxParallelism: maxParalleism,
-                db:             db,
-                stateMap:       make(map[string]uint8),
-                nextStateID:    1,
+                maxParallelism:   maxParalleism,
+                db:               db,
+                stateMap:         make(map[string]uint8),
+                nextStateID:      1,
+                currentNamespace: &core.VoidNamespace{},
         }, nil
 }
 
@@ -60,23 +61,27 @@ func (p *PebbleStateBackend) GetCurrentNamespace() any {
 }
 
 // GetValueState(ctx context.Context, key string) (ValueState[any], error)
-func (p *PebbleStateBackend) GetValueState(ctx context.Context, statename string) (core.ValueState[any], error) {
-        stateID := p.getStateID(statename)
-        return nil, nil
+func (p *PebbleStateBackend) GetValueState(ctx context.Context, stateName string) (core.ValueState[any], error) {
+        stateID := p.getStateID(stateName)
+
+        return &pebbleStateValue{
+                backend: p,
+                stateID: stateID,
+        }, nil
 }
 
-func (p *PebbleStateBackend) getStateID(statename string) uint8 {
+func (p *PebbleStateBackend) getStateID(stateName string) uint8 {
         p.mu.Lock()
         defer p.mu.Unlock()
 
-        if _, ok := p.stateMap[statename]; !ok {
+        if _, ok := p.stateMap[stateName]; !ok {
 
                 id := p.nextStateID
-                p.stateMap[statename] = id
+                p.stateMap[stateName] = id
                 p.nextStateID++
         }
 
-        return p.stateMap[statename]
+        return p.stateMap[stateName]
 }
 
 func (p *PebbleStateBackend) calculateNamespaceSize() int {
@@ -94,7 +99,7 @@ func (p *PebbleStateBackend) calculateNamespaceSize() int {
         return nsLen
 }
 
-func (p *PebbleStateBackend) deserialize(statename string) []byte {
+func (p *PebbleStateBackend) deserialize(stateID uint8) []byte {
         // this will store in SSTable, order is important, keygroup MUST be first for scaling
         // 2B(KeyGroup) + 1B(StateID) + 2B(KeyLen) + len(Key) + 16B(Window)
 
@@ -105,7 +110,6 @@ func (p *PebbleStateBackend) deserialize(statename string) []byte {
         b = binary.BigEndian.AppendUint16(b, uint16(p.currentGroupKey))
 
         // add state id
-        stateID := p.getStateID(statename)
         b = append(b, stateID)
 
         // add key