feat: add content_length to DownloadTaskResponse (#219)

Signed-off-by: Gaius <[email protected]>
dragonflyoss · Nov 20, 2023 · 2840dfe · 2840dfe
1 parent 7f93979
commit 2840dfe
Show file tree

Hide file tree

Showing 7 changed files with 74 additions and 52 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "dragonfly-api"
-version = "2.0.53"
+version = "2.0.54"
 authors = ["Gaius <[email protected]>"]
 edition = "2021"
 license = "Apache-2.0"

diff --git a/pkg/apis/dfdaemon/v2/dfdaemon.pb.go b/pkg/apis/dfdaemon/v2/dfdaemon.pb.go
diff --git a/pkg/apis/dfdaemon/v2/dfdaemon.pb.validate.go b/pkg/apis/dfdaemon/v2/dfdaemon.pb.validate.go
diff --git a/pkg/apis/dfdaemon/v2/dfdaemon.proto b/pkg/apis/dfdaemon/v2/dfdaemon.proto
@@ -77,8 +77,11 @@ message DownloadTaskRequest {
 
 // DownloadTaskResponse represents response of DownloadTask.
 message DownloadTaskResponse {
+  // Task content length.
+  int64 content_length = 1;
+
   // Finished piece of task.
-  common.v2.Piece piece = 1 [(validate.rules).message.required = true];
+  common.v2.Piece piece = 2 [(validate.rules).message.required = true];
 }
 
 // UploadTaskRequest represents request of UploadTask.

diff --git a/proto/dfdaemon.proto b/proto/dfdaemon.proto
@@ -70,8 +70,11 @@ message DownloadTaskRequest {
 
 // DownloadTaskResponse represents response of DownloadTask.
 message DownloadTaskResponse {
+  // Task content length.
+  int64 content_length = 1;
+
   // Finished piece of task.
-  common.v2.Piece piece = 1;
+  common.v2.Piece piece = 2;
 }
 
 // UploadTaskRequest represents request of UploadTask.

diff --git a/src/descriptor.bin b/src/descriptor.bin
diff --git a/src/dfdaemon.v2.rs b/src/dfdaemon.v2.rs
@@ -87,8 +87,11 @@ pub struct DownloadTaskRequest {
 #[allow(clippy::derive_partial_eq_without_eq)]
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct DownloadTaskResponse {
+    /// Task content length.
+    #[prost(int64, tag = "1")]
+    pub content_length: i64,
     /// Finished piece of task.
-    #[prost(message, optional, tag = "1")]
+    #[prost(message, optional, tag = "2")]
     pub piece: ::core::option::Option<super::super::common::v2::Piece>,
 }
 /// UploadTaskRequest represents request of UploadTask.