hyperfile 0.10.4

use std::io::{Result, Error, ErrorKind};
use std::sync::Arc;
use log::{trace, debug, warn, error};
use aws_sdk_s3::Client;
use aws_sdk_s3::types::Object;
use crate::staging::{Staging, FlushInodeFlag};
use crate::file::ReadTiming;
use crate::staging::config::StagingConfig;
use crate::{BlockIndex, BlockPtr, SegmentId};
use crate::segment;
use crate::segment::{Segment, SegmentSum};
use crate::config::{HyperFileMetaConfig, HyperFileRuntimeConfig};
use crate::ondisk::{SegmentHeader, InodeRaw};
use crate::s3uri::S3Uri;
use crate::meta_loader::s3::S3BlockLoader;
use crate::inode::OnDiskState;
use crate::s3commons::{S3Ops, s3_error_kind};
use btree_ondisk::node::{BtreeNode, BTREE_NODE_LEVEL_DATA};
use super::StagingIntercept;

const SEGMENT_HEADER_FETCH_SIZE: usize = 512 * 1024;

/// `uri` as a bucket and a key, or `InvalidInput` naming it: retrying will not
/// make a URI whole.
fn parse_uri(uri: &str) -> Result<S3Uri<'_>> {
    S3Uri::parse(uri).map_err(|e| Error::new(ErrorKind::InvalidInput,
        format!("{:?} is not an S3 URI naming a bucket and a path: {}", uri, e)))
}

#[derive(Clone)]
pub struct S3Staging {
    pub client: Client,
    pub bucket: String,
    pub root_path: String,
    pub root_path_slash: String,  // root path with tail slash
    pub inode_file: String, // full path of inode file
    pub config: StagingConfig,
    pub runtime_config: HyperFileRuntimeConfig,
    pub interceptor: Option<Arc<dyn StagingIntercept<Self>>>,
    /// Shared with every clone of this handle and with the block
    /// loader, so the counts cover the whole file rather than one
    /// handle.
    pub read_timing: Arc<ReadTiming>,
}

impl S3Staging {
    /// The object names a checkpoint's summary could carry, in the order to try.
    ///
    /// A streamed checkpoint keeps its summary in part 0; one written as a single
    /// object has no part suffix at all. Which of the two a container uses is
    /// recorded in the inode — so a reader looking *for* the inode cannot know
    /// first, and has to try both.
    ///
    /// Unsuffixed comes first so that a container written before parting existed
    /// costs exactly what it always did. A streamed one pays one extra request,
    /// and only on the two paths that read a checkpoint's header rather than its
    /// data: opening a specific checkpoint, and asking a checkpoint's timestamp.
    fn candidate_summary_objs(&self, segid: SegmentId) -> [SegmentId; 2] {
        [segid.checkpoint(), segid.at_part(crate::segment::Segment::SUMMARY_PART)]
    }

    /// Read exactly the object `segid` names, with no fallback.
    ///
    /// `SegmentReadWrite::open` resolves a bare checkpoint number to the summary
    /// part when there is no object of that name. Use this to ask about one object,
    /// which is what walking a checkpoint's parts needs.
    pub async fn open_exact(&self, segid: SegmentId) -> Result<SegmentSum> {
        let filename = format!("{}/{}", self.root_path, Segment::segid_to_staging_file_id(segid));
        Self::do_open(&self.client, &self.bucket, &filename).await
    }

    /// Which of the two names actually exists, or the best guess when neither
    /// answers — the caller reports the failure that follows.
    async fn probe_summary_obj(&self, segid: SegmentId) -> SegmentId {
        let candidates = self.candidate_summary_objs(segid);
        for segid in candidates {
            let key = format!("{}/{}", self.root_path, Segment::segid_to_staging_file_id(segid));
            if self.client.head_object().bucket(&self.bucket).key(&key).send().await.is_ok() {
                return segid;
            }
        }
        candidates[0]
    }
}

impl Staging<S3BlockLoader> for S3Staging {
    fn to_block_loader(&self) -> S3BlockLoader {
        S3BlockLoader::new(&self.client, &self.bucket, &self.root_path, self.read_timing.clone())
    }

