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
|
mod builtin_driver;
mod pipeline;
mod platform;
mod util {
use std::collections::HashMap;
use gix_hash::oid;
use gix_object::{bstr::BString, find::Error};
#[derive(Default)]
pub struct ObjectDb {
data_by_id: HashMap<gix_hash::ObjectId, BString>,
}
impl gix_object::FindHeader for ObjectDb {
fn try_header(&self, id: &oid) -> Result<Option<gix_object::Header>, Error> {
match self.data_by_id.get(&id.to_owned()) {
Some(data) => Ok(Some(gix_object::Header {
kind: gix_object::Kind::Blob,
size: data.len() as u64,
})),
None => Ok(None),
}
}
}
impl gix_object::Find for ObjectDb {
fn try_find<'a>(&self, id: &oid, buffer: &'a mut Vec<u8>) -> Result<Option<gix_object::Data<'a>>, Error> {
match self.data_by_id.get(&id.to_owned()) {
Some(data) => {
buffer.clear();
buffer.extend_from_slice(data);
Ok(Some(gix_object::Data {
kind: gix_object::Kind::Blob,
data: buffer.as_slice(),
}))
}
None => Ok(None),
}
}
}
impl ObjectDb {
/// Insert `data` and return its hash. That can be used to find it again.
pub fn insert(&mut self, data: &str) -> gix_hash::ObjectId {
let id = gix_object::compute_hash(gix_hash::Kind::Sha1, gix_object::Kind::Blob, data.as_bytes());
self.data_by_id.insert(id, data.into());
id
}
}
}
|
