aboutsummaryrefslogtreecommitdiff
path: root/src/model/object_table.rs
diff options
context:
space:
mode:
authorAlex Auvolat <alex@adnab.me>2020-07-08 17:33:14 +0200
committerAlex Auvolat <alex@adnab.me>2020-07-08 17:33:14 +0200
commit84bbbfaa7b670d6dd1501aeecc6c44251819d4ae (patch)
tree480c0754feaacc98f11de554c34f3b07d379d665 /src/model/object_table.rs
parenta5fa2a136b4875fb45d990b2d94437e5d77a7ae5 (diff)
downloadgarage-84bbbfaa7b670d6dd1501aeecc6c44251819d4ae.tar.gz
garage-84bbbfaa7b670d6dd1501aeecc6c44251819d4ae.zip
Add multiple headers to object model
Diffstat (limited to 'src/model/object_table.rs')
-rw-r--r--src/model/object_table.rs31
1 files changed, 24 insertions, 7 deletions
diff --git a/src/model/object_table.rs b/src/model/object_table.rs
index 2221c57e..039e985e 100644
--- a/src/model/object_table.rs
+++ b/src/model/object_table.rs
@@ -1,6 +1,7 @@
use async_trait::async_trait;
use serde::{Deserialize, Serialize};
use std::sync::Arc;
+use std::collections::BTreeMap;
use garage_util::background::BackgroundRunner;
use garage_util::data::*;
@@ -66,7 +67,7 @@ pub struct ObjectVersion {
#[derive(PartialEq, Clone, Debug, Serialize, Deserialize)]
pub enum ObjectVersionState {
- Uploading,
+ Uploading(ObjectVersionHeaders),
Complete(ObjectVersionData),
Aborted,
}
@@ -84,12 +85,12 @@ impl ObjectVersionState {
Complete(a) => {
a.merge(b);
}
- Uploading => {
+ Uploading(_) => {
*self = Complete(b.clone());
}
}
}
- Uploading => {}
+ Uploading(_) => {}
}
}
}
@@ -103,11 +104,17 @@ pub enum ObjectVersionData {
#[derive(PartialEq, Clone, Debug, Serialize, Deserialize)]
pub struct ObjectVersionMeta {
- pub mime_type: String,
+ pub headers: ObjectVersionHeaders,
pub size: u64,
pub etag: String,
}
+#[derive(PartialEq, Clone, Debug, Serialize, Deserialize)]
+pub struct ObjectVersionHeaders {
+ pub content_type: String,
+ pub other: BTreeMap<String, String>,
+}
+
impl ObjectVersionData {
fn merge(&mut self, b: &Self) {
if *self != *b {
@@ -120,6 +127,12 @@ impl ObjectVersion {
fn cmp_key(&self) -> (u64, UUID) {
(self.timestamp, self.uuid)
}
+ pub fn is_uploading(&self) -> bool {
+ match self.state {
+ ObjectVersionState::Uploading(_) => true,
+ _ => false,
+ }
+ }
pub fn is_complete(&self) -> bool {
match self.state {
ObjectVersionState::Complete(_) => true,
@@ -232,17 +245,21 @@ impl TableSchema for ObjectTable {
}
fn migrate_version(old: &prev::ObjectVersion) -> ObjectVersion {
+ let headers = ObjectVersionHeaders{
+ content_type: old.mime_type.clone(),
+ other: BTreeMap::new(),
+ };
let meta = ObjectVersionMeta{
+ headers: headers.clone(),
size: old.size,
- mime_type: old.mime_type.clone(),
etag: "".to_string(),
};
let state = match old.state {
- prev::ObjectVersionState::Uploading => ObjectVersionState::Uploading,
+ prev::ObjectVersionState::Uploading => ObjectVersionState::Uploading(headers),
prev::ObjectVersionState::Aborted => ObjectVersionState::Aborted,
prev::ObjectVersionState::Complete => {
match &old.data {
- prev::ObjectVersionData::Uploading => ObjectVersionState::Uploading,
+ prev::ObjectVersionData::Uploading => ObjectVersionState::Uploading(headers),
prev::ObjectVersionData::DeleteMarker => ObjectVersionState::Complete(ObjectVersionData::DeleteMarker),
prev::ObjectVersionData::Inline(x) => ObjectVersionState::Complete(ObjectVersionData::Inline(meta, x.clone())),
prev::ObjectVersionData::FirstBlock(h) => {