    fn read_timing(&self) -> &ReadTiming {
        &self.read_timing
    }

    async fn load_inode(&self, buf: &mut [u8]) -> Result<Option<OnDiskState>> {
        let start = std::time::Instant::now();
        let res = S3Ops::do_get_object(&self.client, &self.bucket, &self.inode_file, buf, None, true).await;
        self.read_timing.add_inode_get(start.elapsed().as_nanos() as u64);
        res
    }

    async fn load_inode_from_segment(&self, buf: &mut [u8], segid: SegmentId) -> Result<Option<OnDiskState>> {
        if segid.as_cno() == 0 {
            // read from latest inode
            return self.load_inode(buf).await;
        }

        let inode_off = std::mem::offset_of!(SegmentHeader, s_inode);
        let inode_bytes = std::mem::size_of::<InodeRaw>();
        let range = format!("bytes={}-{}", inode_off, inode_off + inode_bytes - 1);
        let start = std::time::Instant::now();
        // Which name holds the inode depends on whether the container streams its
        // checkpoints, and that is recorded *in the inode being read* — so it
        // cannot be known first. Both names are tried. The caller's configured
        // format only decides which to try first, which is why a stale one costs
        // a request and never a wrong answer.
        let mut res = Err(Error::new(ErrorKind::NotFound, "no name tried"));
        for segid in self.candidate_summary_objs(segid) {
            let key = format!("{}/{}", self.root_path, Segment::segid_to_staging_file_id(segid));
            res = S3Ops::do_get_object(&self.client, &self.bucket, &key, buf, Some(&range), false).await;
            match &res {
                Err(e) if e.kind() == ErrorKind::NotFound => continue,
                _ => break,
            }
        }
        self.read_timing.add_inode_get(start.elapsed().as_nanos() as u64);
        res
    }

    async fn segment_exists(&self, segid: SegmentId) -> Result<bool> {
        let key = format!("{}/{}", self.root_path, Segment::segid_to_staging_file_id(segid));
        Self::try_exists(&self.client, &self.bucket, &key).await
    }

    async fn load_segment_timestamp(&self, segid: SegmentId) -> Result<(i64, i64)> {
        // get date from server response with interceptor
        use std::fmt;
        use std::sync::Arc;
        use aws_sdk_s3::config::ConfigBag;
        use aws_sdk_s3::config::interceptors::FinalizerInterceptorContextRef;
        use aws_sdk_s3::config::Intercept;
        use aws_sdk_s3::config::RuntimeComponents;
        use tokio::sync::Mutex;
        use chrono::DateTime;

        struct ResponseDateInterceptor {
            inner: Arc<Mutex<Option<i64>>>,
        }

        impl fmt::Debug for ResponseDateInterceptor {
            fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
                write!(f, "")
            }
        }

        impl Intercept for ResponseDateInterceptor {
            fn name(&self) -> &'static str { "date" }

            fn read_after_execution(&self, ctx: &FinalizerInterceptorContextRef<'_>, _: &RuntimeComponents, _: &mut ConfigBag,)
                -> std::result::Result<(), Box<dyn std::error::Error + Send + Sync>>
            {
                if let Some(resp) = ctx.response() {
                    // Left unset when missing or not RFC 2822 — a proxy, an
                    // S3-compatible store — and the caller refuses it then.
                    if let Some(Ok(dt)) = resp.headers().get("date").map(DateTime::parse_from_rfc2822) {
                        if let Ok(mut lock) = self.inner.try_lock() {
                            *lock = Some(dt.timestamp());
                        }
                    }
                }
                Ok(())
            }
        }

        let server_datetime = Arc::new(Mutex::new(None));
        // See `load_inode_from_segment` for why both names are tried.
        let key = format!("{}/{}", self.root_path,
            Segment::segid_to_staging_file_id(self.probe_summary_obj(segid).await));

