feat(core): insert into indexer table

ArkProjectNFTs · Sep 7, 2023 · 521ab13 · 521ab13
1 parent d60e5d2
commit 521ab13
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 10 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -16,6 +16,7 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
+chrono = "0.4.19"
 anyhow = "1.0"
 tokio = { version = "1", features = ["full"] }
 reqwest = { version = "0.11", features = ["json"] }

diff --git a/src/main.rs b/src/main.rs
@@ -7,8 +7,9 @@ use crate::core::block::process_blocks_continuously;
 use anyhow::Result;
 use ark_starknet::{client2::StarknetClient, collection_manager::CollectionManager};
 use aws_config::meta::region::RegionProviderChain;
-use aws_sdk_dynamodb::Client as DynamoClient;
+use aws_sdk_dynamodb::{types::AttributeValue, Client as DynamoClient};
 use aws_sdk_kinesis::Client as KinesisClient;
+use chrono::Utc;
 use dotenv::dotenv;
 use reqwest::{Client as ReqwestClient, Url};
 use simple_logger::SimpleLogger;
@@ -33,6 +34,21 @@ async fn main() -> Result<()> {
     let dynamo_client = DynamoClient::new(&config);
     let reqwest_client = ReqwestClient::new();
 
+    let indexer_table_name =
+        env::var("ARK_INDEXER_TABLE_NAME").expect("ARK_INDEXER_TABLE_NAME must be set");
+
+    let indexer_version = env::var("ARK_INDEXER_VERSION").unwrap_or(String::from("undefined"));
+    let now = Utc::now();
+    let unix_timestamp = now.timestamp();
+
+    dynamo_client
+        .put_item()
+        .table_name(indexer_table_name)
+        .item("PK", AttributeValue::S(String::from("ECS_")))
+        .item("SK", AttributeValue::S(unix_timestamp.to_string()))
+        .item("status", AttributeValue::S(String::from("running")))
+        .item("version", AttributeValue::S(String::from(indexer_version)));
+
     process_blocks_continuously(
         &collection_manager,
         &rpc_client,