comment indexer

Author: stereosteve

indexer/index_comment.go

@@ -0,0 +1,29 @@
+package indexer
+
+import (
+	"encoding/json"
+
+	"github.com/AudiusProject/audiusd/pkg/core/gen/core_proto"
+	"github.com/jackc/pgx/v5"
+)
+
+func (ci *CoreIndexer) createComment(txInfo TxInfo, em *core_proto.ManageEntityLegacy) error {
+	var metadata GenericMetadata
+	if err := json.Unmarshal([]byte(em.Metadata), &metadata); err != nil {
+		return err
+	}
+
+	args := pgx.NamedArgs{
+		"comment_id":  metadata.Data["comment_id"],
+		"user_id":     em.UserId,
+		"text":        metadata.Data["body"],
+		"entity_type": "track",
+		"entity_id":   metadata.Data["track_id"],
+
+		"txhash":      txInfo.txhash,
+		"blockhash":   txInfo.blockhash,
+		"blocknumber": txInfo.blocknumber,
+	}
+
+	return ci.doInsert("comments", args)
+}

indexer/indexer.go

@@ -82,6 +82,9 @@ func (ci *CoreIndexer) handleTx(signedTx *core_proto.SignedTransaction) error {
 		case "UpdatePlaylist":
 			err = ci.updatePlaylist(txInfo, em)
 
+		case "CreateComment":
+			err = ci.createComment(txInfo, em)
+
 		case "FollowUser":
 			err = ci.followUser(txInfo, em)
 		case "UnfollowUser":

indexer/indexer_test.go

@@ -52,6 +52,7 @@ func TestMain(m *testing.M) {
 	alter table saves drop constraint saves_blocknumber_fkey;
 	alter table notification_seen drop constraint notification_seen_blocknumber_fkey;
 	alter table track_downloads drop constraint track_downloads_blocknumber_fkey;
+	alter table comments drop constraint comments_blocknumber_fkey;
 	`)
 	checkErr(err)