1 files changed, 70 insertions, 0 deletions
diff --git a/client/go/cmd/query.go b/client/go/cmd/query.go
new file mode 100644
index 00000000000..58828dbb7cf
--- /dev/null
+++ b/client/go/cmd/query.go
@@ -0,0 +1,70 @@
+// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+// vespa query command
+// author: bratseth
+
+package cmd
+
+import (
+    "errors"
+    "github.com/spf13/cobra"
+    "github.com/vespa-engine/vespa/util"
+    "strings"
+    "net/http"
+    "net/url"
+    "time"
+)
+
+func init() {
+    rootCmd.AddCommand(queryCmd)
+}
+
+var queryCmd = &cobra.Command{
+    Use:   "query",
+    Short: "Issue a query to Vespa",
+    Long:  `TODO, example  \"yql=select from sources * where title contains 'foo'\" hits=5`,
+    // TODO: Support referencing a query json file
+    Args: func(cmd *cobra.Command, args []string) error {
+        if len(args) < 1 {
+            return errors.New("vespa query requires at least one argument containing the query string")
+        }
+        return nil
+    },
+    Run: func(cmd *cobra.Command, args []string) {
+        query(args)
+    },
+}
+
+func query(arguments []string) {
+    url, _ := url.Parse(getTarget(queryContext).query + "/search/")
+    urlQuery := url.Query()
+    for i := 0; i < len(arguments); i++ {
+        key, value := splitArg(arguments[i])
+        urlQuery.Set(key, value)
+    }
+    url.RawQuery = urlQuery.Encode()
+
+    response := util.HttpDo(&http.Request{URL: url,}, time.Second * 10, "Container")
+    if (response == nil) {
+        return
+    }
+    defer response.Body.Close()
+
+    if (response.StatusCode == 200) {
+        util.PrintReader(response.Body)
+    } else if response.StatusCode / 100 == 4 {
+        util.Error("Invalid query (" + response.Status + "):")
+        util.PrintReader(response.Body)
+    } else {
+        util.Error("Error from container at", url.Host, "(" + response.Status + "):")
+        util.PrintReader(response.Body)
+    }
+}
+
+func splitArg(argument string) (string, string) {
+    equalsIndex := strings.Index(argument, "=")
+    if equalsIndex < 1 {
+        return "yql", argument
+    } else {
+        return argument[0:equalsIndex], argument[equalsIndex + 1:len(argument)]
+    }
+}