feat(objectarium): include bucket statistics in bucket information query

ccamel · ccamel · commit 87280490216f · 2024-06-11T16:59:05.000+02:00
diff --git a/contracts/axone-objectarium/src/contract.rs b/contracts/axone-objectarium/src/contract.rs
@@ -315,6 +315,7 @@ pub mod query {
             config: bucket.config.into(),
             limits: bucket.limits.into(),
             pagination: bucket.pagination.into(),
+            stat: bucket.stat.into(),
         })
     }
 
diff --git a/contracts/axone-objectarium/src/msg.rs b/contracts/axone-objectarium/src/msg.rs
@@ -147,6 +147,8 @@ pub struct BucketResponse {
     pub limits: BucketLimits,
     /// The configuration for paginated query.
     pub pagination: PaginationConfig,
+    /// The statistics of the bucket.
+    pub stat: BucketStat,
 }
 
 /// CompressionAlgorithm is an enumeration that defines the different compression algorithms
@@ -351,6 +353,20 @@ impl Default for PaginationConfig {
     }
 }
 
+/// # BucketStat
+///
+/// BucketStat is the type of the statistics of a bucket.
+#[cw_serde]
+#[derive(Default, Builder)]
+pub struct BucketStat {
+    /// The total size of the objects contained in the bucket.
+    pub size: Uint128,
+    /// The total size of the objects contained in the bucket after compression.
+    pub compressed_size: Uint128,
+    /// The number of objects in the bucket.
+    pub object_count: Uint128,
+}
+
 /// # ObjectResponse
 /// ObjectResponse is the response of the Object query.
 #[cw_serde]
diff --git a/contracts/axone-objectarium/src/state.rs b/contracts/axone-objectarium/src/state.rs
@@ -211,6 +211,15 @@ impl From<BucketLimits> for msg::BucketLimits {
     }
 }
 
+impl From<BucketStat> for msg::BucketStat {
+    fn from(stat: BucketStat) -> Self {
+        msg::BucketStat {
+            size: stat.size,
+            compressed_size: stat.compressed_size,
+            object_count: stat.object_count,
+        }
+    }
+}
 impl BucketLimits {
     fn try_new(
         max_total_size: Option<Uint128>,
@@ -262,7 +271,6 @@ impl TryFrom<msg::BucketLimits> for BucketLimits {
         )
     }
 }
-
 /// Pagination is the type carrying configuration for paginated queries.
 #[derive(Serialize, Deserialize, Clone, Debug, PartialEq, Eq, JsonSchema)]
 pub struct Pagination {

Original file line number	Diff line number	Diff line change
`@@ -315,6 +315,7 @@ pub mod query {`
`315`	`315`	`config: bucket.config.into(),`
`316`	`316`	`limits: bucket.limits.into(),`
`317`	`317`	`pagination: bucket.pagination.into(),`
	`318`	`+ stat: bucket.stat.into(),`
`318`	`319`	`})`
`319`	`320`	`}`
`320`	`321`
Original file line number	Diff line number	Diff line change
`@@ -211,6 +211,15 @@ impl From<BucketLimits> for msg::BucketLimits {`
`211`	`211`	`}`
`212`	`212`	`}`
`213`	`213`
	`214`	`+impl From<BucketStat> for msg::BucketStat {`
	`215`	`+ fn from(stat: BucketStat) -> Self {`
	`216`	`+ msg::BucketStat {`
	`217`	`+ size: stat.size,`
	`218`	`+ compressed_size: stat.compressed_size,`
	`219`	`+ object_count: stat.object_count,`
	`220`	`+ }`
	`221`	`+ }`
	`222`	`+}`
`214`	`223`	`impl BucketLimits {`
`215`	`224`	`fn try_new(`
`216`	`225`	`max_total_size: Option<Uint128>,`
`@@ -262,7 +271,6 @@ impl TryFrom<msg::BucketLimits> for BucketLimits {`
`262`	`271`	`)`
`263`	`272`	`}`
`264`	`273`	`}`
`265`		`-`
`266`	`274`	`/// Pagination is the type carrying configuration for paginated queries.`
`267`	`275`	`#[derive(Serialize, Deserialize, Clone, Debug, PartialEq, Eq, JsonSchema)]`
`268`	`276`	`pub struct Pagination {`