chore: revert drop of allocation_nonce from users table

golemfoundation · Mar 27, 2024 · 6c617af · 6c617af
1 parent e04c1bc
commit 6c617af
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 29 deletions.
diff --git a/backend/app/infrastructure/database/models.py b/backend/app/infrastructure/database/models.py
@@ -20,6 +20,11 @@ class User(BaseModel):
 
     id = Column(db.Integer, primary_key=True)
     address = Column(db.String(42), unique=True, nullable=False)
+    allocation_nonce = Column(
+        db.Integer,
+        nullable=True,
+        comment="Allocations signing nonce, last used value. Range [0..inf)",
+    )
 
     def get_effective_deposit(self, epoch: int) -> Optional[int]:
         effective_deposit = None

diff --git a/backend/app/modules/user/allocations/service/pending.py b/backend/app/modules/user/allocations/service/pending.py
@@ -48,6 +48,9 @@ def allocate(
         )
 
         self.revoke_previous_allocation(context, user_address)
+
+        user = database.user.get_by_address(user_address)
+        user.allocation_nonce = expected_nonce
         database.allocations.store_allocation_request(
             user_address, context.epoch_details.epoch_num, payload, **kwargs
         )

diff --git a/backend/migrations/versions/1f89a0fae4ae_drop_allocation_nonce_column.py b/backend/migrations/versions/1f89a0fae4ae_drop_allocation_nonce_column.py