sui_indexer/backfill/backfill_instances/
mod.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
// Copyright (c) Mysten Labs, Inc.
// SPDX-License-Identifier: Apache-2.0

use crate::backfill::backfill_instances::ingestion_backfills::digest_task::DigestBackfill;
use crate::backfill::backfill_instances::ingestion_backfills::ingestion_backfill_task::IngestionBackfillTask;
use crate::backfill::backfill_instances::ingestion_backfills::raw_checkpoints::RawCheckpointsBackFill;
use crate::backfill::backfill_instances::ingestion_backfills::tx_affected_objects::TxAffectedObjectsBackfill;
use crate::backfill::backfill_task::BackfillTask;
use crate::backfill::{BackfillTaskKind, IngestionBackfillKind};
use std::sync::Arc;
use sui_types::messages_checkpoint::CheckpointSequenceNumber;

mod ingestion_backfills;
mod sql_backfill;
mod system_state_summary_json;

pub async fn get_backfill_task(
    kind: BackfillTaskKind,
    range_start: usize,
) -> Arc<dyn BackfillTask> {
    match kind {
        BackfillTaskKind::SystemStateSummaryJson => {
            Arc::new(system_state_summary_json::SystemStateSummaryJsonBackfill)
        }
        BackfillTaskKind::Sql { sql, key_column } => {
            Arc::new(sql_backfill::SqlBackFill::new(sql, key_column))
        }
        BackfillTaskKind::Ingestion {
            kind,
            remote_store_url,
        } => match kind {
            IngestionBackfillKind::Digest => Arc::new(
                IngestionBackfillTask::<DigestBackfill>::new(
                    remote_store_url,
                    range_start as CheckpointSequenceNumber,
                )
                .await,
            ),
            IngestionBackfillKind::RawCheckpoints => Arc::new(
                IngestionBackfillTask::<RawCheckpointsBackFill>::new(
                    remote_store_url,
                    range_start as CheckpointSequenceNumber,
                )
                .await,
            ),
            IngestionBackfillKind::TxAffectedObjects => Arc::new(
                IngestionBackfillTask::<TxAffectedObjectsBackfill>::new(
                    remote_store_url,
                    range_start as CheckpointSequenceNumber,
                )
                .await,
            ),
        },
    }
}