Replaced channel pointer field with non pointer channel
This commit is contained in:
parent
ead3dd9759
commit
2c229bac00
|
@ -45,7 +45,7 @@ func Start(pipe *xeth.XEth, config RpcConfig) error {
|
||||||
c := cors.New(opts)
|
c := cors.New(opts)
|
||||||
handler = NewStoppableHandler(c.Handler(JSONRPC(pipe)), l.stop)
|
handler = NewStoppableHandler(c.Handler(JSONRPC(pipe)), l.stop)
|
||||||
} else {
|
} else {
|
||||||
handler = JSONRPC(pipe)
|
handler = NewStoppableHandler(JSONRPC(pipe), l.stop)
|
||||||
}
|
}
|
||||||
|
|
||||||
go http.Serve(l, handler)
|
go http.Serve(l, handler)
|
||||||
|
|
23
rpc/types.go
23
rpc/types.go
|
@ -275,20 +275,21 @@ func (self ListenerStoppedError) Error() string {
|
||||||
|
|
||||||
var listenerStoppedError = ListenerStoppedError{"Listener stopped"}
|
var listenerStoppedError = ListenerStoppedError{"Listener stopped"}
|
||||||
|
|
||||||
|
// When https://github.com/golang/go/issues/4674 is fixed this could be replaced
|
||||||
type StoppableTCPListener struct {
|
type StoppableTCPListener struct {
|
||||||
*net.TCPListener
|
*net.TCPListener
|
||||||
stop *chan struct{} // closed when the listener must stop
|
stop chan struct{} // closed when the listener must stop
|
||||||
}
|
}
|
||||||
|
|
||||||
// Wraps the default handler and checks if the RPC service was stopped. In that case it returns an
|
// Wraps the default handler and checks if the RPC service was stopped. In that case it returns an
|
||||||
// error indicating that the service was stopped. This will only happen for connections which are
|
// error indicating that the service was stopped. This will only happen for connections which are
|
||||||
// kept open (HTTP keep-alive) when the RPC service was shutdown.
|
// kept open (HTTP keep-alive) when the RPC service was shutdown.
|
||||||
func NewStoppableHandler(h http.Handler, stop *chan struct{}) http.Handler {
|
func NewStoppableHandler(h http.Handler, stop chan struct{}) http.Handler {
|
||||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
select {
|
select {
|
||||||
case <-*stop:
|
case <-stop:
|
||||||
w.Header().Set("Content-Type", "application/json")
|
w.Header().Set("Content-Type", "application/json")
|
||||||
jsonerr := &RpcErrorObject{-32603, "RPC service stopt"}
|
jsonerr := &RpcErrorObject{-32603, "RPC service stopped"}
|
||||||
send(w, &RpcErrorResponse{Jsonrpc: jsonrpcver, Id: nil, Error: jsonerr})
|
send(w, &RpcErrorResponse{Jsonrpc: jsonrpcver, Id: nil, Error: jsonerr})
|
||||||
default:
|
default:
|
||||||
h.ServeHTTP(w, r)
|
h.ServeHTTP(w, r)
|
||||||
|
@ -298,7 +299,7 @@ func NewStoppableHandler(h http.Handler, stop *chan struct{}) http.Handler {
|
||||||
|
|
||||||
// Stop the listener and all accepted and still active connections.
|
// Stop the listener and all accepted and still active connections.
|
||||||
func (self *StoppableTCPListener) Stop() {
|
func (self *StoppableTCPListener) Stop() {
|
||||||
close(*self.stop)
|
close(self.stop)
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewStoppableTCPListener(addr string) (*StoppableTCPListener, error) {
|
func NewStoppableTCPListener(addr string) (*StoppableTCPListener, error) {
|
||||||
|
@ -309,7 +310,7 @@ func NewStoppableTCPListener(addr string) (*StoppableTCPListener, error) {
|
||||||
|
|
||||||
if tcpl, ok := wl.(*net.TCPListener); ok {
|
if tcpl, ok := wl.(*net.TCPListener); ok {
|
||||||
stop := make(chan struct{})
|
stop := make(chan struct{})
|
||||||
l := &StoppableTCPListener{tcpl, &stop}
|
l := &StoppableTCPListener{tcpl, stop}
|
||||||
return l, nil
|
return l, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -322,8 +323,10 @@ func (self *StoppableTCPListener) Accept() (net.Conn, error) {
|
||||||
c, err := self.TCPListener.AcceptTCP()
|
c, err := self.TCPListener.AcceptTCP()
|
||||||
|
|
||||||
select {
|
select {
|
||||||
case <-*self.stop:
|
case <-self.stop:
|
||||||
c.Close()
|
if c != nil { // accept timeout
|
||||||
|
c.Close()
|
||||||
|
}
|
||||||
self.TCPListener.Close()
|
self.TCPListener.Close()
|
||||||
return nil, listenerStoppedError
|
return nil, listenerStoppedError
|
||||||
default:
|
default:
|
||||||
|
@ -341,12 +344,12 @@ func (self *StoppableTCPListener) Accept() (net.Conn, error) {
|
||||||
|
|
||||||
type ClosableConnection struct {
|
type ClosableConnection struct {
|
||||||
*net.TCPConn
|
*net.TCPConn
|
||||||
closed *chan struct{}
|
closed chan struct{}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (self *ClosableConnection) Read(b []byte) (n int, err error) {
|
func (self *ClosableConnection) Read(b []byte) (n int, err error) {
|
||||||
select {
|
select {
|
||||||
case <-*self.closed:
|
case <-self.closed:
|
||||||
self.TCPConn.Close()
|
self.TCPConn.Close()
|
||||||
return 0, io.EOF
|
return 0, io.EOF
|
||||||
default:
|
default:
|
||||||
|
|
Loading…
Reference in New Issue