Introduce drun command, refactor code to use context package

1 files changed, 35 insertions, 0 deletions

diff --git a/internal/server/handlers/mapcommand.go b/internal/server/handlers/mapcommand.go
new file mode 100644
index 0000000..10372da
--- /dev/null
+++ b/internal/server/handlers/mapcommand.go
@@ -0,0 +1,35 @@
+package handlers
+
+import (
+	"context"
+	"strings"
+
+	"github.com/mimecast/dtail/internal/mapr/server"
+)
+
+// Map command implements the mapreduce command server side.
+type mapCommand struct {
+	aggregate *server.Aggregate
+	server    *ServerHandler
+}
+
+// NewMapCommand returns a new server side mapreduce command.
+func newMapCommand(serverHandler *ServerHandler, argc int, args []string) (mapCommand, *server.Aggregate, error) {
+	mapCommand := mapCommand{
+		server: serverHandler,
+	}
+
+	queryStr := strings.Join(args[1:], " ")
+	aggregate, err := server.NewAggregate(queryStr)
+	if err != nil {
+		return mapCommand, nil, err
+	}
+
+	mapCommand.aggregate = aggregate
+	return mapCommand, aggregate, nil
+
+}
+
+func (m mapCommand) Start(ctx context.Context, aggregatedMessages chan<- string) {
+	m.aggregate.Start(ctx, aggregatedMessages)
+}