        match self.client
            .head_object()
            .bucket(&self.bucket)
            .key(&key)
            .customize()
            .interceptor(ResponseDateInterceptor { inner: server_datetime.clone() })
            .send()
            .await
        {
            Ok(output) => {
                // Both are needed to tell whether a checkpoint is still being
                // published. Guessing is not an option: a time of 0 made every
                // checkpoint look fresh, and a writable open waited for it for good.
                let Some(server_dt) = *server_datetime.lock().await else {
                    let err_str = format!("HeadObject s3://{}/{}: the response has no Date header \
                        in RFC 2822 form, so the server's time is unknown", self.bucket, key);
                    error!("{}", err_str);
                    return Err(Error::new(ErrorKind::InvalidData, err_str));
                };
                let Some(lm) = output.last_modified.map(|t| t.secs()) else {
                    let err_str = format!("HeadObject s3://{}/{}: the response has no Last-Modified",
                        self.bucket, key);
                    error!("{}", err_str);
                    return Err(Error::new(ErrorKind::InvalidData, err_str));
                };
                return Ok((server_dt, lm));
            },
            Err(sdk_err) => {
                if sdk_err.as_service_error().map(|e| e.is_not_found()) == Some(true) {
                    let err_str = format!("HeadObject s3://{}/{} not exist", self.bucket, key);
                    warn!("{}", err_str);
                    return Err(Error::new(ErrorKind::NotFound, err_str));
                }
                let err_str = format!("HeadObject s3://{}/{} error: {}", self.bucket, key, sdk_err);
                error!("{}", err_str);
                return Err(Error::new(s3_error_kind(&sdk_err), err_str));
            },
        }
    }

    async fn flush_inode(&self, buf: &[u8], inode_state: &Option<OnDiskState>, flag: FlushInodeFlag) -> Result<Option<OnDiskState>> {
        if let Some(i) = &self.interceptor {
            let _ = i.before_flush_inode(&self, buf, flag.clone()).await?;
        }
        let inode_state = S3Ops::do_put_object(&self.client, &self.bucket, &self.inode_file, buf, inode_state).await?;
        if let Some(i) = &self.interceptor {
            let _ = i.after_flush_inode(&self, buf, flag).await;
        }
        Ok(inode_state)
    }

    async fn remove_inode(&self, inode_state: &Option<OnDiskState>) -> Result<()> {
        if self.is_directory_bucket() {
            let _  = S3Ops::do_delete_object(&self.client, &self.bucket, &self.inode_file, inode_state).await?;
        } else {
            let _  = S3Ops::do_delete_object(&self.client, &self.bucket, &self.inode_file, &None).await?;
        }
        if let Some(i) = &self.interceptor {
            let _ = i.after_remove_inode(&self).await;
        }
        Ok(())
    }

    // load a single data block from staging
    async fn load_data_block(&self, segid: SegmentId, staging_off: usize, offset: usize, block_size: usize, buf: &mut [u8]) -> Result<()> {
        // open staging file for read
        let start_off = staging_off + offset;
        let end = start_off + block_size - 1;
        let key = format!("{}/{}", self.root_path, Segment::segid_to_staging_file_id(segid));
        let range = format!("bytes={}-{}", start_off, end);
        debug!("load_data_block from s3 staging s3://{}/{} at offset: {} range: {} block size: {}", &self.bucket, &key, start_off, &range, block_size);
        let start = std::time::Instant::now();
        let res = S3Ops::do_get_object(&self.client, &self.bucket, &key, buf, Some(&range), false).await;
        self.read_timing.add_data_get(block_size, start.elapsed().as_nanos() as u64);
        let _ = res?;
        Ok(())
    }

    async fn load_range(&self, segid: SegmentId, s3_off: usize, buf: &mut [u8]) -> Result<()> {
        let len = buf.len();
        debug_assert!(len > 0, "load_range with zero length");
        let end = s3_off + len - 1;
        let key = format!(
            "{}/{}",
            self.root_path,
            Segment::segid_to_staging_file_id(segid),
        );
        let range = format!("bytes={}-{}", s3_off, end);
        debug!(
            "load_range from s3 staging s3://{}/{} at offset: {} range: {} len: {}",
            &self.bucket, &key, s3_off, &range, len,
        );
        let start = std::time::Instant::now();
        let res = S3Ops::do_get_object(&self.client, &self.bucket, &key, buf, Some(&range), false).await;
        self.read_timing.add_data_get(len, start.elapsed().as_nanos() as u64);
        let _ = res?;
        Ok(())
    }

