holochain_data 0.7.0-rc.0

//! `DbRead<Dht>` / `DbWrite<Dht>` API for the `Entry` and `PrivateEntry` tables.

use super::super::inner::entry;
use crate::handles::{DbRead, DbWrite};
use crate::kind::Dht;
use holo_hash::{AgentPubKey, EntryHash};
use holochain_integrity_types::entry::Entry;
use std::collections::HashMap;

impl DbWrite<Dht> {
    pub async fn insert_entry(&self, hash: &EntryHash, entry: &Entry) -> sqlx::Result<()> {
        entry::insert_entry(self.pool(), hash, entry).await
    }

    pub async fn insert_private_entry(
        &self,
        hash: &EntryHash,
        author: &AgentPubKey,
        entry: &Entry,
    ) -> sqlx::Result<()> {
        entry::insert_private_entry(self.pool(), hash, author, entry).await
    }
}

impl DbRead<Dht> {
    /// Reads an entry by hash. Pass `author = Some(_)` to also surface a
    /// matching `PrivateEntry` owned by that author.
    pub async fn get_entry(
        &self,
        hash: EntryHash,
        author: Option<&AgentPubKey>,
    ) -> sqlx::Result<Option<Entry>> {
        let mut conn = self.timed_conn().await?;
        entry::get_entry(&mut *conn, hash, author).await
    }

    /// Batch-reads entries by hash in a single query per chunk, keyed by entry
    /// hash. Visibility matches [`get_entry`](Self::get_entry): public entries
    /// always resolve, and `author = Some(_)` additionally surfaces that
    /// author's private entries. Hashes with no entry are absent from the map.
    pub async fn get_entries_by_hashes(
        &self,
        hashes: &[EntryHash],
        author: Option<&AgentPubKey>,
    ) -> sqlx::Result<HashMap<EntryHash, Entry>> {
        let mut conn = self.timed_conn().await?;
        entry::get_entries_by_hashes(&mut conn, hashes, author).await
    }
}