Merge pull request #41 from PowerLoom/feat/epoch-monitoring

Record state transition: snapshot submission to relayer
PowerLoom · Aug 21, 2023 · 935fd4d · 935fd4d
2 parents 1d8b1ae + 406fdba
commit 935fd4d
Show file tree

Hide file tree

Showing 4 changed files with 44 additions and 1 deletion.
diff --git a/go/caching/redis.go b/go/caching/redis.go
@@ -33,6 +33,36 @@ func NewRedisCache(readClient, writeClient *redis.Client) *RedisCache {
 	return cache
 }
 
+func (r *RedisCache) UpdateEpochProcessingStatus(ctx context.Context, projectID string, epochId int, status string, err string) error {
+	htable := fmt.Sprintf(redisutils.REDIS_KEY_EPOCH_STATE_ID, strconv.Itoa(epochId), datamodel.RELAYER_SEND_STATE_ID)
+
+	l := log.WithField("hash table", htable).
+		WithField("projectID", projectID).WithField("epochId", epochId).WithField("status", status)
+
+	transition_status_item := datamodel.SnapshotterStateUpdate{
+		Status:    status,
+		Error:     err,
+		Timestamp: time.Now().Unix(),
+	}
+	state_update_bytes, err2 := json.Marshal(transition_status_item)
+
+	if err2 != nil {
+		log.WithError(err2).Error("failed to marshal state update message on relayer submission")
+		return err2
+	}
+	err_ := r.writeClient.HSet(
+		ctx,
+		htable,
+		projectID,
+		string(state_update_bytes),
+	).Err()
+	if err_ != nil {
+		l.WithError(err_).Error("failed to update epoch processing status in redis")
+		return err_
+	}
+	return nil
+}
+
 func (r *RedisCache) GetUnfinalizedSnapshotAtEpochID(ctx context.Context, projectID string, epochId int) (*datamodel.UnfinalizedSnapshot, error) {
 	key := fmt.Sprintf(redisutils.REDIS_KEY_PROJECT_UNFINALIZED_SNAPSHOT_CIDS, projectID)
 	snapshotCid := ""

diff --git a/go/goutils/datamodel/data_model.go b/go/goutils/datamodel/data_model.go
@@ -9,6 +9,15 @@ const (
 	IncorrectSnapshotSubmission SnapshotSubmissionState = "SUBMITTED_INCORRECT_SNAPSHOT"
 )
 
+type SnapshotterStateUpdate struct {
+	Status    string                 `json:"status"`
+	Error     string                 `json:"error"`
+	Extra     map[string]interface{} `json:"extra"`
+	Timestamp int64                  `json:"timestamp"`
+}
+
+const RELAYER_SEND_STATE_ID string = "RELAYER_SEND"
+
 type SummaryProjectVerificationStatus struct {
 	ProjectId     string `json:"projectId"`
 	ProjectHeight string `json:"chainHeight"`

diff --git a/go/goutils/redisutils/redis_keys.go b/go/goutils/redisutils/redis_keys.go
@@ -9,4 +9,5 @@ const (
 	REDIS_KEY_TOTAL_INCORRECT_SNAPSHOT_COUNT    string = "projectID:%s:totalIncorrectSnapshotCount"
 	REDIS_KEY_LAST_FINALIZED_EPOCH              string = "projectID:%s:lastFinalizedEpoch"
 	REDIS_KEY_FINALIZED_SNAPSHOTS               string = "projectID:%s:finalizedSnapshots"
+	REDIS_KEY_EPOCH_STATE_ID					string = "epochID:%s:stateID:%s:processingStatus"
 )
diff --git a/go/payload-commit/service/service.go b/go/payload-commit/service/service.go
@@ -252,6 +252,7 @@ func (s *PayloadCommitService) HandlePayloadCommitTask(msg *datamodel.PayloadCom
 					"issueDetails": "Error: " + err.Error(),
 					"msg":          "failed to submit snapshot to contract",
 				})
+			go s.redisCache.UpdateEpochProcessingStatus(context.Background(), msg.ProjectID, msg.EpochID, "failed", err.Error())
 
 			go s.redisCache.AddSnapshotterStatusReport(context.Background(), msg.EpochID, msg.ProjectID, &datamodel.SnapshotterStatusReport{
 				SubmittedSnapshotCid: msg.SnapshotCID,
@@ -261,6 +262,7 @@ func (s *PayloadCommitService) HandlePayloadCommitTask(msg *datamodel.PayloadCom
 
 			return err
 		}
+		go s.redisCache.UpdateEpochProcessingStatus(context.Background(), msg.ProjectID, msg.EpochID, "success", "")
 	} else {
 		// send payload commit message with signature to relayer
 		err = s.sendSignatureToRelayer(txPayload)
@@ -275,7 +277,7 @@ func (s *PayloadCommitService) HandlePayloadCommitTask(msg *datamodel.PayloadCom
 					"issueDetails": "Error: " + err.Error(),
 					"msg":          "failed to submit snapshot to relayer",
 				})
-
+			go s.redisCache.UpdateEpochProcessingStatus(context.Background(), msg.ProjectID, msg.EpochID, "failed", err.Error())
 			go s.redisCache.AddSnapshotterStatusReport(context.Background(), msg.EpochID, msg.ProjectID, &datamodel.SnapshotterStatusReport{
 				SubmittedSnapshotCid: msg.SnapshotCID,
 				State:                datamodel.MissedSnapshotSubmission,
@@ -284,6 +286,7 @@ func (s *PayloadCommitService) HandlePayloadCommitTask(msg *datamodel.PayloadCom
 
 			return err
 		}
+		go s.redisCache.UpdateEpochProcessingStatus(context.Background(), msg.ProjectID, msg.EpochID, "success", "")
 	}
 
 	// store unfinalized payload cid in redis