    fn new_segwr(&self, segid: SegmentId, hyper_file_config: &HyperFileMetaConfig) -> segment::Writer<S3Staging> {
        segment::Writer::<S3Staging>::new(self.clone(), self.runtime_config.segment_buffer_size, segid, hyper_file_config)
    }

    fn new_segwr_at(&self, segid: SegmentId, hyper_file_config: &HyperFileMetaConfig) -> segment::Writer<S3Staging> {
        segment::Writer::<S3Staging>::new_at(self.clone(), self.runtime_config.segment_buffer_size, segid, hyper_file_config)
    }

    fn dir_filename(&self) -> (&str, &str) {
        if let Some((dir, filename)) = self.root_path.rsplit_once('/') {
            return (dir, filename);
        }
        return ("", &self.root_path);
    }

    fn root_path(&self) -> &str {
        &self.root_path
    }

    fn config(&self) -> &StagingConfig {
        &self.config
    }

    async fn unlink(&self) -> Result<()> {
        self.remove_inode(&None).await?;

        let mut delete_keys = Vec::new();
        let filter = |o: &Object| delete_keys.push(o.key().unwrap().to_string());
        let _ = S3Ops::do_list_objects(&self.client, &self.bucket, &self.root_path_slash, filter).await?;
        debug!("unlink - found delete keys {:?}", delete_keys);

        let _ = S3Ops::do_delete_objects(&self.client, &self.bucket, delete_keys).await?;
        Ok(())
    }

    fn interceptor(&mut self, i: impl StagingIntercept<Self> + 'static) {
        self.interceptor = Some(Arc::new(i));
    }
}

impl S3Staging {
    async fn try_exists(client: &Client, bucket: &str, key: &str) -> Result<bool> {
        match S3Ops::do_head_object(client, bucket, key).await {
            Ok(_) => {
                return Ok(true);
            },
            Err(e) => {
                if e.kind() == ErrorKind::NotFound {
                    return Ok(false);
                }
                return Err(e);
            },
        }
    }

    pub fn pesudo(client: &Client, config: StagingConfig, runtime_config: HyperFileRuntimeConfig) -> Self {
        Self {
            client: client.to_owned(),
            bucket: String::new(),
            root_path: String::new(),
            root_path_slash: String::new(),
            inode_file: String::new(),
            config: config,
            runtime_config: runtime_config,
            interceptor: None,
            read_timing: Arc::new(ReadTiming::default()),
        }
    }

    pub async fn from(client: &Client, config: StagingConfig, runtime_config: HyperFileRuntimeConfig) -> Result<Self> {
        // Both parsed before anything is sent, so a bad one costs no request.
        let s3uri = parse_uri(&config.root_uri)?;
        let inode_uri = parse_uri(&config.inode_file_uri)?;
        let inode_file_path = inode_uri.key.to_string();
        let inode_exists = Self::try_exists(client, inode_uri.bucket, &inode_file_path).await?;
        if !inode_exists {
            return Err(Error::new(ErrorKind::NotFound, format!("inode not exists: {}", config.inode_file_uri)));
        }

        let root_path = s3uri.key.to_string();
        let mut root_path_slash = root_path.clone();
        root_path_slash.push('/');
        Ok(Self {
            client: client.to_owned(),
            bucket: s3uri.bucket.to_string(),
            root_path: root_path,
            root_path_slash: root_path_slash,
            inode_file: inode_file_path,
            config: config,
            runtime_config: runtime_config,
            interceptor: None,
            read_timing: Arc::new(ReadTiming::default()),
        })
    }

