diff options
Diffstat (limited to 'internal/clients/baseclient.go')
| -rw-r--r-- | internal/clients/baseclient.go | 23 |
1 files changed, 13 insertions, 10 deletions
diff --git a/internal/clients/baseclient.go b/internal/clients/baseclient.go index 102fd7c..ba18f95 100644 --- a/internal/clients/baseclient.go +++ b/internal/clients/baseclient.go @@ -38,18 +38,9 @@ type baseClient struct { Regex regex.Regex } -func (c *baseClient) init(maker maker) { +func (c *baseClient) init() { logger.Info("Initiating base client") - c.maker = maker - c.sshAuthMethods, c.hostKeyCallback = client.InitSSHAuthMethods(c.Args.SSHAuthMethods, c.Args.SSHHostKeyCallback, c.Args.TrustAllHosts, c.throttleCh, c.Args.PrivateKeyPathFile) - - discoveryService := discovery.New(c.Discovery, c.ServersStr, discovery.Shuffle) - - for _, server := range discoveryService.ServerList() { - c.connections = append(c.connections, c.makeConnection(server, c.sshAuthMethods, c.hostKeyCallback)) - } - flag := regex.Default if c.Args.RegexInvert { flag = regex.Invert @@ -59,6 +50,18 @@ func (c *baseClient) init(maker maker) { logger.FatalExit(c.Regex, "invalid regex!", err, regex) } c.Regex = regex + logger.Debug("Regex", c.Regex) + + c.sshAuthMethods, c.hostKeyCallback = client.InitSSHAuthMethods(c.Args.SSHAuthMethods, c.Args.SSHHostKeyCallback, c.Args.TrustAllHosts, c.throttleCh, c.Args.PrivateKeyPathFile) +} + +func (c *baseClient) makeConnections(maker maker) { + c.maker = maker + + discoveryService := discovery.New(c.Discovery, c.ServersStr, discovery.Shuffle) + for _, server := range discoveryService.ServerList() { + c.connections = append(c.connections, c.makeConnection(server, c.sshAuthMethods, c.hostKeyCallback)) + } c.stats = newTailStats(len(c.connections)) } |
