From 9b60784dbbbbe257d967e5f9e1b0f7c09f07af6c Mon Sep 17 00:00:00 2001 From: Arnas Udovic Date: Tue, 27 Jun 2023 16:57:14 +0300 Subject: [PATCH] DownloadAttachment --- go.mod | 2 +- go.sum | 4 +- server.go | 30 +++++++ tree.pb.go | 210 +++++++++++++++++++++++++++++++++++++++--------- tree.proto | 9 +++ tree_grpc.pb.go | 68 +++++++++++++++- 6 files changed, 279 insertions(+), 44 deletions(-) diff --git a/go.mod b/go.mod index d6219f8..a10cce5 100644 --- a/go.mod +++ b/go.mod @@ -3,7 +3,7 @@ module g.arns.lt/zordsdavini/arns-lt-tree-push-service go 1.19 require ( - g.arns.lt/zordsdavini/zord-tree v0.2.6 + g.arns.lt/zordsdavini/zord-tree v0.2.7 google.golang.org/grpc v1.48.0 google.golang.org/protobuf v1.28.1 ) diff --git a/go.sum b/go.sum index f46646b..77761ae 100644 --- a/go.sum +++ b/go.sum @@ -2,8 +2,8 @@ cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMT cloud.google.com/go v0.34.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw= g.arns.lt/zordsdavini/abcex v1.0.0 h1:qQqlZ4DMfethCGK4I6yGaLqMrTzKNIshqpINd1l3t0E= g.arns.lt/zordsdavini/abcex v1.0.0/go.mod h1:YRcJgts3XZwI+LEkngpfUab3DkUAW387Irpr43hIym8= -g.arns.lt/zordsdavini/zord-tree v0.2.6 h1:utUtk7OTjhWC4Fltnzcn/QB7Kpc0yW9IBuyaYRhi+as= -g.arns.lt/zordsdavini/zord-tree v0.2.6/go.mod h1:HG++MTNUD8oQHrkTdVnClm0q82lGumASn4GWJtWQApY= +g.arns.lt/zordsdavini/zord-tree v0.2.7 h1:h19O+VmzaXA9E68XWIDTF4VIU4I1TnTd8TPigRCljLQ= +g.arns.lt/zordsdavini/zord-tree v0.2.7/go.mod h1:HG++MTNUD8oQHrkTdVnClm0q82lGumASn4GWJtWQApY= github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= github.com/antihax/optional v1.0.0/go.mod h1:uupD/76wgC+ih3iEmQUL+0Ugr19nfwCT1kdvxnR2qWY= github.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU= diff --git a/server.go b/server.go index 2c34e27..ef65abb 100644 --- a/server.go +++ b/server.go @@ -2,10 +2,13 @@ package main import ( "context" + "fmt" "google.golang.org/grpc" + "io" "log" "net" "os" + "path" "path/filepath" zordTree "g.arns.lt/zordsdavini/zord-tree" @@ -79,6 +82,33 @@ func (TreeManagerServer *server) RebuildTree(ctx context.Context, in *TreeSecret return &response, nil } +func (TreeManagerServer *server) DownloadAttachment(in *AttachmentRequest, responseStream TreeManager_DownloadAttachmentServer) error { + file, err := os.Open(path.Clean(fmt.Sprintf("%s/%s/%s", dest, "__a", in.Path))) + if err != nil { + return err + } + defer file.Close() + + bufferSize := 64 * 1024 + buff := make([]byte, bufferSize) + for { + bytesRead, err := file.Read(buff) + if err != nil { + if err != io.EOF { + log.Println("Error during downloadingAttachment: ", err) + } + break + } + resp := &AttachmentResponse{Chunk: buff[:bytesRead]} + + err = responseStream.Send(resp) + if err != nil { + return err + } + } + return nil +} + func removeContents(dir string) error { d, err := os.Open(dir) if err != nil { diff --git a/tree.pb.go b/tree.pb.go index 6c7598f..582f456 100644 --- a/tree.pb.go +++ b/tree.pb.go @@ -1,7 +1,7 @@ // Code generated by protoc-gen-go. DO NOT EDIT. // versions: -// protoc-gen-go v1.28.0 -// protoc v3.21.4 +// protoc-gen-go v1.30.0 +// protoc v3.21.12 // source: tree.proto package main @@ -461,6 +461,100 @@ func (x *FileContent) GetContent() string { return "" } +type AttachmentRequest struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + Path string `protobuf:"bytes,1,opt,name=path,proto3" json:"path,omitempty"` +} + +func (x *AttachmentRequest) Reset() { + *x = AttachmentRequest{} + if protoimpl.UnsafeEnabled { + mi := &file_tree_proto_msgTypes[7] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *AttachmentRequest) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*AttachmentRequest) ProtoMessage() {} + +func (x *AttachmentRequest) ProtoReflect() protoreflect.Message { + mi := &file_tree_proto_msgTypes[7] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use AttachmentRequest.ProtoReflect.Descriptor instead. +func (*AttachmentRequest) Descriptor() ([]byte, []int) { + return file_tree_proto_rawDescGZIP(), []int{7} +} + +func (x *AttachmentRequest) GetPath() string { + if x != nil { + return x.Path + } + return "" +} + +type AttachmentResponse struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + Chunk []byte `protobuf:"bytes,1,opt,name=chunk,proto3" json:"chunk,omitempty"` +} + +func (x *AttachmentResponse) Reset() { + *x = AttachmentResponse{} + if protoimpl.UnsafeEnabled { + mi := &file_tree_proto_msgTypes[8] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *AttachmentResponse) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*AttachmentResponse) ProtoMessage() {} + +func (x *AttachmentResponse) ProtoReflect() protoreflect.Message { + mi := &file_tree_proto_msgTypes[8] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use AttachmentResponse.ProtoReflect.Descriptor instead. +func (*AttachmentResponse) Descriptor() ([]byte, []int) { + return file_tree_proto_rawDescGZIP(), []int{8} +} + +func (x *AttachmentResponse) GetChunk() []byte { + if x != nil { + return x.Chunk + } + return nil +} + type TreeRequest_Filter struct { state protoimpl.MessageState sizeCache protoimpl.SizeCache @@ -473,7 +567,7 @@ type TreeRequest_Filter struct { func (x *TreeRequest_Filter) Reset() { *x = TreeRequest_Filter{} if protoimpl.UnsafeEnabled { - mi := &file_tree_proto_msgTypes[7] + mi := &file_tree_proto_msgTypes[9] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } @@ -486,7 +580,7 @@ func (x *TreeRequest_Filter) String() string { func (*TreeRequest_Filter) ProtoMessage() {} func (x *TreeRequest_Filter) ProtoReflect() protoreflect.Message { - mi := &file_tree_proto_msgTypes[7] + mi := &file_tree_proto_msgTypes[9] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { @@ -576,19 +670,29 @@ var file_tree_proto_rawDesc = []byte{ 0x04, 0x66, 0x69, 0x6c, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x0e, 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x54, 0x72, 0x65, 0x65, 0x46, 0x69, 0x6c, 0x65, 0x52, 0x04, 0x66, 0x69, 0x6c, 0x65, 0x12, 0x18, 0x0a, 0x07, 0x63, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x18, 0x02, 0x20, 0x01, - 0x28, 0x09, 0x52, 0x07, 0x63, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x32, 0xa9, 0x01, 0x0a, 0x0b, - 0x54, 0x72, 0x65, 0x65, 0x4d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x12, 0x2d, 0x0a, 0x0a, 0x47, - 0x65, 0x74, 0x53, 0x75, 0x6d, 0x6d, 0x65, 0x72, 0x79, 0x12, 0x11, 0x2e, 0x6d, 0x61, 0x69, 0x6e, - 0x2e, 0x54, 0x72, 0x65, 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x0a, 0x2e, 0x6d, - 0x61, 0x69, 0x6e, 0x2e, 0x54, 0x72, 0x65, 0x65, 0x22, 0x00, 0x12, 0x31, 0x0a, 0x07, 0x47, 0x65, - 0x74, 0x46, 0x69, 0x6c, 0x65, 0x12, 0x11, 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x46, 0x69, 0x6c, - 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x11, 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, - 0x46, 0x69, 0x6c, 0x65, 0x43, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x22, 0x00, 0x12, 0x38, 0x0a, - 0x0b, 0x52, 0x65, 0x62, 0x75, 0x69, 0x6c, 0x64, 0x54, 0x72, 0x65, 0x65, 0x12, 0x10, 0x2e, 0x6d, - 0x61, 0x69, 0x6e, 0x2e, 0x54, 0x72, 0x65, 0x65, 0x53, 0x65, 0x63, 0x72, 0x65, 0x74, 0x1a, 0x15, - 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x52, 0x65, 0x62, 0x75, 0x69, 0x6c, 0x64, 0x52, 0x65, 0x73, - 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x42, 0x03, 0x5a, 0x01, 0x2e, 0x62, 0x06, 0x70, 0x72, - 0x6f, 0x74, 0x6f, 0x33, + 0x28, 0x09, 0x52, 0x07, 0x63, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x22, 0x27, 0x0a, 0x11, 0x41, + 0x74, 0x74, 0x61, 0x63, 0x68, 0x6d, 0x65, 0x6e, 0x74, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, + 0x12, 0x12, 0x0a, 0x04, 0x70, 0x61, 0x74, 0x68, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x04, + 0x70, 0x61, 0x74, 0x68, 0x22, 0x2a, 0x0a, 0x12, 0x41, 0x74, 0x74, 0x61, 0x63, 0x68, 0x6d, 0x65, + 0x6e, 0x74, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x14, 0x0a, 0x05, 0x63, 0x68, + 0x75, 0x6e, 0x6b, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x05, 0x63, 0x68, 0x75, 0x6e, 0x6b, + 0x32, 0xf6, 0x01, 0x0a, 0x0b, 0x54, 0x72, 0x65, 0x65, 0x4d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, + 0x12, 0x2d, 0x0a, 0x0a, 0x47, 0x65, 0x74, 0x53, 0x75, 0x6d, 0x6d, 0x65, 0x72, 0x79, 0x12, 0x11, + 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x54, 0x72, 0x65, 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, + 0x74, 0x1a, 0x0a, 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x54, 0x72, 0x65, 0x65, 0x22, 0x00, 0x12, + 0x31, 0x0a, 0x07, 0x47, 0x65, 0x74, 0x46, 0x69, 0x6c, 0x65, 0x12, 0x11, 0x2e, 0x6d, 0x61, 0x69, + 0x6e, 0x2e, 0x46, 0x69, 0x6c, 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x11, 0x2e, + 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x46, 0x69, 0x6c, 0x65, 0x43, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, + 0x22, 0x00, 0x12, 0x38, 0x0a, 0x0b, 0x52, 0x65, 0x62, 0x75, 0x69, 0x6c, 0x64, 0x54, 0x72, 0x65, + 0x65, 0x12, 0x10, 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x54, 0x72, 0x65, 0x65, 0x53, 0x65, 0x63, + 0x72, 0x65, 0x74, 0x1a, 0x15, 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x52, 0x65, 0x62, 0x75, 0x69, + 0x6c, 0x64, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x12, 0x4b, 0x0a, 0x12, + 0x44, 0x6f, 0x77, 0x6e, 0x6c, 0x6f, 0x61, 0x64, 0x41, 0x74, 0x74, 0x61, 0x63, 0x68, 0x6d, 0x65, + 0x6e, 0x74, 0x12, 0x17, 0x2e, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x41, 0x74, 0x74, 0x61, 0x63, 0x68, + 0x6d, 0x65, 0x6e, 0x74, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x18, 0x2e, 0x6d, 0x61, + 0x69, 0x6e, 0x2e, 0x41, 0x74, 0x74, 0x61, 0x63, 0x68, 0x6d, 0x65, 0x6e, 0x74, 0x52, 0x65, 0x73, + 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x30, 0x01, 0x42, 0x03, 0x5a, 0x01, 0x2e, 0x62, 0x06, + 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, } var ( @@ -603,7 +707,7 @@ func file_tree_proto_rawDescGZIP() []byte { return file_tree_proto_rawDescData } -var file_tree_proto_msgTypes = make([]protoimpl.MessageInfo, 10) +var file_tree_proto_msgTypes = make([]protoimpl.MessageInfo, 12) var file_tree_proto_goTypes = []interface{}{ (*TreeSecret)(nil), // 0: main.TreeSecret (*RebuildResponse)(nil), // 1: main.RebuildResponse @@ -612,28 +716,32 @@ var file_tree_proto_goTypes = []interface{}{ (*TreeFile)(nil), // 4: main.TreeFile (*Tree)(nil), // 5: main.Tree (*FileContent)(nil), // 6: main.FileContent - (*TreeRequest_Filter)(nil), // 7: main.TreeRequest.Filter - nil, // 8: main.Tree.TagsEntry - nil, // 9: main.Tree.CategoriesEntry + (*AttachmentRequest)(nil), // 7: main.AttachmentRequest + (*AttachmentResponse)(nil), // 8: main.AttachmentResponse + (*TreeRequest_Filter)(nil), // 9: main.TreeRequest.Filter + nil, // 10: main.Tree.TagsEntry + nil, // 11: main.Tree.CategoriesEntry } var file_tree_proto_depIdxs = []int32{ - 7, // 0: main.TreeRequest.filter:type_name -> main.TreeRequest.Filter - 4, // 1: main.Tree.files:type_name -> main.TreeFile - 4, // 2: main.Tree.rootFiles:type_name -> main.TreeFile - 8, // 3: main.Tree.tags:type_name -> main.Tree.TagsEntry - 9, // 4: main.Tree.categories:type_name -> main.Tree.CategoriesEntry - 4, // 5: main.FileContent.file:type_name -> main.TreeFile - 3, // 6: main.TreeManager.GetSummery:input_type -> main.TreeRequest - 2, // 7: main.TreeManager.GetFile:input_type -> main.FileRequest - 0, // 8: main.TreeManager.RebuildTree:input_type -> main.TreeSecret - 5, // 9: main.TreeManager.GetSummery:output_type -> main.Tree - 6, // 10: main.TreeManager.GetFile:output_type -> main.FileContent - 1, // 11: main.TreeManager.RebuildTree:output_type -> main.RebuildResponse - 9, // [9:12] is the sub-list for method output_type - 6, // [6:9] is the sub-list for method input_type - 6, // [6:6] is the sub-list for extension type_name - 6, // [6:6] is the sub-list for extension extendee - 0, // [0:6] is the sub-list for field type_name + 9, // 0: main.TreeRequest.filter:type_name -> main.TreeRequest.Filter + 4, // 1: main.Tree.files:type_name -> main.TreeFile + 4, // 2: main.Tree.rootFiles:type_name -> main.TreeFile + 10, // 3: main.Tree.tags:type_name -> main.Tree.TagsEntry + 11, // 4: main.Tree.categories:type_name -> main.Tree.CategoriesEntry + 4, // 5: main.FileContent.file:type_name -> main.TreeFile + 3, // 6: main.TreeManager.GetSummery:input_type -> main.TreeRequest + 2, // 7: main.TreeManager.GetFile:input_type -> main.FileRequest + 0, // 8: main.TreeManager.RebuildTree:input_type -> main.TreeSecret + 7, // 9: main.TreeManager.DownloadAttachment:input_type -> main.AttachmentRequest + 5, // 10: main.TreeManager.GetSummery:output_type -> main.Tree + 6, // 11: main.TreeManager.GetFile:output_type -> main.FileContent + 1, // 12: main.TreeManager.RebuildTree:output_type -> main.RebuildResponse + 8, // 13: main.TreeManager.DownloadAttachment:output_type -> main.AttachmentResponse + 10, // [10:14] is the sub-list for method output_type + 6, // [6:10] is the sub-list for method input_type + 6, // [6:6] is the sub-list for extension type_name + 6, // [6:6] is the sub-list for extension extendee + 0, // [0:6] is the sub-list for field type_name } func init() { file_tree_proto_init() } @@ -727,6 +835,30 @@ func file_tree_proto_init() { } } file_tree_proto_msgTypes[7].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*AttachmentRequest); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + file_tree_proto_msgTypes[8].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*AttachmentResponse); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + file_tree_proto_msgTypes[9].Exporter = func(v interface{}, i int) interface{} { switch v := v.(*TreeRequest_Filter); i { case 0: return &v.state @@ -746,7 +878,7 @@ func file_tree_proto_init() { GoPackagePath: reflect.TypeOf(x{}).PkgPath(), RawDescriptor: file_tree_proto_rawDesc, NumEnums: 0, - NumMessages: 10, + NumMessages: 12, NumExtensions: 0, NumServices: 1, }, diff --git a/tree.proto b/tree.proto index e135fa9..45aaeb9 100644 --- a/tree.proto +++ b/tree.proto @@ -11,6 +11,7 @@ service TreeManager { rpc GetSummery(TreeRequest) returns (Tree) {} rpc GetFile(FileRequest) returns (FileContent) {} rpc RebuildTree(TreeSecret) returns (RebuildResponse) {} + rpc DownloadAttachment(AttachmentRequest) returns (stream AttachmentResponse) {} } message TreeSecret { @@ -59,3 +60,11 @@ message FileContent { TreeFile file = 1; string content = 2; } + +message AttachmentRequest { + string path = 1; +} + +message AttachmentResponse { + bytes chunk = 1; +} diff --git a/tree_grpc.pb.go b/tree_grpc.pb.go index ba9822e..0f284c9 100644 --- a/tree_grpc.pb.go +++ b/tree_grpc.pb.go @@ -1,7 +1,7 @@ // Code generated by protoc-gen-go-grpc. DO NOT EDIT. // versions: // - protoc-gen-go-grpc v1.2.0 -// - protoc v3.21.4 +// - protoc v3.21.12 // source: tree.proto package main @@ -25,6 +25,7 @@ type TreeManagerClient interface { GetSummery(ctx context.Context, in *TreeRequest, opts ...grpc.CallOption) (*Tree, error) GetFile(ctx context.Context, in *FileRequest, opts ...grpc.CallOption) (*FileContent, error) RebuildTree(ctx context.Context, in *TreeSecret, opts ...grpc.CallOption) (*RebuildResponse, error) + DownloadAttachment(ctx context.Context, in *AttachmentRequest, opts ...grpc.CallOption) (TreeManager_DownloadAttachmentClient, error) } type treeManagerClient struct { @@ -62,6 +63,38 @@ func (c *treeManagerClient) RebuildTree(ctx context.Context, in *TreeSecret, opt return out, nil } +func (c *treeManagerClient) DownloadAttachment(ctx context.Context, in *AttachmentRequest, opts ...grpc.CallOption) (TreeManager_DownloadAttachmentClient, error) { + stream, err := c.cc.NewStream(ctx, &TreeManager_ServiceDesc.Streams[0], "/main.TreeManager/DownloadAttachment", opts...) + if err != nil { + return nil, err + } + x := &treeManagerDownloadAttachmentClient{stream} + if err := x.ClientStream.SendMsg(in); err != nil { + return nil, err + } + if err := x.ClientStream.CloseSend(); err != nil { + return nil, err + } + return x, nil +} + +type TreeManager_DownloadAttachmentClient interface { + Recv() (*AttachmentResponse, error) + grpc.ClientStream +} + +type treeManagerDownloadAttachmentClient struct { + grpc.ClientStream +} + +func (x *treeManagerDownloadAttachmentClient) Recv() (*AttachmentResponse, error) { + m := new(AttachmentResponse) + if err := x.ClientStream.RecvMsg(m); err != nil { + return nil, err + } + return m, nil +} + // TreeManagerServer is the server API for TreeManager service. // All implementations must embed UnimplementedTreeManagerServer // for forward compatibility @@ -69,6 +102,7 @@ type TreeManagerServer interface { GetSummery(context.Context, *TreeRequest) (*Tree, error) GetFile(context.Context, *FileRequest) (*FileContent, error) RebuildTree(context.Context, *TreeSecret) (*RebuildResponse, error) + DownloadAttachment(*AttachmentRequest, TreeManager_DownloadAttachmentServer) error mustEmbedUnimplementedTreeManagerServer() } @@ -85,6 +119,9 @@ func (UnimplementedTreeManagerServer) GetFile(context.Context, *FileRequest) (*F func (UnimplementedTreeManagerServer) RebuildTree(context.Context, *TreeSecret) (*RebuildResponse, error) { return nil, status.Errorf(codes.Unimplemented, "method RebuildTree not implemented") } +func (UnimplementedTreeManagerServer) DownloadAttachment(*AttachmentRequest, TreeManager_DownloadAttachmentServer) error { + return status.Errorf(codes.Unimplemented, "method DownloadAttachment not implemented") +} func (UnimplementedTreeManagerServer) mustEmbedUnimplementedTreeManagerServer() {} // UnsafeTreeManagerServer may be embedded to opt out of forward compatibility for this service. @@ -152,6 +189,27 @@ func _TreeManager_RebuildTree_Handler(srv interface{}, ctx context.Context, dec return interceptor(ctx, in, info, handler) } +func _TreeManager_DownloadAttachment_Handler(srv interface{}, stream grpc.ServerStream) error { + m := new(AttachmentRequest) + if err := stream.RecvMsg(m); err != nil { + return err + } + return srv.(TreeManagerServer).DownloadAttachment(m, &treeManagerDownloadAttachmentServer{stream}) +} + +type TreeManager_DownloadAttachmentServer interface { + Send(*AttachmentResponse) error + grpc.ServerStream +} + +type treeManagerDownloadAttachmentServer struct { + grpc.ServerStream +} + +func (x *treeManagerDownloadAttachmentServer) Send(m *AttachmentResponse) error { + return x.ServerStream.SendMsg(m) +} + // TreeManager_ServiceDesc is the grpc.ServiceDesc for TreeManager service. // It's only intended for direct use with grpc.RegisterService, // and not to be introspected or modified (even as a copy) @@ -172,6 +230,12 @@ var TreeManager_ServiceDesc = grpc.ServiceDesc{ Handler: _TreeManager_RebuildTree_Handler, }, }, - Streams: []grpc.StreamDesc{}, + Streams: []grpc.StreamDesc{ + { + StreamName: "DownloadAttachment", + Handler: _TreeManager_DownloadAttachment_Handler, + ServerStreams: true, + }, + }, Metadata: "tree.proto", }