[gql][indexer] index chain identifier into its own table

crates/sui-graphql-e2e-tests/tests/epoch/chain_identifier.exp

@@ -0,0 +1,16 @@
+processed 3 tasks
+
+init:
+C: object(0,0)
+
+task 1, line 7:
+//# create-checkpoint
+Checkpoint created: 1
+
+task 2, lines 9-12:
+//# run-graphql
+Response: {
+  "data": {
+    "chainIdentifier": "8f58ad28"
+  }
+}

crates/sui-graphql-e2e-tests/tests/epoch/chain_identifier.move

@@ -0,0 +1,12 @@
+// Copyright (c) Mysten Labs, Inc.
+// SPDX-License-Identifier: Apache-2.0
+
+//# init --protocol-version 48 --simulator --accounts C
+
+
+//# create-checkpoint
+
+//# run-graphql
+{
+    chainIdentifier
+}

crates/sui-graphql-rpc/src/types/chain_identifier.rs

@@ -6,8 +6,8 @@ use crate::{
     error::Error,
 };
 use async_graphql::*;
-use diesel::{ExpressionMethods, QueryDsl};
-use sui_indexer::schema::checkpoints;
+use diesel::QueryDsl;
+use sui_indexer::schema::chain_identifier;
 use sui_types::{
     digests::ChainIdentifier as NativeChainIdentifier, messages_checkpoint::CheckpointDigest,
 };
