ref(entities): split

Author: emptybutton

pyproject.toml

@@ -76,7 +76,7 @@ lines-after-imports = 2
 "src/tgdb/entities/*" = ["PLR2004"]
 "src/tgdb/application/*" = ["PLR0917"]
 "src/tgdb/infrastructure/adapters/*" = ["RUF029"]
-"src/tgdb/entities/transaction.py" = ["SLF001"]
+"src/tgdb/entities/horizon/transaction.py" = ["SLF001"]
 "tests/*" = ["PLR0124", "PLR0917", "S106", "C901", "PLR2004"]
 "__init__.py" = ["PLC0414"]
 

src/tgdb/entities/entity.py

@@ -0,0 +1,77 @@
+from collections.abc import Sequence, Set
+from dataclasses import dataclass
+from uuid import UUID
+
+from tgdb.entities.topic.partition import PartitionTuple, PartitionTupleID
+
+
+@dataclass(frozen=True)
+class ViewedTuple:
+    id: PartitionTupleID
+
+    def __and__(self, effect: "TupleEffect") -> "TupleEffect":
+        return effect
+
+
+@dataclass(frozen=True)
+class NewTuple:
+    tuple: PartitionTuple
+
+    def __and__(self, effect: "TupleEffect") -> "TupleEffect":
+        match effect:
+            case ViewedTuple():
+                return self
+            case MutatedTuple(tuple):
+                return NewTuple(tuple)
+            case NewTuple() | DeletedTuple():
+                return effect
+
+    @property
+    def id(self) -> PartitionTupleID:
+        return self.tuple.id
+
+
+@dataclass(frozen=True)
+class MutatedTuple:
+    tuple: PartitionTuple
+
+    def __and__(self, effect: "TupleEffect") -> "TupleEffect":
+        match effect:
+            case ViewedTuple():
+                return self
+            case NewTuple(tuple):
+                return MutatedTuple(tuple)
+            case MutatedTuple() | DeletedTuple():
+                return effect
+
+    @property
+    def id(self) -> PartitionTupleID:
+        return self.tuple.id
+
+
+@dataclass(frozen=True)
+class DeletedTuple:
+    id: PartitionTupleID
+
+    def __and__(self, effect: "TupleEffect") -> "TupleEffect":
+        match effect:
+            case ViewedTuple() | MutatedTuple() | DeletedTuple():
+                return self
+            case NewTuple(tuple):
+                return MutatedTuple(tuple)
+
+
+type TupleEffect = NewTuple | ViewedTuple | MutatedTuple | DeletedTuple
+
+
+@dataclass(frozen=True)
+class Claim:
+    id: UUID
+    object: str
+
+
+type ConflictableTransactionScalarEffect = TupleEffect | Claim
+type ConflictableTransactionEffect = Sequence[TupleEffect | Claim]
+
+type TransactionScalarEffect = NewTuple | MutatedTuple | DeletedTuple
+type TransactionEffect = Set[TransactionScalarEffect]

src/tgdb/entities/horizon/__init__.py

@@ -3,17 +3,15 @@
 from dataclasses import dataclass
 from uuid import UUID
 
-from tgdb.entities.assert_ import not_none
-from tgdb.entities.logic_time import LogicTime, age
-from tgdb.entities.operator import (
+from tgdb.entities.horizon.operator import (
     AppliedOperator,
     CommitOperator,
     IntermediateOperator,
     Operator,
     RollbackOperator,
     StartOperator,
 )
-from tgdb.entities.transaction import (
+from tgdb.entities.horizon.transaction import (
     NoTransaction,
     Transaction,
     TransactionCommit,
@@ -22,42 +20,44 @@
     TransactionState,
     start_transaction,
 )
+from tgdb.entities.time.logic_time import LogicTime, age
+from tgdb.entities.tools.assert_ import not_none
 
 
 class NonLinearizedOperatorError(Exception): ...
 
 
-class UnlimitedTransactionHorizonError(Exception): ...
+class UnlimitedHorizonError(Exception): ...
 
 
-class UnattainableTransactionHorizonError(Exception): ...
+class UnattainableHorizonError(Exception): ...
 
 
 class UselessMaxHeightError(Exception): ...
 
 
 @dataclass
-class TransactionHorizon:
+class Horizon:
     _max_width: LogicTime | None
     _max_height: int | None
     _time: LogicTime | None
     _active_transaction_by_id: OrderedDict[UUID, Transaction]
 
     def __post_init__(self) -> None:
         """
-        :raises tgdb.entities.transaction_horizon.UnlimitedTransactionHorizonError:
-        :raises tgdb.entities.transaction_horizon.UnattainableTransactionHorizonError:
-        :raises tgdb.entities.transaction_horizon.UselessMaxHeightError:
-        """  # noqa: E501
+        :raises tgdb.entities.horizon.horizon.UnlimitedHorizonError:
+        :raises tgdb.entities.horizon.horizon.UnattainableHorizonError:
+        :raises tgdb.entities.horizon.horizon.UselessMaxHeightError:
+        """
 
         if self._max_width is None and self._max_height is None:
-            raise UnlimitedTransactionHorizonError
+            raise UnlimitedHorizonError
 
         if self._max_width is not None and self._max_width <= 0:
-            raise UnattainableTransactionHorizonError
+            raise UnattainableHorizonError
 
         if self._max_height is not None and self._max_height <= 0:
-            raise UnattainableTransactionHorizonError
+            raise UnattainableHorizonError
 
         if (
             self._max_height is not None and self._max_width is not None
@@ -107,7 +107,7 @@ def add(
         self, applied_operator: AppliedOperator
     ) -> TransactionPreparedCommit | None:
         """
-        :raises tgdb.entities.transaction_horizon.NonLinearizedOperatorError:
+        :raises tgdb.entities.horizon.horizon.NonLinearizedOperatorError:
         """
 
         if self._time is not None and applied_operator.time <= self._time:
@@ -138,15 +138,15 @@ def _add_operator(
                 Transaction(),
                 TransactionState.active
             ):
-                transaction.add_effect(effect)
+                transaction.include(effect)
 
             case (
                 CommitOperator(_, intermediate_operators),
                 Transaction(),
                 TransactionState.active
             ):
                 for intermediate_operator in intermediate_operators:
-                    transaction.add_effect(intermediate_operator.effect)
+                    transaction.include(intermediate_operator.effect)
 
                 return transaction.prepare_commit()
 
@@ -191,17 +191,17 @@ def _oldest_transaction(self) -> Transaction | None:
             return None
 
 
-def create_transaction_horizon(
+def create_horizon(
     max_width: LogicTime | None,
     max_height: int | None,
-) -> TransactionHorizon:
+) -> Horizon:
+    """
+    :raises tgdb.entities.horizon.horizon.UnlimitedHorizonError:
+    :raises tgdb.entities.horizon.horizon.UnattainableHorizonError:
+    :raises tgdb.entities.horizon.horizon.UselessMaxHeightError:
     """
-    :raises tgdb.entities.transaction_horizon.UnlimitedTransactionHorizonError:
-    :raises tgdb.entities.transaction_horizon.UnattainableTransactionHorizonError:
-    :raises tgdb.entities.transaction_horizon.UselessMaxHeightError:
-    """  # noqa: E501
 
-    return TransactionHorizon(
+    return Horizon(
         _max_width=max_width,
         _max_height=max_height,
         _time=None,