opensource/grpc-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #587 from tamird/perf

Browse Source 
transport: allocate bigger slabs and reuse
This commit is contained in:
Qi Zhao
2016-04-04 11:30:38 -07:00
parent 5aeebcd810 9e8752e7a2
commit 466b6e4052
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
transport/handler_server.go
View File

@ -313,16 +313,22 @@ func (ht *serverHandlerTransport) HandleStreams(startStream func(*Stream)) {
readerDone := make(chan struct{}) readerDone := make(chan struct{})
go func() { go func() {
defer close(readerDone) defer close(readerDone)
for {
buf := make([]byte, 1024) // TODO: minimize garbage, optimize recvBuffer code/ownership // TODO: minimize garbage, optimize recvBuffer code/ownership
const readSize = 8196
for buf := make([]byte, readSize); ; {
n, err := req.Body.Read(buf) n, err := req.Body.Read(buf)
if n > 0 { if n > 0 {
s.buf.put(&recvMsg{data: buf[:n]}) s.buf.put(&recvMsg{data: buf[:n:n]})
buf = buf[n:]
} }
if err != nil { if err != nil {
s.buf.put(&recvMsg{err: mapRecvMsgError(err)}) s.buf.put(&recvMsg{err: mapRecvMsgError(err)})
return return
} }
if len(buf) == 0 {
buf = make([]byte, readSize)
}
} }
}() }()