@@ -17,15 +17,11 @@ pub(crate) struct ChainIdentifier;
 impl ChainIdentifier {
     /// Query the Chain Identifier from the DB.
     pub(crate) async fn query(db: &Db) -> Result<NativeChainIdentifier, Error> {
-        use checkpoints::dsl;
+        use chain_identifier::dsl;
 
         let digest_bytes = db
             .execute(move |conn| {
-                conn.first(move || {
-                    dsl::checkpoints
-                        .select(dsl::checkpoint_digest)
-                        .order_by(dsl::sequence_number.asc())
-                })
+                conn.first(move || dsl::chain_identifier.select(dsl::checkpoint_digest))
             })
             .await
             .map_err(|e| Error::Internal(format!("Failed to fetch genesis digest: {e}")))?;

crates/sui-indexer/migrations/pg/2024-07-13-003534_chain_identifier/down.sql

@@ -0,0 +1,2 @@
+-- This file should undo anything in `up.sql`
+DROP TABLE IF EXISTS chain_identifier;

crates/sui-indexer/migrations/pg/2024-07-13-003534_chain_identifier/up.sql

@@ -0,0 +1,6 @@
+-- Your SQL goes here
+CREATE TABLE chain_identifier
+(
+    checkpoint_digest   BYTEA    NOT NULL,
+    PRIMARY KEY(checkpoint_digest)
+);

crates/sui-indexer/src/models/checkpoints.rs

@@ -9,9 +9,15 @@ use sui_types::digests::CheckpointDigest;
 use sui_types::gas::GasCostSummary;
 
 use crate::errors::IndexerError;
-use crate::schema::{checkpoints, pruner_cp_watermark};
+use crate::schema::{chain_identifier, checkpoints, pruner_cp_watermark};
 use crate::types::IndexedCheckpoint;
 
+#[derive(Queryable, Insertable, Selectable, Debug, Clone, Default)]
+#[diesel(table_name = chain_identifier)]
+pub struct StoredChainIdentifier {
+    pub checkpoint_digest: Vec<u8>,
+}
+
 #[derive(Queryable, Insertable, Selectable, Debug, Clone, Default)]
 #[diesel(table_name = checkpoints)]
 pub struct StoredCheckpoint {

crates/sui-indexer/src/schema/mod.rs

@@ -12,6 +12,7 @@ mod pg;
 
 #[cfg(feature = "postgres-feature")]
 mod inner {
+    pub use crate::schema::pg::chain_identifier;
     pub use crate::schema::pg::checkpoints;
     pub use crate::schema::pg::display;
     pub use crate::schema::pg::epochs;
@@ -33,6 +34,7 @@ mod inner {
 #[cfg(feature = "mysql-feature")]
 #[cfg(not(feature = "postgres-feature"))]
 mod inner {
+    pub use crate::schema::mysql::chain_identifier;
     pub use crate::schema::mysql::checkpoints;
     pub use crate::schema::mysql::display;
     pub use crate::schema::mysql::epochs;
@@ -51,6 +53,7 @@ mod inner {
     pub use crate::schema::mysql::tx_senders;
 }
 
+pub use inner::chain_identifier;
 pub use inner::checkpoints;
 pub use inner::display;
 pub use inner::epochs;

crates/sui-indexer/src/schema/mysql.rs

@@ -2,6 +2,12 @@
 // SPDX-License-Identifier: Apache-2.0
 // @generated automatically by Diesel CLI.
 
+diesel::table! {
+    chain_identifier (checkpoint_digest) {
+        checkpoint_digest -> Blob,
+    }
+}
+
 diesel::table! {
     checkpoints (sequence_number) {
         sequence_number -> Bigint,
@@ -227,6 +233,7 @@ diesel::table! {
 macro_rules! for_all_tables {
     ($action:path) => {
         $action!(
+            chain_identifier,
             checkpoints,
             epochs,
             events,

crates/sui-indexer/src/schema/pg.rs

@@ -2,6 +2,12 @@
 // SPDX-License-Identifier: Apache-2.0
 // @generated automatically by Diesel CLI.
 
+diesel::table! {
+    chain_identifier (checkpoint_digest) {
+        checkpoint_digest -> Bytea,
+    }
+}
+
 diesel::table! {
     checkpoints (sequence_number) {
         sequence_number -> Int8,
@@ -284,6 +290,7 @@ diesel::table! {
 macro_rules! for_all_tables {
     ($action:path) => {
         $action!(
+            chain_identifier,
             checkpoints,
             pruner_cp_watermark,
             display,

crates/sui-indexer/src/store/pg_indexer_store.rs

@@ -27,6 +27,7 @@ use crate::errors::{Context, IndexerError};
 use crate::handlers::EpochToCommit;
 use crate::handlers::TransactionObjectChangesToCommit;
 use crate::metrics::IndexerMetrics;
+use crate::models::checkpoints::StoredChainIdentifier;
 use crate::models::checkpoints::StoredCheckpoint;
 use crate::models::checkpoints::StoredCpTx;
 use crate::models::display::StoredDisplay;
@@ -39,9 +40,9 @@ use crate::models::objects::{
 use crate::models::packages::StoredPackage;
 use crate::models::transactions::StoredTransaction;
 use crate::schema::{
-    checkpoints, display, epochs, events, objects, objects_history, objects_snapshot, packages,
-    pruner_cp_watermark, transactions, tx_calls, tx_changed_objects, tx_digests, tx_input_objects,
-    tx_recipients, tx_senders,
+    chain_identifier, checkpoints, display, epochs, events, objects, objects_history,
+    objects_snapshot, packages, pruner_cp_watermark, transactions, tx_calls, tx_changed_objects,
+    tx_digests, tx_input_objects, tx_recipients, tx_senders,
 };
 use crate::types::{IndexedCheckpoint, IndexedEvent, IndexedPackage, IndexedTransaction, TxIndex};
 use crate::{
@@ -598,8 +599,27 @@ impl<T: R2D2Connection + 'static> PgIndexerStore<T> {
     }
 
     fn persist_checkpoints(&self, checkpoints: Vec<IndexedCheckpoint>) -> Result<(), IndexerError> {
-        if checkpoints.is_empty() {
+        let Some(first_checkpoint) = checkpoints.first() else {
             return Ok(());
+        };
+
+        // If the first checkpoint has sequence number 0, we need to persist the digest as
+        // chain identifier.
+        if first_checkpoint.sequence_number == 0 {
+            transactional_blocking_with_retry!(
+                &self.blocking_cp,
+                |conn| {
+                    let checkpoint_digest =
+                        first_checkpoint.checkpoint_digest.into_inner().to_vec();
+                    insert_or_ignore_into!(
+                        chain_identifier::table,
+                        StoredChainIdentifier { checkpoint_digest },
+                        conn
+                    );
+                    Ok::<(), IndexerError>(())
+                },
+                PG_DB_COMMIT_SLEEP_DURATION
+            )?;
         }
         let guard = self
             .metrics