summaryrefslogtreecommitdiff
path: root/internal/clients/catclient.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/clients/catclient.go')
-rw-r--r--internal/clients/catclient.go24
1 files changed, 15 insertions, 9 deletions
diff --git a/internal/clients/catclient.go b/internal/clients/catclient.go
index bd65560..e2e247e 100644
--- a/internal/clients/catclient.go
+++ b/internal/clients/catclient.go
@@ -2,9 +2,7 @@ package clients
import (
"errors"
- "fmt"
"runtime"
- "strings"
"github.com/mimecast/dtail/internal/clients/handlers"
"github.com/mimecast/dtail/internal/config"
@@ -26,14 +24,18 @@ func NewCatClient(args config.Args) (*CatClient, error) {
c := CatClient{
baseClient: baseClient{
+ mu: newBaseClientMu(),
Args: args,
throttleCh: make(chan struct{}, args.ConnectionsPerCPU*runtime.NumCPU()),
retry: false,
+ runtime: newClientRuntimeBoundary(config.CurrentRuntime()),
},
}
c.init()
- c.makeConnections(c)
+ if err := c.makeConnections(c); err != nil {
+ return nil, err
+ }
return &c, nil
}
@@ -41,14 +43,18 @@ func (c CatClient) makeHandler(server string) handlers.Handler {
return handlers.NewClientHandler(server)
}
+func (c CatClient) makeSessionSpec() (SessionSpec, error) {
+ return NewSessionSpec(c.Args), nil
+}
+
func (c CatClient) makeCommands() (commands []string) {
- regex, err := c.Regex.Serialize()
+ sessionSpec, err := c.makeSessionSpec()
if err != nil {
- dlog.Client.FatalPanic(err)
+ dlog.Client.FatalPanic("unable to build cat session spec", err)
}
- for _, file := range strings.Split(c.What, ",") {
- commands = append(commands, fmt.Sprintf("%s:%s %s %s",
- c.Mode.String(), c.Args.SerializeOptions(), file, regex))
+ commands, err = sessionSpec.Commands()
+ if err != nil {
+ dlog.Client.FatalPanic("unable to build cat commands from session spec", err)
}
- return
+ return commands
}