feat: add stream response to DownloadTask (#212)

Signed-off-by: Gaius <[email protected]>
dragonflyoss · Nov 15, 2023 · 6d86592 · 6d86592
1 parent 3394a57
commit 6d86592
Show file tree

Hide file tree

Showing 9 changed files with 640 additions and 127 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "dragonfly-api"
-version = "2.0.49"
+version = "2.0.50"
 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
@@ -75,6 +75,12 @@ message DownloadTaskRequest {
   common.v2.Download download = 1 [(validate.rules).message.required = true];
 }
 
+// DownloadTaskResponse represents response of DownloadTask.
+message DownloadTaskResponse {
+  // Finished piece of task.
+  common.v2.Piece piece = 1 [(validate.rules).message.required = true];
+}
+
 // UploadTaskRequest represents request of UploadTask.
 message UploadTaskRequest {
   // Task metadata.
@@ -102,7 +108,7 @@ service Dfdaemon {
   rpc SyncPieces(stream SyncPiecesRequest)returns(stream SyncPiecesResponse);
 
   // DownloadTask downloads task back-to-source.
-  rpc DownloadTask(DownloadTaskRequest) returns(google.protobuf.Empty);
+  rpc DownloadTask(DownloadTaskRequest) returns(stream DownloadTaskResponse);
 
   // UploadTask uploads task to p2p network.
   rpc UploadTask(UploadTaskRequest) returns(google.protobuf.Empty);

diff --git a/pkg/apis/dfdaemon/v2/dfdaemon_grpc.pb.go b/pkg/apis/dfdaemon/v2/dfdaemon_grpc.pb.go