sapling/eden/mononoke/derived_data/changeset_info/derive.rs

/*
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * This software may be used and distributed according to the terms of the
 * GNU General Public License version 2.
 */

use anyhow::Error;
use std::{collections::HashMap, iter::FromIterator, sync::Arc};

use blobrepo::BlobRepo;
use blobstore::Blobstore;
use context::CoreContext;
use derived_data::{BonsaiDerived, BonsaiDerivedMapping};
use fbthrift::compact_protocol;
use futures::{future, stream::FuturesUnordered, Future, Stream};
use futures_ext::{BoxFuture, FutureExt};
use mononoke_types::{BlobstoreBytes, BonsaiChangeset, ChangesetId};

use crate::ChangesetInfo;

impl BonsaiDerived for ChangesetInfo {
    const NAME: &'static str = "changeset_info";
    type Mapping = ChangesetInfoMapping;

    fn mapping(_ctx: &CoreContext, repo: &BlobRepo) -> Self::Mapping {
        ChangesetInfoMapping::new(repo.blobstore().boxed())
    }

    fn derive_from_parents(
        _ctx: CoreContext,
        _repo: BlobRepo,
        bonsai: BonsaiChangeset,
        _parents: Vec<Self>,
    ) -> BoxFuture<Self, Error> {
        let csid = bonsai.get_changeset_id();
        future::ok(ChangesetInfo::new(csid, bonsai)).boxify()
    }
}

#[derive(Clone)]
pub struct ChangesetInfoMapping {
    blobstore: Arc<dyn Blobstore>,
}

impl ChangesetInfoMapping {
    pub fn new(blobstore: Arc<dyn Blobstore>) -> Self {
        Self { blobstore }
    }

    fn format_key(&self, csid: &ChangesetId) -> String {
        format!("changeset_info.blake2.{}", csid)
    }
}

impl BonsaiDerivedMapping for ChangesetInfoMapping {
    type Value = ChangesetInfo;

    fn get(
        &self,
        ctx: CoreContext,
        csids: Vec<ChangesetId>,
    ) -> BoxFuture<HashMap<ChangesetId, Self::Value>, Error> {
        let futs = csids.into_iter().map(|csid| {
            self.blobstore
                .get(ctx.clone(), self.format_key(&csid))
                .map(move |value| {
                    value.map(|bytes| {
                        let info = ChangesetInfo::from_bytes(bytes.as_raw_bytes())?;
                        Ok((csid, info))
                    })
                })
        });
        FuturesUnordered::from_iter(futs)
            .filter_map(|maybe_info| maybe_info)
            .collect()
            .and_then(move |infos| infos.into_iter().collect::<Result<HashMap<_, _>, Error>>())
            .boxify()
    }

    fn put(&self, ctx: CoreContext, csid: ChangesetId, info: Self::Value) -> BoxFuture<(), Error> {
        let data = {
            let data = compact_protocol::serialize(&info.into_thrift());
            BlobstoreBytes::from_bytes(data)
        };
        self.blobstore.put(ctx, self.format_key(&csid), data)
    }
}

#[cfg(test)]
mod test {
    use super::*;

    use blobstore::Loadable;
    use fbinit::FacebookInit;
    use fixtures::linear;
    use mercurial_types::HgChangesetId;
    use mononoke_types::BonsaiChangeset;
    use std::collections::BTreeMap;
    use std::str::FromStr;
    use tokio_compat::runtime::Runtime;

    #[fbinit::test]
    fn derive_info_test(fb: FacebookInit) {
        let mut runtime = Runtime::new().unwrap();
        let repo = runtime.block_on_std(linear::getrepo(fb));
        let ctx = CoreContext::test_mock(fb);

        let hg_cs_id = HgChangesetId::from_str("3c15267ebf11807f3d772eb891272b911ec68759").unwrap();
        let bcs_id = runtime
            .block_on(repo.get_bonsai_from_hg(ctx.clone(), hg_cs_id))
            .unwrap()
            .unwrap();
        let bcs = runtime
            .block_on(bcs_id.load(ctx.clone(), repo.blobstore()))
            .unwrap();

        // Make sure that the changeset info was saved in the blobstore
        let info = runtime
            .block_on(ChangesetInfo::derive(
                ctx.clone(),
                repo.clone(),
                bcs_id.clone(),
            ))
            .unwrap();

        check_info(&info, &bcs);
    }

    fn check_info(info: &ChangesetInfo, bcs: &BonsaiChangeset) {
        assert_eq!(*info.changeset_id(), bcs.get_changeset_id());
        assert_eq!(info.message(), bcs.message());
        assert_eq!(
            info.parents().collect::<Vec<_>>(),
            bcs.parents().collect::<Vec<_>>()
        );
        assert_eq!(info.author(), bcs.author());
        assert_eq!(info.author_date(), bcs.author_date());
        assert_eq!(info.committer(), bcs.committer());
        assert_eq!(info.committer_date(), bcs.committer_date());
        assert_eq!(
            info.extra().collect::<BTreeMap<_, _>>(),
            bcs.extra().collect::<BTreeMap<_, _>>()
        );
    }
}