Skip to content

Commit 28f83a2

Browse files
committed
fixup
1 parent 8cf7f5c commit 28f83a2

File tree

1 file changed

+2
-11
lines changed

1 file changed

+2
-11
lines changed

provisionerd/provisionerd.go

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -534,12 +534,7 @@ func (p *Server) UploadModuleFiles(ctx context.Context, moduleFiles []byte) erro
534534

535535
dataUp, chunks := sdkproto.BytesToDataUpload(sdkproto.DataUploadType_UPLOAD_TYPE_MODULE_FILES, moduleFiles)
536536

537-
err = stream.Send(&proto.UploadFileRequest{Type: &proto.UploadFileRequest_DataUpload{DataUpload: &proto.DataUpload{
538-
UploadType: proto.DataUploadType(dataUp.UploadType),
539-
DataHash: dataUp.DataHash,
540-
FileSize: dataUp.FileSize,
541-
Chunks: dataUp.Chunks,
542-
}}})
537+
err = stream.Send(&proto.UploadFileRequest{Type: &proto.UploadFileRequest_DataUpload{DataUpload: dataUp}})
543538
if err != nil {
544539
if retryable(err) { // Do not retry
545540
return nil, xerrors.Errorf("send data upload: %s", err.Error())
@@ -548,11 +543,7 @@ func (p *Server) UploadModuleFiles(ctx context.Context, moduleFiles []byte) erro
548543
}
549544

550545
for i, chunk := range chunks {
551-
err = stream.Send(&proto.UploadFileRequest{Type: &proto.UploadFileRequest_ChunkPiece{ChunkPiece: &proto.ChunkPiece{
552-
Data: chunk.Data,
553-
FullDataHash: chunk.FullDataHash,
554-
PieceIndex: chunk.PieceIndex,
555-
}}})
546+
err = stream.Send(&proto.UploadFileRequest{Type: &proto.UploadFileRequest_ChunkPiece{ChunkPiece: chunk}})
556547
if err != nil {
557548
if retryable(err) { // Do not retry
558549
return nil, xerrors.Errorf("send chunk piece: %s", err.Error())

0 commit comments

Comments
 (0)