    pub async fn create(client: &Client, config: StagingConfig, runtime_config: HyperFileRuntimeConfig) -> Result<Self> {
        // Both parsed before anything is sent, so a bad one costs no request.
        let s3uri = parse_uri(&config.root_uri)?;
        let inode_uri = parse_uri(&config.inode_file_uri)?;
        let inode_file_path = inode_uri.key.to_string();
        let inode_exists = Self::try_exists(client, inode_uri.bucket, &inode_file_path).await?;
        if inode_exists {
            return Err(Error::new(ErrorKind::AlreadyExists, "inode exists"));
        }

        let root_path = s3uri.key.to_string();
        let mut root_path_slash = root_path.clone();
        root_path_slash.push('/');
        Ok(Self {
            client: client.to_owned(),
            bucket: s3uri.bucket.to_string(),
            root_path: root_path,
            root_path_slash: root_path_slash,
            inode_file: inode_file_path,
            config: config,
            runtime_config: runtime_config,
            interceptor: None,
            read_timing: Arc::new(ReadTiming::default()),
        })
    }

    #[inline]
    fn is_directory_bucket(&self) -> bool {
        self.bucket.ends_with("--x-s3")
    }
}

impl segment::SegmentReadWrite for S3Staging {
    fn append(&self, segid: SegmentId, buf: &[u8]) -> Result<()> {
        trace!("appending to inner buffer {}/{} len {}", self.root_path, Segment::segid_to_staging_file_id(segid), buf.len());
        Ok(())
    }

    async fn done(&self, segid: SegmentId, buf: &[u8], len: usize) -> Result<()> {
        if let Some(i) = &self.interceptor {
            let _ = i.before_segment_done(&self, segid, buf, len).await?;
        }
        let key = format!("{}/{}", self.root_path, Segment::segid_to_staging_file_id(segid));
        debug!("bufwr done s3://{}/{} {}", &self.bucket, &key, len);
        let (data, _) = buf.split_at(len);

        // check if should use MPU
        if self.segment_should_mp_upload(len) {
            let _ = S3Ops::do_mp_upload(&self.client, &self.bucket, &key, data, &None,
                self.runtime_config.segment_mpu_chunk_size, self.runtime_config.segment_mpu_max_concurrency).await?;
            return Ok(());
        }

        let _ = S3Ops::do_put_object(&self.client, &self.bucket, &key, buf, &None).await?;
        Ok(())
    }

    async fn done_pieces(&self, segid: SegmentId, body: crate::segment_body::SegmentBody) -> Result<()> {
        // The fault-injection interceptors used in
        // `tests/integration_s3_*.rs` take a `&[u8]` snapshot of
        // the body. To preserve their ergonomics we build a
        // contiguous copy here ONLY when an interceptor is
        // installed (i.e. during tests). The production fast path
        // skips this and goes straight to the scatter upload.
        if let Some(i) = &self.interceptor {
            let mut snapshot = Vec::with_capacity(body.len());
            for piece in body.clone().materialize().await?.byte_pieces() {
                snapshot.extend_from_slice(piece);
            }
            let _ = i.before_segment_done(&self, segid, &snapshot, snapshot.len()).await?;
        }
        let key = format!("{}/{}", self.root_path, Segment::segid_to_staging_file_id(segid));
        let len = body.len();
        debug!("bufwr done s3://{}/{} {} (scatter)", &self.bucket, &key, len);

        if self.segment_should_mp_upload(len) {
            let _ = S3Ops::do_mp_upload_pieces(
                &self.client,
                &self.bucket,
                &key,
                body,
                &None,
                self.runtime_config.segment_mpu_chunk_size,
                self.runtime_config.segment_mpu_max_concurrency,
            ).await?;
            return Ok(());
        }

        let _ = S3Ops::do_put_object_pieces(
            &self.client,
            &self.bucket,
            &key,
            body,
            &None,
        ).await?;
        Ok(())
    }

    async fn remove(&self, segid: SegmentId) -> Result<()> {
        let key = format!("{}/{}", self.root_path, Segment::segid_to_staging_file_id(segid));
        Self::do_remove(&self.client, &self.bucket, &key).await
    }

