diff options
author | Petar Maymounkov <petarm@gmail.com> | 2010-02-19 08:21:21 -0800 |
---|---|---|
committer | Russ Cox <rsc@golang.org> | 2010-02-19 08:21:21 -0800 |
commit | 8d9c2b2eab723f442ea623ec59300a6cb9765448 (patch) | |
tree | f66c7b169cc59534a4d7d687aea645ccda5850b5 | |
parent | 666abfb64c6c61a215cea58092d0e1dd6c8e7c7b (diff) | |
download | go-8d9c2b2eab723f442ea623ec59300a6cb9765448.tar.gz go-8d9c2b2eab723f442ea623ec59300a6cb9765448.zip |
http: add Pending method to ServerConn, ClientConn
R=rsc
CC=golang-dev
https://golang.org/cl/216052
-rw-r--r-- | src/pkg/http/persist.go | 20 |
1 files changed, 19 insertions, 1 deletions
diff --git a/src/pkg/http/persist.go b/src/pkg/http/persist.go index a4da3da657..8bfc097558 100644 --- a/src/pkg/http/persist.go +++ b/src/pkg/http/persist.go @@ -45,6 +45,8 @@ func NewServerConn(c net.Conn, r *bufio.Reader) *ServerConn { // called before Read has signaled the end of the keep-alive logic. The user // should not call Close while Read or Write is in progress. func (sc *ServerConn) Close() (c net.Conn, r *bufio.Reader) { + sc.lk.Lock() + defer sc.lk.Unlock() c = sc.c r = sc.r sc.c = nil @@ -111,6 +113,14 @@ func (sc *ServerConn) Read() (req *Request, err os.Error) { return } +// Pending returns the number of unanswered requests +// that have been received on the connection. +func (sc *ServerConn) Pending() int { + sc.lk.Lock() + defer sc.lk.Unlock() + return sc.nread - sc.nwritten +} + // Write writes a repsonse. To close the connection gracefully, set the // Response.Close field to true. Write should be considered operational until // it returns an error, regardless of any errors returned on the Read side. @@ -176,11 +186,11 @@ func NewClientConn(c net.Conn, r *bufio.Reader) *ClientConn { // called before the user or Read have signaled the end of the keep-alive // logic. The user should not call Close while Read or Write is in progress. func (cc *ClientConn) Close() (c net.Conn, r *bufio.Reader) { + cc.lk.Lock() c = cc.c r = cc.r cc.c = nil cc.r = nil - cc.lk.Lock() cc.reqm.Init() cc.lk.Unlock() return @@ -228,6 +238,14 @@ func (cc *ClientConn) Write(req *Request) os.Error { return nil } +// Pending returns the number of unanswered requests +// that have been sent on the connection. +func (cc *ClientConn) Pending() int { + cc.lk.Lock() + defer cc.lk.Unlock() + return cc.nwritten - cc.nread +} + // Read reads the next response from the wire. A valid response might be // returned together with an ErrPersistEOF, which means that the remote // requested that this be the last request serviced. Read can be called |