summaryrefslogtreecommitdiff
path: root/internal/clients/runclient.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/clients/runclient.go')
-rw-r--r--internal/clients/runclient.go40
1 files changed, 40 insertions, 0 deletions
diff --git a/internal/clients/runclient.go b/internal/clients/runclient.go
new file mode 100644
index 0000000..7a62fcc
--- /dev/null
+++ b/internal/clients/runclient.go
@@ -0,0 +1,40 @@
+package clients
+
+import (
+ "fmt"
+ "runtime"
+
+ "github.com/mimecast/dtail/internal/clients/handlers"
+ "github.com/mimecast/dtail/internal/omode"
+)
+
+// RunClient is a client to run various commands on the server.
+type RunClient struct {
+ baseClient
+}
+
+// NewRunClient returns a new cat client.
+func NewRunClient(args Args) (*RunClient, error) {
+ args.Mode = omode.RunClient
+
+ c := RunClient{
+ baseClient: baseClient{
+ Args: args,
+ throttleCh: make(chan struct{}, args.ConnectionsPerCPU*runtime.NumCPU()),
+ retry: false,
+ },
+ }
+
+ c.init(c)
+ return &c, nil
+}
+
+func (c RunClient) makeHandler(server string) handlers.Handler {
+ return handlers.NewClientHandler(server)
+}
+
+func (c RunClient) makeCommands() (commands []string) {
+ // Send "run COMMAND" to server!
+ commands = append(commands, fmt.Sprintf("%s %s", c.Mode.String(), c.What))
+ return
+}