NewXXXServer doesnt run Start or return error

This commit is contained in:
Ethan Buchman
2017-05-15 12:51:24 -04:00
parent 21fff49f2c
commit d07b2352ad
9 changed files with 35 additions and 25 deletions

View File

@ -27,6 +27,10 @@ func main() {
os.Exit(1) os.Exit(1)
} }
srv.SetLogger(logger.With("module", "abci-server")) srv.SetLogger(logger.With("module", "abci-server"))
if _, err := srv.Start(); err != nil {
logger.Error(err.Error())
os.Exit(1)
}
// Wait forever // Wait forever
cmn.TrapSignal(func() { cmn.TrapSignal(func() {

View File

@ -36,6 +36,10 @@ func main() {
os.Exit(1) os.Exit(1)
} }
srv.SetLogger(logger.With("module", "abci-server")) srv.SetLogger(logger.With("module", "abci-server"))
if _, err := srv.Start(); err != nil {
logger.Error(err.Error())
os.Exit(1)
}
// Wait forever // Wait forever
cmn.TrapSignal(func() { cmn.TrapSignal(func() {

View File

@ -25,6 +25,10 @@ func main() {
os.Exit(1) os.Exit(1)
} }
srv.SetLogger(logger.With("module", "abci-server")) srv.SetLogger(logger.With("module", "abci-server"))
if _, err := srv.Start(); err != nil {
logger.Error(err.Error())
os.Exit(1)
}
// Wait forever // Wait forever
cmn.TrapSignal(func() { cmn.TrapSignal(func() {

View File

@ -20,6 +20,9 @@ func TestChainAware(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
srv.SetLogger(log.TestingLogger().With("module", "abci-server")) srv.SetLogger(log.TestingLogger().With("module", "abci-server"))
if _, err := srv.Start(); err != nil {
t.Fatal(err.Error())
}
defer srv.Stop() defer srv.Stop()
// Connect to the socket // Connect to the socket

View File

@ -214,22 +214,21 @@ func makeSocketClientServer(app types.Application, name string) (abcicli.Client,
socket := cmn.Fmt("unix://%s.sock", name) socket := cmn.Fmt("unix://%s.sock", name)
logger := log.TestingLogger() logger := log.TestingLogger()
server, err := server.NewSocketServer(socket, app) server := server.NewSocketServer(socket, app)
if err != nil { server.SetLogger(logger.With("module", "abci-server"))
if _, err := server.Start(); err != nil {
return nil, nil, err return nil, nil, err
} }
server.SetLogger(logger.With("module", "abci-server"))
// Connect to the socket // Connect to the socket
client := abcicli.NewSocketClient(socket, false) client := abcicli.NewSocketClient(socket, false)
client.SetLogger(logger.With("module", "abci-client"))
if _, err := client.Start(); err != nil { if _, err := client.Start(); err != nil {
server.Stop() server.Stop()
return nil, nil, err return nil, nil, err
} }
client.SetLogger(logger.With("module", "abci-client"))
client.Start()
return client, server, err return client, server, nil
} }
func makeGRPCClientServer(app types.Application, name string) (abcicli.Client, cmn.Service, error) { func makeGRPCClientServer(app types.Application, name string) (abcicli.Client, cmn.Service, error) {
@ -238,19 +237,19 @@ func makeGRPCClientServer(app types.Application, name string) (abcicli.Client, c
logger := log.TestingLogger() logger := log.TestingLogger()
gapp := types.NewGRPCApplication(app) gapp := types.NewGRPCApplication(app)
server, err := server.NewGRPCServer(socket, gapp) server := server.NewGRPCServer(socket, gapp)
if err != nil { server.SetLogger(logger.With("module", "abci-server"))
if _, err := server.Start(); err != nil {
return nil, nil, err return nil, nil, err
} }
server.SetLogger(logger.With("module", "abci-server"))
client := abcicli.NewGRPCClient(socket, true) client := abcicli.NewGRPCClient(socket, true)
client.SetLogger(logger.With("module", "abci-client"))
if _, err := client.Start(); err != nil { if _, err := client.Start(); err != nil {
server.Stop() server.Stop()
return nil, nil, err return nil, nil, err
} }
client.SetLogger(logger.With("module", "abci-client")) return client, server, nil
return client, server, err
} }
func TestClientServer(t *testing.T) { func TestClientServer(t *testing.T) {

View File

@ -38,8 +38,8 @@ func testStream(t *testing.T, app types.Application) {
numDeliverTxs := 200000 numDeliverTxs := 200000
// Start the listener // Start the listener
server, err := server.NewSocketServer("unix://test.sock", app) server := server.NewSocketServer("unix://test.sock", app)
if err != nil { if _, err := server.Start(); err != nil {
t.Fatalf("Error starting socket server: %v", err.Error()) t.Fatalf("Error starting socket server: %v", err.Error())
} }
server.SetLogger(log.TestingLogger().With("module", "abci-server")) server.SetLogger(log.TestingLogger().With("module", "abci-server"))
@ -112,8 +112,8 @@ func testGRPCSync(t *testing.T, app *types.GRPCApplication) {
numDeliverTxs := 2000 numDeliverTxs := 2000
// Start the listener // Start the listener
server, err := server.NewGRPCServer("unix://test.sock", app) server := server.NewGRPCServer("unix://test.sock", app)
if err != nil { if _, err := server.Start(); err != nil {
t.Fatalf("Error starting GRPC server: %v", err.Error()) t.Fatalf("Error starting GRPC server: %v", err.Error())
} }
server.SetLogger(log.TestingLogger().With("module", "abci-server")) server.SetLogger(log.TestingLogger().With("module", "abci-server"))

View File

@ -22,7 +22,7 @@ type GRPCServer struct {
} }
// NewGRPCServer returns a new gRPC ABCI server // NewGRPCServer returns a new gRPC ABCI server
func NewGRPCServer(protoAddr string, app types.ABCIApplicationServer) (cmn.Service, error) { func NewGRPCServer(protoAddr string, app types.ABCIApplicationServer) cmn.Service {
parts := strings.SplitN(protoAddr, "://", 2) parts := strings.SplitN(protoAddr, "://", 2)
proto, addr := parts[0], parts[1] proto, addr := parts[0], parts[1]
s := &GRPCServer{ s := &GRPCServer{
@ -32,8 +32,7 @@ func NewGRPCServer(protoAddr string, app types.ABCIApplicationServer) (cmn.Servi
app: app, app: app,
} }
s.BaseService = *cmn.NewBaseService(nil, "ABCIServer", s) s.BaseService = *cmn.NewBaseService(nil, "ABCIServer", s)
_, err := s.Start() // Just start it return s
return s, err
} }
// OnStart starts the gRPC service // OnStart starts the gRPC service

View File

@ -21,9 +21,9 @@ func NewServer(protoAddr, transport string, app types.Application) (cmn.Service,
var err error var err error
switch transport { switch transport {
case "socket": case "socket":
s, err = NewSocketServer(protoAddr, app) s = NewSocketServer(protoAddr, app)
case "grpc": case "grpc":
s, err = NewGRPCServer(protoAddr, types.NewGRPCApplication(app)) s = NewGRPCServer(protoAddr, types.NewGRPCApplication(app))
default: default:
err = fmt.Errorf("Unknown server type %s", transport) err = fmt.Errorf("Unknown server type %s", transport)
} }

View File

@ -29,7 +29,7 @@ type SocketServer struct {
app types.Application app types.Application
} }
func NewSocketServer(protoAddr string, app types.Application) (cmn.Service, error) { func NewSocketServer(protoAddr string, app types.Application) cmn.Service {
parts := strings.SplitN(protoAddr, "://", 2) parts := strings.SplitN(protoAddr, "://", 2)
proto, addr := parts[0], parts[1] proto, addr := parts[0], parts[1]
s := &SocketServer{ s := &SocketServer{
@ -40,10 +40,7 @@ func NewSocketServer(protoAddr string, app types.Application) (cmn.Service, erro
conns: make(map[int]net.Conn), conns: make(map[int]net.Conn),
} }
s.BaseService = *cmn.NewBaseService(nil, "ABCIServer", s) s.BaseService = *cmn.NewBaseService(nil, "ABCIServer", s)
// FIXME we are loosing "Starting ABCIServer" message here return s
// add logger to params?
_, err := s.Start() // Just start it
return s, err
} }
func (s *SocketServer) OnStart() error { func (s *SocketServer) OnStart() error {