certusone
/
tendermint
mirror of https://github.com/certusone/tendermint.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Use go-wire's new Read/Write LengthPrefixed methods

This commit is contained in:
Jae Kwon 2015-12-20 09:23:30 -08:00
parent f288e58c81
commit e57f3596a1
1 changed files with 4 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
proxy/remote_app_context.go
View File

@ -2,7 +2,6 @@ package proxy
import (
"bufio"
"bytes"
"container/list"
"errors"
"fmt"
@ -84,7 +83,7 @@ func (app *remoteAppContext) Error() error {
func (app *remoteAppContext) sendRequestsRoutine() {
for {
var n, n2 int
var n int
var err error
select {
case <-app.QuitService.Quit:
@ -93,10 +92,7 @@ func (app *remoteAppContext) sendRequestsRoutine() {
app.willSendReq(reqres)
buf := new(bytes.Buffer)
wire.WriteBinary(reqres.Request, buf, &n2, &err) // Length prefix
wire.WriteVarint(buf.Len(), app.bufWriter, &n, &err)
wire.WriteTo(buf.Bytes(), app.bufWriter, &n, &err)
wire.WriteBinaryLengthPrefixed(reqres.Request, app.bufWriter, &n, &err) // Length prefix
if err != nil {
app.StopForError(err)
return
@ -117,17 +113,13 @@ func (app *remoteAppContext) recvResponseRoutine() {
r := bufio.NewReader(app.conn) // Buffer reads
for {
var res tmsp.Response
var n, n2 int
var n int
var err error
size := wire.ReadVarint(r, &n2, &err)
wire.ReadBinaryPtr(&res, r, maxResponseSize, &n, &err)
wire.ReadBinaryPtrLengthPrefixed(&res, r, maxResponseSize, &n, &err)
if err != nil {
app.StopForError(err)
return
}
if size != n {
app.StopForError(fmt.Errorf("Prefixed length not equal to actual length. %v vs %v", size, n))
}
switch res := res.(type) {
case tmsp.ResponseException:
app.StopForError(errors.New(res.Error))