add data source column

crates/sui-bridge-indexer/src/lib.rs

@@ -20,6 +20,7 @@ pub struct TokenTransfer {
  txn_hash: Vec<u8>,
  status: TokenTransferStatus,
  gas_usage: i64,
+ data_source: BridgeDataSource,
  data: Option<TokenTransferData>,
 }
 
@@ -41,6 +42,7 @@ impl From<TokenTransfer> for DBTokenTransfer {
  txn_hash: value.txn_hash,
  status: value.status.to_string(),
  gas_usage: value.gas_usage,
+ data_source: value.data_source.to_string(),
  }
  }
 }
@@ -86,3 +88,18 @@ impl Display for TokenTransferStatus {
  write!(f, "{str}")
  }
 }
+
+enum BridgeDataSource {
+ SUI,
+ ETH,
+}
+
+impl Display for BridgeDataSource {
+ fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+ let str = match self {
+ BridgeDataSource::ETH => "ETH",
+ BridgeDataSource::SUI => "SUI",
+ };
+ write!(f, "{str}")
+ }
+}

crates/sui-bridge-indexer/src/main.rs

@@ -75,7 +75,7 @@ async fn main() -> Result<()> {
 
  let contract_addresses = HashMap::from_iter(vec![(bridge_address, config.start_block)]);
 
- let (_task_handles, mut eth_events_rx, _) = EthSyncer::new(eth_client, contract_addresses)
+ let (_task_handles, eth_events_rx, _) = EthSyncer::new(eth_client, contract_addresses)
  .run()
  .await
  .expect("Failed to start eth syncer");

crates/sui-bridge-indexer/src/migrations/00000000000000_diesel_initial_setup/up.sql

@@ -21,5 +21,6 @@ CREATE TABLE token_transfer
  timestamp_ms BIGINT NOT NULL,
  txn_hash bytea NOT NULL,
  gas_usage BIGINT NOT NULL,
+ data_source TEXT NOT NULL,
  PRIMARY KEY(chain_id, nonce, status)
 );

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

@@ -14,6 +14,7 @@ pub struct TokenTransfer {
  pub timestamp_ms: i64,
  pub txn_hash: Vec<u8>,
  pub gas_usage: i64,
+ pub data_source: String,
 }
 
 #[derive(Queryable, Selectable, Insertable, Identifiable, Debug)]

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

@@ -12,6 +12,7 @@ diesel::table! {
  timestamp_ms -> Int8,
  txn_hash -> Bytea,
  gas_usage -> Int8,
+ data_source -> Text,
  }
 }
 

crates/sui-bridge-indexer/src/worker.rs

@@ -2,7 +2,7 @@
 // SPDX-License-Identifier: Apache-2.0
 
 use crate::postgres_writer::{get_connection_pool, write, PgPool};
-use crate::{TokenTransfer, TokenTransferData, TokenTransferStatus};
+use crate::{BridgeDataSource, TokenTransfer, TokenTransferData, TokenTransferStatus};
 use anyhow::Result;
 use async_trait::async_trait;
 use ethers::providers::Provider;
@@ -74,6 +74,7 @@ impl BridgeWorker {
  txn_hash: tx.transaction.digest().inner().to_vec(),
  status: TokenTransferStatus::Deposited,
  gas_usage: tx.effects.gas_cost_summary().net_gas_usage(),
+ data_source: BridgeDataSource::SUI,
  data: Some(TokenTransferData {
  sender_address: event.sender_address,
  destination_chain: event.target_chain,
@@ -95,6 +96,7 @@ impl BridgeWorker {
  txn_hash: tx.transaction.digest().inner().to_vec(),
  status: TokenTransferStatus::Approved,
  gas_usage: tx.effects.gas_cost_summary().net_gas_usage(),
+ data_source: BridgeDataSource::SUI,
  data: None,
  })
  }
@@ -110,6 +112,7 @@ impl BridgeWorker {
  txn_hash: tx.transaction.digest().inner().to_vec(),
  status: TokenTransferStatus::Claimed,
  gas_usage: tx.effects.gas_cost_summary().net_gas_usage(),
+ data_source: BridgeDataSource::SUI,
  data: None,
  })
  }
@@ -163,6 +166,7 @@ pub async fn process_eth_transaction(
  txn_hash: tx_hash.as_bytes().to_vec(),
  status: TokenTransferStatus::Deposited,
  gas_usage: gas.as_u64() as i64,
+ data_source: BridgeDataSource::ETH,
  data: Some(TokenTransferData {
  sender_address: bridge_event.sender_address.as_bytes().to_vec(),
  destination_chain: bridge_event.destination_chain_id,
@@ -184,15 +188,16 @@ pub async fn process_eth_transaction(
  txn_hash: tx_hash.as_bytes().to_vec(),
  status: TokenTransferStatus::Claimed,
  gas_usage: gas.as_u64() as i64,
+ data_source: BridgeDataSource::ETH,
  data: None,
  };
 
  write(&pool, transfer);
  }
- EthSuiBridgeEvents::PausedFilter(_bridge_event) => (),
- EthSuiBridgeEvents::UnpausedFilter(_bridge_event) => (),
- EthSuiBridgeEvents::UpgradedFilter(_bridge_event) => (),
- EthSuiBridgeEvents::InitializedFilter(_bridge_event) => (),
+ EthSuiBridgeEvents::PausedFilter(_)
+ | EthSuiBridgeEvents::UnpausedFilter(_)
+ | EthSuiBridgeEvents::UpgradedFilter(_)
+ | EthSuiBridgeEvents::InitializedFilter(_) => (),
  },
  }
  }