fix

Author: luoluoyuyu

go-sdk/state/keyed/keyed_map_state.go

@@ -181,8 +181,14 @@ func (s *KeyedMapState[MK, MV]) All() iter.Seq2[MK, MV] {
 				return
 			}
 
-			k, _ := s.factory.mapKeyCodec.Decode(keyRaw)
-			v, _ := s.factory.mapValueCodec.Decode(valRaw)
+			k, err := s.factory.mapKeyCodec.Decode(keyRaw)
+			if err != nil {
+				continue // skip entry on decode error to avoid yielding corrupted zero values
+			}
+			v, err := s.factory.mapValueCodec.Decode(valRaw)
+			if err != nil {
+				continue
+			}
 
 			if !yield(k, v) {
 				return

go-sdk/state/keyed/keyed_priority_queue_state.go

@@ -145,7 +145,10 @@ func (s *KeyedPriorityQueueState[V]) Poll() (V, bool, error) {
 		return val, found, err
 	}
 
-	userKey, _ := s.factory.valueCodec.Encode(val)
+	userKey, err := s.factory.valueCodec.Encode(val)
+	if err != nil {
+		return val, true, fmt.Errorf("encode pq element for delete failed: %w", err)
+	}
 	ck := api.ComplexKey{
 		KeyGroup:  s.factory.groupKey,
 		Key:       s.primaryKey,
@@ -188,7 +191,10 @@ func (s *KeyedPriorityQueueState[V]) All() iter.Seq[V] {
 				return
 			}
 
-			v, _ := s.factory.valueCodec.Decode(userKey)
+			v, err := s.factory.valueCodec.Decode(userKey)
+			if err != nil {
+				continue // skip entry on decode error to avoid yielding corrupted zero values
+			}
 			if !yield(v) {
 				return
 			}

go-sdk/state/structures/map.go

@@ -156,8 +156,14 @@ func (m *MapState[K, V]) All() iter.Seq2[K, V] {
 				return
 			}
 
-			k, _ := m.keyCodec.Decode(keyRaw)
-			v, _ := m.valueCodec.Decode(valRaw)
+			k, err := m.keyCodec.Decode(keyRaw)
+			if err != nil {
+				continue // skip entry on decode error to avoid yielding corrupted zero values
+			}
+			v, err := m.valueCodec.Decode(valRaw)
+			if err != nil {
+				continue
+			}
 
 			if !yield(k, v) {
 				return

go-sdk/state/structures/priority_queue.go

@@ -115,7 +115,10 @@ func (q *PriorityQueueState[T]) Poll() (T, bool, error) {
 		return val, found, err
 	}
 
-	userKey, _ := q.valueCodec.Encode(val)
+	userKey, err := q.valueCodec.Encode(val)
+	if err != nil {
+		return val, true, fmt.Errorf("encode pq element for delete failed: %w", err)
+	}
 	if err = q.store.Delete(q.ck(userKey)); err != nil {
 		return val, true, err
 	}
@@ -149,7 +152,10 @@ func (q *PriorityQueueState[T]) All() iter.Seq[T] {
 				return
 			}
 
-			v, _ := q.valueCodec.Decode(userKey)
+			v, err := q.valueCodec.Decode(userKey)
+			if err != nil {
+				continue // skip entry on decode error to avoid yielding corrupted zero values
+			}
 			if !yield(v) {
 				return
 			}