ing

client.go

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

server.go

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