feat: add StatusAt field in update event

sentinel-official · Jul 15, 2024 · be8450c · be8450c
1 parent 6278a21
commit be8450c
Show file tree

Hide file tree

Showing 6 changed files with 80 additions and 30 deletions.
diff --git a/proto/sentinel/session/v3/events.proto b/proto/sentinel/session/v3/events.proto
@@ -40,4 +40,5 @@ message EventUpdateStatus {
   string acc_address = 2;
   string node_address = 3;
   sentinel.types.v1.Status status = 4;
+  string status_at = 5;
 }
diff --git a/x/session/keeper/abci.go b/x/session/keeper/abci.go
@@ -60,6 +60,7 @@ func (k *Keeper) handleInactiveSessions(ctx sdk.Context) {
 				AccAddress:  item.GetAccAddress(),
 				NodeAddress: item.GetNodeAddress(),
 				Status:      v1base.StatusInactive,
+				StatusAt:    ctx.BlockTime().String(),
 			},
 		)
 

diff --git a/x/session/keeper/msg_handler.go b/x/session/keeper/msg_handler.go
@@ -48,6 +48,7 @@ func (k *Keeper) HandleMsgCancelSession(ctx sdk.Context, msg *v3.MsgCancelSessio
 			AccAddress:  session.GetAccAddress(),
 			NodeAddress: session.GetNodeAddress(),
 			Status:      session.GetStatus(),
+			StatusAt:    session.GetStatusAt().String(),
 		},
 	)
 

diff --git a/x/session/types/v3/events.pb.go b/x/session/types/v3/events.pb.go
diff --git a/x/session/types/v3/session.go b/x/session/types/v3/session.go
@@ -19,6 +19,7 @@ type Session interface {
 	GetInactiveAt() time.Time
 	GetNodeAddress() string
 	GetStatus() v1base.Status
+	GetStatusAt() time.Time
 	GetUploadBytes() sdkmath.Int
 
 	SetDownloadBytes(sdkmath.Int)

diff --git a/x/subscription/keeper/abci.go b/x/subscription/keeper/abci.go
@@ -55,6 +55,7 @@ func (k *Keeper) handleInactiveSubscriptions(ctx sdk.Context) {
 				PlanID:     item.PlanID,
 				AccAddress: item.AccAddress,
 				Status:     v1base.StatusInactive,
+				StatusAt:   ctx.BlockTime().String(),
 			},
 		)