    async fn open(&self, segid: SegmentId) -> Result<SegmentSum> {
        // A bare checkpoint number may name an object that does not exist, because
        // a container that streams its checkpoints keeps the summary in part 0. Both
        // are tried for the same reason `load_inode_from_segment` tries both.
        let mut last = Err(Error::new(ErrorKind::NotFound, "no name tried"));
        for obj in [segid, segid.at_part(Segment::SUMMARY_PART)] {
            last = self.open_exact(obj).await;
            match &last {
                Err(e) if e.kind() == ErrorKind::NotFound && segid.part_id().is_none() => continue,
                _ => break,
            }
        }
        last
    }


    async fn list(&self, segid: SegmentId) -> Result<Vec<SegmentId>> {
        Self::do_list(&self.client, &self.bucket, &self.root_path_slash, segid).await
    }

    async fn build_block_map(&self, segid: SegmentId) -> Result<Vec<(BlockIndex, BlockPtr)>> {
        let filename = format!("{}/{}", self.root_path, Segment::segid_to_staging_file_id(segid));
        Self::do_build_block_map(&self.client, &self.bucket, &filename).await
    }
}

impl S3Staging {
    pub(crate) async fn do_remove(client: &Client, bucket: &str, key: &str) -> Result<()> {
        S3Ops::do_delete_object(client, bucket, key, &None).await
    }

    pub(crate) async fn do_open(client: &Client, bucket: &str, key: &str) -> Result<SegmentSum> {
        // read at least min size of header data
        let range = format!("bytes={}-{}", 0, SEGMENT_HEADER_FETCH_SIZE - 1);
        let (bytes, _) = S3Ops::do_get_object_speculative(client, bucket, key, Some(&range), false).await?;

        let mut buf = bytes.to_vec();

        let hdr_size = std::mem::size_of::<SegmentHeader>();
        if buf.len() < hdr_size {
            // if segment size is too small
            return Err(Error::new(ErrorKind::InvalidData, "incorrect segment header size"));
        }

        // check actual segment sum bytes
        let hdr_buf = &buf[0..hdr_size];
        let hdr = SegmentHeader::from_slice(&hdr_buf);
        let actual_ss_bytes = hdr.s_bytes as usize;
        if buf.len() < actual_ss_bytes {
            // Read the rest of the summary. This uses the speculative
            // read on purpose: sizing a buffer from `s_bytes` would let a
            // corrupted header (u32, so up to ~4 GiB) dictate the
            // allocation. Letting S3 clamp the range to the end of the
            // object instead bounds the read by data that actually
            // exists.
            let range = format!("bytes={}-{}", buf.len(), actual_ss_bytes - 1);
            let (rest, _) = S3Ops::do_get_object_speculative(client, bucket, key, Some(&range), false).await?;
            buf.extend_from_slice(&rest);

            // Still short: `s_bytes` claims more than the object holds, so
            // the header is corrupt or the object was truncated.
            if buf.len() < actual_ss_bytes {
                let err_str = format!(
                    "segment summary truncated on s3://{}/{}: header claims {} bytes, object has {}",
                    bucket, key, actual_ss_bytes, buf.len(),
                );
                error!("{}", err_str);
                return Err(Error::new(ErrorKind::InvalidData, err_str));
            }
        }

        Ok(SegmentSum::from_slice(&buf))
    }

    // list staging dir for all segments id <= input segid, skip anythig else
    // if input segment id is 0, return all
    /// The checkpoints at or below `segid`, ascending, each counted once.
    ///
    /// Checkpoints, not objects: one written as several parts appears once, because
    /// what a caller walking a container's history wants is its history and not its
    /// storage layout.
    ///
    /// A checkpoint's objects are named by a padded number with an optional part
    /// suffix, so the number is what comes before the first `.`. Parsing the whole
    /// name as a number — which is what this did — skips every object of a container
    /// that streams its checkpoints, and skipping them all is indistinguishable from
    /// a container with no history.
    pub(crate) async fn do_list(client: &Client, bucket: &str, prefix: &str, segid: SegmentId) -> Result<Vec<SegmentId>> {
        let mut output = Vec::new();
        let filter = |o: &Object| {
            let key = o.key().unwrap();
            let filename = if let Some((_, f)) = key.rsplit_once('/') {
                f
            } else {
                key
            };
            // The checkpoint number is the name up to the part suffix, if any. The
            // inode object and anything else under the prefix parse as neither.
            let number = filename.split('.').next().unwrap_or(filename);
            if let Ok(id) = number.parse::<u64>() {
                if segid.as_cno() == 0 || id <= segid.as_cno() {
                    output.push(id)
                }
            }
        };
        let _ = S3Ops::do_list_objects(client, bucket, prefix, filter).await?;
        output.sort();
        output.dedup();
        Ok(output.into_iter().map(SegmentId::new_from_cno).collect())
    }

