Introduce drun command, refactor code to use context package

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
+}