feat(validator): handle remove validator event

okp4 · Dec 23, 2022 · 237740b · 237740b
1 parent fbfb01a
commit 237740b
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/app/actor/subscription/actor.go b/app/actor/subscription/actor.go
@@ -96,6 +96,8 @@ func (a *Actor) receiveNewEvent(e event.Event) {
 		a.handleValidatorRegisteredEvent(e.Data)
 	case graphql.ValidatorUpdatedEventType:
 		a.handleValidatorUpdatedEvent(e.Data)
+	case graphql.ValidatorRemovedEventType:
+		a.handleValidatorRemovedEvent(e.Data)
 	case tweet.NewTweetEventType:
 		a.handleNewTweetEvent(e.Date, e.Data)
 	case graphql.TaskCompletedEventType:
@@ -232,6 +234,20 @@ func (a *Actor) handleValidatorUpdatedEvent(data map[string]interface{}) {
 	}
 }
 
+func (a *Actor) handleValidatorRemovedEvent(data map[string]interface{}) {
+	log.Info().Interface("event", data).Msg("Handle ValidatorRemoved event")
+
+	e := &graphql.ValidatorRemovedEvent{}
+	if err := e.Unmarshal(data); err != nil {
+		log.Panic().Err(err).Msg("❌ Failed unmarshal event to ValidatorRemoved")
+		return
+	}
+
+	if err := a.store.RemoveValidator(a.ctx, e.Validator); err != nil {
+		log.Err(err).Interface("data", data).Msg("🤕 Couldn't remove validator")
+	}
+}
+
 func (a *Actor) handleNewTweetEvent(when time.Time, data map[string]interface{}) {
 	log.Info().Interface("event", data).Msg("Handle NewTweet event")
 

diff --git a/app/nemeton/store.go b/app/nemeton/store.go
@@ -375,6 +375,16 @@ func (s *Store) UpdateValidator(
 	return err
 }
 
+func (s *Store) RemoveValidator(ctx context.Context, valoper types.ValAddress) error {
+	_, err := s.db.Collection(validatorsCollectionName).
+		DeleteOne(
+			ctx,
+			bson.M{
+				"valoper": valoper,
+			})
+	return err
+}
+
 func (s *Store) RegisterValidatorURL(ctx context.Context,
 	when time.Time,
 	urlType string,