From ed3b39b6ac2a56b3f4eff405daa46e6ea7d89eff Mon Sep 17 00:00:00 2001 From: Tobie Morgan Hitchcock Date: Wed, 15 Nov 2017 18:24:09 +0000 Subject: [PATCH] Add sql command to the command line ui Add sql command to the command line ui , useful for executing and processing an sql file against a local or remote database. --- cli/cli.go | 1 + cli/sql.go | 149 +++++++++++++++++++++++++++++++++++++++++++++++++++++ cnf/cnf.go | 4 ++ 3 files changed, 154 insertions(+) create mode 100644 cli/sql.go diff --git a/cli/cli.go b/cli/cli.go index 455b02a9..183d6ed5 100644 --- a/cli/cli.go +++ b/cli/cli.go @@ -32,6 +32,7 @@ var mainCmd = &cobra.Command{ func init() { mainCmd.AddCommand( + sqlCmd, startCmd, importCmd, exportCmd, diff --git a/cli/sql.go b/cli/sql.go new file mode 100644 index 00000000..7283a0a5 --- /dev/null +++ b/cli/sql.go @@ -0,0 +1,149 @@ +// Copyright © 2016 Abcum Ltd +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +package cli + +import ( + "fmt" + "net/http" + "os" + + "io/ioutil" + + "github.com/spf13/cobra" + + "github.com/abcum/surreal/log" +) + +var sqlCmd = &cobra.Command{ + Use: "sql [flags] ", + Short: "Execute a SQL script against an existing database", + Example: " surreal sql --auth root:root script.sql", + RunE: func(cmd *cobra.Command, args []string) (err error) { + + var fle *os.File + var req *http.Request + var res *http.Response + + // Ensure that the command has a filepath + // as the output file argument. If no filepath + // has been provided then return an error. + + if len(args) != 1 { + log.Fatalln("No filepath provided.") + return + } + + // Attempt to open or create the specified file + // in write-only mode, and if there is a problem + // creating the file, then return an error. + + if fle, err = os.OpenFile(args[0], os.O_RDONLY, 0644); err != nil { + log.Fatalln("Import failed - please check the filepath and try again.") + return + } + + // Ensure that we properly close the file handle + // when we have finished with the file so that + // the file descriptor is released. + + defer fle.Close() + + // Configure the export connection endpoint url + // and specify the authentication header using + // basic auth for root login. + + url := fmt.Sprintf("http://%s@%s:%s/sql", opts.Auth.Auth, opts.DB.Host, opts.DB.Port) + + // Create a new http request object that we + // can use to connect to the import endpoint + // using a POST http request type. + + if req, err = http.NewRequest("POST", url, fle); err != nil { + log.Fatalln("Connection failed - check the connection details and try again.") + return + } + + // Specify that the request is an octet stream + // so that we can stream the file contents to + // the server without reading the whole file. + + switch opts.Format.Type { + case "pack": + req.Header.Set("Content-Type", "application/msgpack") + case "json": + req.Header.Set("Content-Type", "application/json") + case "cork": + req.Header.Set("Content-Type", "application/cork") + default: + req.Header.Set("Content-Type", "text/plain") + } + + // Attempt to dial the import endpoint and + // if there is an error then stop execution + // and return the connection error. + + if res, err = http.DefaultClient.Do(req); err != nil { + log.Fatalln("Connection failed - check the connection details and try again.") + return + } + + // Ensure that we close the body, otherwise + // if the Body is not closed, the Client can + // not use the underlying transport again. + + defer res.Body.Close() + + // Ensure that we didn't receive a 401 status + // code back from the server, otherwise there + // was a problem with our authentication. + + if res.StatusCode == 401 { + log.Fatalln("Authentication failed - check the connection details and try again.") + return + } + + // Ensure that we received a http 200 status + // code back from the server, otherwise there + // was a problem with our request. + + if res.StatusCode != 200 { + bdy, _ := ioutil.ReadAll(res.Body) + log.Fatalf("%s", bdy) + return + } + + // Otherwise let's output the whole response + // body to the terminal, using the output + // format specified in the request. + + if bdy, _ := ioutil.ReadAll(res.Body); len(bdy) > 0 { + log.SetLevel("INFO") + log.Infof("%s", bdy) + } + + return + + }, +} + +func init() { + + sqlCmd.PersistentFlags().StringVar(&opts.Auth.Auth, "auth", "root:root", "Master authentication details to use when connecting.") + sqlCmd.PersistentFlags().StringVar(&opts.DB.Host, "host", "surreal.io", "Database server host to connect to.") + sqlCmd.PersistentFlags().StringVar(&opts.DB.Port, "port", "80", "Database server port to connect to.") + + sqlCmd.PersistentFlags().StringVar(&opts.Format.Type, "format", "text", "The output format for the server response data.") + +} diff --git a/cnf/cnf.go b/cnf/cnf.go index 007e3237..42423dcd 100644 --- a/cnf/cnf.go +++ b/cnf/cnf.go @@ -112,6 +112,10 @@ type Options struct { Join []string // Slice of cluster peers to join } + Format struct { + Type string // Stores the cli output format + } + Logging struct { Level string // Stores the configured logging level Output string // Stores the configured logging output