    pub async fn cli_list(client: &Client, bucket: &str, prefix: &str, segid: SegmentId) -> Result<Vec<SegmentId>> {
        Self::do_list(client, bucket, prefix, segid).await
    }

    // fetch meta blocks area from segment
    // return: (meta_block_area_offset, meta_block_size, meta_block count, data chunk)
    /// Fetch a segment's meta-block chunk.
    ///
    /// Issues up to two requests — the header, then the blocks — so it
    /// takes the counter rather than letting the caller record one
    /// request for what may be two.
    pub(crate) async fn do_fetch_meta_blocks_chunk(client: &Client, bucket: &str, key: &str,
        read_timing: Option<&ReadTiming>) -> Result<(usize, usize, usize, Vec<u8>)> {
        let mut buf = Vec::with_capacity(SegmentHeader::size());
        buf.resize(SegmentHeader::size(), 0);
        // read in header
        let range = format!("bytes={}-{}", 0, SegmentHeader::size() - 1);
        let start = std::time::Instant::now();
        let res = S3Ops::do_get_object(client, bucket, key, &mut buf, Some(&range), false).await;
        if let Some(t) = read_timing {
            t.add_meta_get(SegmentHeader::size(), start.elapsed().as_nanos() as u64);
        }
        let _ = res?;

        let hdr = SegmentHeader::from_slice(&buf);
        let meta_block_size = (1 << hdr.s_meta_blk_shift) as usize;
        let meta_blocks = hdr.s_nmetablk as usize;
        let meta_block_off = hdr.aligned_ss_bytes();
        if meta_blocks == 0 {
            return Ok((meta_block_off, meta_block_size, meta_blocks, Vec::new()));
        }

        let meta_blocks_len = (meta_block_size * meta_blocks) as usize;

        let mut buf = Vec::with_capacity(meta_blocks_len);
        buf.resize(meta_blocks_len, 0);
        // readin meta blocks
        let range = format!("bytes={}-{}", meta_block_off, meta_block_off + meta_blocks_len - 1);
        let start = std::time::Instant::now();
        let res = S3Ops::do_get_object(client, bucket, key, &mut buf, Some(&range), false).await;
        if let Some(t) = read_timing {
            t.add_meta_get(meta_blocks_len, start.elapsed().as_nanos() as u64);
        }
        let _ = res?;

        Ok((meta_block_off, meta_block_size, meta_blocks, buf))
    }

    pub(crate) async fn do_build_block_map(client: &Client, bucket: &str, key: &str) -> Result<Vec<(BlockIndex, BlockPtr)>> {

        let (_meta_block_off, meta_block_size, _meta_blocks, buf) = Self::do_fetch_meta_blocks_chunk(client, bucket, key, None).await?;

        // collect all valid entry from all level 0 node
        let mut v = Vec::new();
        for meta_block_slice in buf.chunks(meta_block_size) {
            let node = BtreeNode::<BlockIndex, BlockPtr, BlockPtr>::from_slice_ref(meta_block_slice)?;
            if node.get_level() == BTREE_NODE_LEVEL_DATA + 1 {
                for idx in 0..node.get_nchild() {
                    v.push((*node.get_key(idx), *node.get_val(idx)));
                }
            }
        }
        return Ok(v);
    }

    pub async fn cli_build_block_map(client: &Client, bucket: &str, key: &str) -> Result<Vec<(BlockIndex, BlockPtr)>> {
        Self::do_build_block_map(client, bucket, key).await
    }

    fn segment_should_mp_upload(&self, len: usize) -> bool {
        if len > self.runtime_config.segment_mpu_chunk_size {
            return true;
        }
        false
    }
}