druid_client 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9a81c003d77d8da968f48e5ff8702892b98449819b5629f0eb83fb34e7930cdd
-  data.tar.gz: 882564c48dd0551757b273c4f2e703d922b64574a61ab4ff9102657beca915e0
+  metadata.gz: d73b3aacd3fd035ffa9c8d172ca2f185c274b09ade9d7e859cb45d3d5132cb37
+  data.tar.gz: 116b09541bf93686a6d305ed4e21e643113f3a0273428eb702314cb95d106f79
 SHA512:
-  metadata.gz: e723a1dc81da695bde0987398597105f75aa878ae21d6c22c4406ccf09f3abfe4e51f4dfe350fa13c72e9a0641df7e5a1aa6044195c1b2c83fa1ab8d98564065
-  data.tar.gz: 91660b9d43dcbe92b9d4ea38c963566b8dd36358705b532974a7fb0186f6eea8be9a0be6bc58fd7f3185b739359004dc39d8b8caeb855ad5b24cde2ce4149b72
+  metadata.gz: c4de4aaa5b97488d401b249ec2d0ac44c9cbb18aea557967ec53dce890ca092ae0385c6b11eeef54d026ed214746f18a451779793a8c88cc11981ce56cb2c6b1
+  data.tar.gz: 40368bd913861a91db6a915cb96a623f328d5fb002d9ef66e5407796eea072d7439212fd6acd412a79bd2e70a220f5e08ab53b9c4d389d8e40edce814ae52aef

data/exe/druid_client

@@ -0,0 +1,64 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require 'gli'
+require 'druid_client'
+require 'druid_client/cli'
+
+class DruidClientCli
+  extend GLI::App
+
+  program_desc 'Druid client'
+  version DruidClient::VERSION
+
+  subcommand_option_handling :normal
+  arguments :strict
+
+  desc 'Druid coordinator URL'
+  flag %i[url U], default_value: 'http://localhost:8081'
+
+  desc 'Username for basic authentication'
+  flag %i[username u]
+
+  desc 'Password for basic authentication'
+  flag %i[password p]
+
+  desc 'Output format (table, json)'
+  flag [:F, :format],
+       default_value: 'table',
+       must_match: %w[table json]
+
+  desc 'Submit query to Druid'
+  command :query do |q|
+    q.desc 'Submit SQL query'
+    q.arg_name 'sql_query', 'The SQL query to execute'
+    q.command :sql do |sql|
+
+      desc 'Output format (table, json)'
+      sql.flag [:F, :format],
+               default_value: 'table',
+               must_match: %w[table json]
+
+      sql.action do |global_options, options, args|
+        DruidClient::Cli.new(options: global_options)
+          .sql
+          .query(query: args.join(' ').strip,
+                                     options: options)
+      end
+    end
+  end
+
+  pre do |_global, _command, _options, _args|
+    true
+  end
+
+  post do |_global, _command, _options, _args|
+    true
+  end
+
+  on_error do |_exception|
+    true
+  end
+end
+
+exit DruidClientCli.run(ARGV)

data/lib/druid_client/cli/json_format.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require 'json'
+
+require_relative 'output_format'
+
+module DruidClient
+  class CLi
+    class JsonFormat
+      include OutputFormat
+
+      def format(data)
+        JSON.pretty_generate(data)
+      end
+    end
+  end
+end

data/lib/druid_client/cli/output_format.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module DruidClient
+  class CLi
+    module OutputFormat
+      def format(data)
+        raise NotImplementedError, "Subclasses must implement #format method"
+      end
+    end
+  end
+end

data/lib/druid_client/cli/output_format_factory.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require_relative 'table_format'
+require_relative 'json_format'
+
+module DruidClient
+  class CLi
+    class OutputFormatFactory
+      puts self
+      FORMATS = {
+        table: TableFormat,
+        json: JsonFormat,
+      }.freeze
+
+      def self.create(format_name)
+        format_class = FORMATS[format_name.to_sym]
+        raise "Unknown format: #{format_name}" unless format_class
+        format_class.new
+      end
+    end
+  end
+end

data/lib/druid_client/cli/sql.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require_relative 'output_format_factory'
+
+module DruidClient
+  class Cli
+    class Sql
+      def initialize(parent: nil)
+        @parent = parent
+      end
+
+      def query(query:, options: {})
+        query_body = { query: query }
+        query_response = client.sql.query(
+          **query_body
+        )
+
+        unless  query_response.success?
+          display_error(
+            status_code: query_response.status_code,
+            error_message: query_response.body
+          )
+          return
+        end
+
+        formater = DruidClient::CLi::OutputFormatFactory.create(options.fetch(:format, :table))
+        puts formater.format(query_response.body)
+      end
+
+      def display_error(status_code: , error_message: )
+        puts "failed: status_code:#{status_code}, message:#{error_message}"
+      end
+
+      def client
+        @parent.client
+      end
+    end
+  end
+end

data/lib/druid_client/cli/table_format.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require 'terminal-table'
+
+require_relative 'output_format'
+
+module DruidClient
+  class CLi
+    class TableFormat
+      include OutputFormat
+
+      def format(data)
+        table = Terminal::Table.new do |t|
+          headings = data.first.keys
+          t.headings = headings
+          data[1..-1].each do |row|
+            t << headings.map { |heading| row[heading] }
+          end
+        end
+        table.to_s
+      end
+    end
+  end
+end

data/lib/druid_client/cli.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require_relative 'api'
+require_relative 'cli/sql'
+
+module DruidClient
+  class Cli
+    def initialize(options: {})
+      @options = options
+    end
+
+    def sql
+      @sql ||= Cli::Sql.new(parent: self)
+    end
+
+    def client
+      @client ||= DruidClient::Api.new(
+          url: @options[:url],
+          username: @options[:username],
+          password: @options[:password],
+      )
+    end
+  end
+end

data/lib/druid_client/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module DruidClient
-  VERSION = '0.1.2'
+  VERSION = '0.1.3'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: druid_client
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Thomas Tych
@@ -230,13 +230,15 @@ dependencies:
 description: Apache Druid DB client
 email:
 - thomas.tych@gmail.com
-executables: []
+executables:
+- druid_client
 extensions: []
 extra_rdoc_files: []
 files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- exe/druid_client
 - extra/fake_druid_api.rb
 - lib/druid_client.rb
 - lib/druid_client/api.rb
@@ -245,6 +247,12 @@ files:
 - lib/druid_client/api/response.rb
 - lib/druid_client/api/rest_client.rb
 - lib/druid_client/api/sql.rb
+- lib/druid_client/cli.rb
+- lib/druid_client/cli/json_format.rb
+- lib/druid_client/cli/output_format.rb
+- lib/druid_client/cli/output_format_factory.rb
+- lib/druid_client/cli/sql.rb
+- lib/druid_client/cli/table_format.rb
 - lib/druid_client/druid_config.rb
 - lib/druid_client/error.rb
 - lib/druid_client/version.rb