feat: set output_path to optional and add need_piece_content for down…

…loading persistent cache Signed-off-by: Gaius <[email protected]>
dragonflyoss · Jan 2, 2025 · 38c8f3a · 38c8f3a
1 parent 190d05d
commit 38c8f3a
Show file tree

Hide file tree

Showing 10 changed files with 498 additions and 190 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "dragonfly-api"
-version = "2.1.1"
+version = "2.1.2"
 authors = ["Gaius <[email protected]>"]
 edition = "2021"
 license = "Apache-2.0"

diff --git a/build.rs b/build.rs
@@ -10,7 +10,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
         .extern_path(".google.protobuf.Timestamp", "::prost_wkt_types::Timestamp")
         .extern_path(".google.protobuf.Duration", "::prost_wkt_types::Duration")
         .out_dir("src")
-        .compile(
+        .compile_protos(
             &[
                 "proto/common.proto",
                 "proto/errordetails.proto",

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
@@ -138,9 +138,11 @@ message DownloadPersistentCacheTaskRequest {
   // Application of task.
   optional string application = 4;
   // File path to be exported.
-  string output_path = 5 [(validate.rules).string.min_len = 1];
+  optional string output_path = 5 [(validate.rules).string = {min_len: 1, ignore_empty: true}];
   // Download timeout.
   optional google.protobuf.Duration timeout = 6;
+  // need_piece_content is the flag to indicate whether the response needs to return piece content.
+  bool need_piece_content = 7;
 }
 
 // DownloadPersistentCacheTaskStartedResponse represents task download started response of DownloadPersistentCacheTaskResponse.

diff --git a/pkg/apis/dfdaemon/v2/mocks/dfdaemon_mock.go b/pkg/apis/dfdaemon/v2/mocks/dfdaemon_mock.go