Peter Kurfer
d89ab3a576
- introduce CLI binary - move server binary to subdirectory to have a uniformed directory structure - update goreleaser to build all binaries and create dist packages - update ignore files accordingly
116 lines
2.7 KiB
Go
116 lines
2.7 KiB
Go
package cmd
|
|
|
|
import (
|
|
"github.com/baez90/inetmock/internal/endpoints"
|
|
"github.com/baez90/inetmock/internal/plugins"
|
|
"github.com/baez90/inetmock/internal/rpc"
|
|
"github.com/baez90/inetmock/pkg/api"
|
|
"github.com/baez90/inetmock/pkg/config"
|
|
"github.com/baez90/inetmock/pkg/logging"
|
|
"github.com/baez90/inetmock/pkg/path"
|
|
"github.com/spf13/cobra"
|
|
"go.uber.org/zap"
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
endpointManager endpoints.EndpointManager
|
|
serveCmd = &cobra.Command{
|
|
Use: "serve",
|
|
Short: "Starts the INetMock server",
|
|
Long: ``,
|
|
Run: startINetMock,
|
|
}
|
|
)
|
|
|
|
func onServerInit() {
|
|
logging.ConfigureLogging(
|
|
logging.ParseLevel(logLevel),
|
|
developmentLogs,
|
|
map[string]interface{}{"cwd": path.WorkingDirectory()},
|
|
)
|
|
|
|
logger, _ = logging.CreateLogger()
|
|
config.CreateConfig(serverCmd.Flags())
|
|
appConfig := config.Instance()
|
|
|
|
if err := appConfig.ReadConfig(configFilePath); err != nil {
|
|
logger.Error(
|
|
"failed to read config file",
|
|
zap.Error(err),
|
|
)
|
|
}
|
|
|
|
if err := api.InitServices(appConfig, logger); err != nil {
|
|
logger.Error(
|
|
"failed to initialize app services",
|
|
zap.Error(err),
|
|
)
|
|
}
|
|
|
|
registry := plugins.Registry()
|
|
cfg := config.Instance()
|
|
|
|
pluginLoadStartTime := time.Now()
|
|
if err := registry.LoadPlugins(cfg.PluginsDir()); err != nil {
|
|
logger.Error("Failed to load plugins",
|
|
zap.String("pluginsDirectory", cfg.PluginsDir()),
|
|
zap.Error(err),
|
|
)
|
|
}
|
|
pluginLoadDuration := time.Since(pluginLoadStartTime)
|
|
logger.Info(
|
|
"loading plugins completed",
|
|
zap.Duration("pluginLoadDuration", pluginLoadDuration),
|
|
)
|
|
}
|
|
|
|
func startINetMock(_ *cobra.Command, _ []string) {
|
|
onServerInit()
|
|
endpointManager = endpoints.NewEndpointManager(logger)
|
|
cfg := config.Instance()
|
|
rpcAPI := rpc.NewINetMockAPI(
|
|
cfg,
|
|
endpointManager,
|
|
plugins.Registry(),
|
|
)
|
|
|
|
for endpointName, endpointHandler := range cfg.EndpointConfigs() {
|
|
handlerSubConfig := cfg.Viper().Sub(strings.Join([]string{config.EndpointsKey, endpointName, config.OptionsKey}, "."))
|
|
endpointHandler.Options = handlerSubConfig
|
|
if err := endpointManager.CreateEndpoint(endpointName, endpointHandler); err != nil {
|
|
logger.Warn(
|
|
"error occurred while creating endpoint",
|
|
zap.String("endpointName", endpointName),
|
|
zap.String("handlerName", endpointHandler.Handler),
|
|
zap.Error(err),
|
|
)
|
|
}
|
|
}
|
|
|
|
endpointManager.StartEndpoints()
|
|
if err := rpcAPI.StartServer(); err != nil {
|
|
logger.Error(
|
|
"failed to start gRPC API",
|
|
zap.Error(err),
|
|
)
|
|
}
|
|
|
|
signalChannel := make(chan os.Signal, 1)
|
|
signal.Notify(signalChannel, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
|
|
|
|
// block until canceled
|
|
s := <-signalChannel
|
|
|
|
logger.Info(
|
|
"got signal to quit",
|
|
zap.String("signal", s.String()),
|
|
)
|
|
|
|
rpcAPI.StopServer()
|
|
endpointManager.ShutdownEndpoints()
|
|
}
|