This commit is contained in:
crusader 2017-10-27 16:38:35 +09:00
parent ee5c8cf463
commit 6dc7895947
2 changed files with 12 additions and 0 deletions

View File

@ -13,6 +13,8 @@ import (
type Client interface { type Client interface {
Start() error Start() error
Stop() Stop()
GetClientHandler() ClientHandler
} }
func NewClient(ch ClientHandler) Client { func NewClient(ch ClientHandler) Client {
@ -31,6 +33,10 @@ type client struct {
stopWg sync.WaitGroup stopWg sync.WaitGroup
} }
func (c *client) GetClientHandler() ClientHandler {
return c.clientHandler
}
func (c *client) Start() error { func (c *client) Start() error {
if nil == c.clientHandler { if nil == c.clientHandler {
panic("Client: client handler must be specified.") panic("Client: client handler must be specified.")

View File

@ -15,6 +15,8 @@ type Server interface {
Start() error Start() error
Stop() Stop()
Serve() error Serve() error
GetServerHandler() ServerHandler
} }
func NewServer(sh ServerHandler) Server { func NewServer(sh ServerHandler) Server {
@ -35,6 +37,10 @@ type server struct {
stopWg sync.WaitGroup stopWg sync.WaitGroup
} }
func (s *server) GetServerHandler() ServerHandler {
return s.serverHandler
}
func (s *server) Start() error { func (s *server) Start() error {
if nil == s.serverHandler { if nil == s.serverHandler {
panic("Server: server handler must be specified.") panic("Server: server handler must be specified.")