Add ability to have several servers
diff --git a/core/kg/cmd/commands/root.go b/core/kg/cmd/commands/root.go
index 812cdf1..b5c92e2 100644
--- a/core/kg/cmd/commands/root.go
+++ b/core/kg/cmd/commands/root.go
@@ -33,17 +33,10 @@
FileLocation: "server.log",
}
logger := log.NewLogger(config)
- srv, err := server.NewServer(logger)
- if err != nil {
- logger.Error(err.Error())
- return err
- }
- defer srv.Shutdown()
+ grpcServer := server.NewGRPCServer(logger)
+ servers := server.New(logger)
+ servers.AddServers(grpcServer)
+ servers.Run()
- serverErr := srv.Start()
- if serverErr != nil {
- logger.Error(err.Error())
- return serverErr
- }
return nil
}