Move cache helpers to separate module

Author: gwaramadze

quixstreams/state/rocksdb/cache.py

@@ -0,0 +1,16 @@
+from dataclasses import dataclass, field
+from typing import Optional
+
+
+@dataclass
+class TimestampsCache:
+    key: bytes
+    cf_name: str
+    timestamps: dict[bytes, Optional[int]] = field(default_factory=dict)
+
+
+@dataclass
+class CounterCache:
+    key: bytes
+    cf_name: str
+    counter: Optional[int] = None

quixstreams/state/rocksdb/timestamped.py

@@ -7,12 +7,12 @@
 )
 from quixstreams.state.metadata import SEPARATOR
 from quixstreams.state.recovery import ChangelogProducer
+from quixstreams.state.rocksdb.cache import TimestampsCache
 from quixstreams.state.rocksdb.metadata import (
     LATEST_TIMESTAMP_KEY,
     LATEST_TIMESTAMPS_CF_NAME,
 )
 from quixstreams.state.rocksdb.types import RocksDBOptionsType
-from quixstreams.state.rocksdb.windowed.transaction import TimestampsCache
 from quixstreams.state.serialization import (
     DumpsFunc,
     LoadsFunc,

quixstreams/state/rocksdb/windowed/transaction.py

@@ -1,5 +1,4 @@
 import itertools
-from dataclasses import dataclass, field
 from typing import TYPE_CHECKING, Any, Iterable, Optional, cast
 
 from quixstreams.state.base.transaction import (
@@ -9,6 +8,7 @@
 )
 from quixstreams.state.metadata import DEFAULT_PREFIX, SEPARATOR
 from quixstreams.state.recovery import ChangelogProducer
+from quixstreams.state.rocksdb.cache import CounterCache, TimestampsCache
 from quixstreams.state.serialization import (
     DumpsFunc,
     LoadsFunc,
@@ -41,20 +41,6 @@
     from .partition import WindowedRocksDBStorePartition
 
 
-@dataclass
-class TimestampsCache:
-    key: bytes
-    cf_name: str
-    timestamps: dict[bytes, Optional[int]] = field(default_factory=dict)
-
-
-@dataclass
-class CounterCache:
-    key: bytes
-    cf_name: str
-    counter: Optional[int] = None
-
-
 class WindowedRocksDBPartitionTransaction(PartitionTransaction):
     def __init__(
         self,