2014-10-21 11:24:48 +00:00
|
|
|
package rpc
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"net"
|
|
|
|
"net/rpc"
|
|
|
|
"net/rpc/jsonrpc"
|
|
|
|
|
2014-10-23 13:01:27 +00:00
|
|
|
"github.com/ethereum/go-ethereum/ethpipe"
|
2014-10-31 11:56:05 +00:00
|
|
|
"github.com/ethereum/go-ethereum/logger"
|
2014-10-21 11:24:48 +00:00
|
|
|
)
|
|
|
|
|
2014-10-31 11:56:05 +00:00
|
|
|
var jsonlogger = logger.NewLogger("JSON")
|
2014-10-21 11:24:48 +00:00
|
|
|
|
|
|
|
type JsonRpcServer struct {
|
|
|
|
quit chan bool
|
|
|
|
listener net.Listener
|
|
|
|
pipe *ethpipe.JSPipe
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *JsonRpcServer) exitHandler() {
|
|
|
|
out:
|
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case <-s.quit:
|
|
|
|
s.listener.Close()
|
|
|
|
break out
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-10-31 11:56:05 +00:00
|
|
|
jsonlogger.Infoln("Shutdown JSON-RPC server")
|
2014-10-21 11:24:48 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (s *JsonRpcServer) Stop() {
|
|
|
|
close(s.quit)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *JsonRpcServer) Start() {
|
2014-10-31 11:56:05 +00:00
|
|
|
jsonlogger.Infoln("Starting JSON-RPC server")
|
2014-10-21 11:24:48 +00:00
|
|
|
go s.exitHandler()
|
|
|
|
rpc.Register(&EthereumApi{pipe: s.pipe})
|
|
|
|
rpc.HandleHTTP()
|
|
|
|
|
|
|
|
for {
|
|
|
|
conn, err := s.listener.Accept()
|
|
|
|
if err != nil {
|
2014-10-31 11:56:05 +00:00
|
|
|
jsonlogger.Infoln("Error starting JSON-RPC:", err)
|
2014-10-21 11:24:48 +00:00
|
|
|
break
|
|
|
|
}
|
2014-10-31 11:56:05 +00:00
|
|
|
jsonlogger.Debugln("Incoming request.")
|
2014-10-21 11:24:48 +00:00
|
|
|
go jsonrpc.ServeConn(conn)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewJsonRpcServer(pipe *ethpipe.JSPipe, port int) (*JsonRpcServer, error) {
|
|
|
|
sport := fmt.Sprintf(":%d", port)
|
|
|
|
l, err := net.Listen("tcp", sport)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return &JsonRpcServer{
|
|
|
|
listener: l,
|
|
|
|
quit: make(chan bool),
|
|
|
|
pipe: pipe,
|
|
|
|
}, nil
|
|
|
|
}
|