Function snapshot

apps/hash-graph/src/subcommand/snapshot.rs:84–146

(args: SnapshotArgs)

}

pub async fn snapshot(args: SnapshotArgs) -> Result<(), Report<GraphError>> {
    SnapshotEntry::install_error_stack_hook();

    let mut settings = PostgresStoreSettings::default();
    if let SnapshotCommand::Restore(args) = &args.command {
        settings.validate_links = !args.skip_validation;
    }

    let pool = PostgresStorePool::new(&args.db_info, &args.pool_config, NoTls, settings)
        .await
        .change_context(GraphError)
        .map_err(|report| {
            tracing::error!(error = ?report, "Failed to connect to database");
            report
        })?;

    match args.command {
        SnapshotCommand::Dump(args) => {
            let write = FramedWrite::new(
                io::BufWriter::new(io::stdout()),
                JsonLinesEncoder::default(),
            );
            let settings = SnapshotDumpSettings {
                chunk_size: 10_000,
                dump_principals: !args.no_principals,
                dump_actions: !args.no_actions,
                dump_policies: !args.no_policies,
                dump_entities: !args.no_entities,
                dump_entity_types: !args.no_entity_types,
                dump_property_types: !args.no_property_types,
                dump_data_types: !args.no_data_types,
                dump_embeddings: !args.no_embeddings,
            };

            pool.dump_snapshot(write, settings)
                .change_context(GraphError)
                .attach("Failed to produce snapshot dump")?;

            tracing::info!("Snapshot dumped successfully");
        }
        SnapshotCommand::Restore(args) => {
            let read =
                FramedRead::new(io::BufReader::new(io::stdin()), JsonLinesDecoder::default());
            SnapshotStore::new(
                pool.acquire(None)
                    .await
                    .change_context(GraphError)
                    .map_err(|report| {
                        tracing::error!(error = ?report, "Failed to acquire database connection");
                        report
                    })?,
            )
            .restore_snapshot(read, 10_000, args.ignore_validation_errors)
            .await
            .change_context(GraphError)
            .attach("Failed to restore snapshot")?;

            tracing::info!("Snapshot restored successfully");