extern/zrepl
1
0
Fork 0
mirror of https://github.com/zrepl/zrepl.git synced 2024-11-22 00:13:52 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

rpc: fix incremental transfer request handling

Browse Source
This commit is contained in:
Christian Schwarz 2017-05-07 11:53:47 +02:00
parent 44d20ac256
commit c71be910f9
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
rpc/rpc.go
View File

@ -186,6 +186,15 @@ func ListenByteStreamRPC(conn io.ReadWriteCloser, handler RPCHandler) error {
if err != nil {
respondWithError(encoder, EHandler, err)
} else {
r := ResponseHeader{
RequestId: header.Id,
ResponseType: RChunkedStream,
}
if err := encoder.Encode(&r); err != nil {
panic(err)
}
chunker := NewChunker(snapReader)
_, err := io.Copy(conn, &chunker)
if err != nil {
@ -225,6 +234,8 @@ func inferRequestType(v interface{}) (RequestType, error) {
return RTFilesystemVersionsRequest, nil
case InitialTransferRequest:
return RTInitialTransferRequest, nil
case IncrementalTransferRequest:
return RTIncrementalTransferRequest, nil
default:
return 0, errors.New(fmt.Sprintf("cannot infer request type for type '%v'",
reflect.TypeOf(v)))