pinot 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3fa42f97735f850dc0f870866e7d6da882227a998533b3aaa94f7c6a320d20ec
-  data.tar.gz: '019daf46b029bdac3a7c196d0a5f2927234f63ba513873318c5e0f6d48e1d3c6'
+  metadata.gz: 6443fe94263c767cc5039c400a7684fae467ab856e2d04253959ef6e0042e1d1
+  data.tar.gz: 4cb05bc8fe5c6f9b83cffa623e359f3c90fdc768575033e13968a78c5f7d43ff
 SHA512:
-  metadata.gz: 03e285bd5884bae307bd58797738d478f96cf97466ec0e6992dba0de6ab5120f935fba9245c4c5f1466794cf83da4a42dcc30b6819289e736c40cc374a054639
-  data.tar.gz: 07345f9e2e93442855887c208d586d98a0922bc346e19cac7bc58efbea1b4a79aadafaf9abd70692957890211583b2202502fe92996278f7e87f28a503fcfd88
+  metadata.gz: f217a53c1c2712e8d098aa94e5f3a4a93995e0839cad0f1a640f78c6e5269f7847afe9ef6d8344c2eb1d1df85ea087a5fb0db853ce3eb8c2d65d029ddbc6d554
+  data.tar.gz: 70694ca05dd5de01ef4e1d8af6e8bc61b899840e73ccb2b9680150d5d9dd48553fba2002607e6b22a9e3e1939d08d625e8c1d672adc6f4001345ad0ea5da9c5a

data/CHANGELOG.md

@@ -1,5 +1,19 @@
 ## [Unreleased]
 
+## [0.1.3] - 2024-07-02
+
+- feature: add tables command
+
+## [0.1.2] - 2024-05-29
+
+- feature: support socks5
+- feature: support authentication bearer token
+
+## [0.1.1] - 2024-05-16
+
+- feature: Add support for controller host
+- fix: handle nil column names
+
 ## [0.1.0] - 2024-01-16
 
 - Initial release

data/README.md

@@ -1,24 +1,71 @@
 # Pinot
 
-TODO: Delete this and the text below, and describe your gem
-
-Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/pinot`. To experiment with that code, run `bin/console` for an interactive prompt.
-
 ## Installation
 
-TODO: Replace `UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG` with your gem name right after releasing it to RubyGems.org. Please do not do it earlier due to security reasons. Alternatively, replace this section with instructions to install your gem from git if you don't plan to release to RubyGems.org.
-
 Install the gem and add to the application's Gemfile by executing:
 
-    $ bundle add UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG
+    $ bundle add pinot
 
 If bundler is not being used to manage dependencies, install the gem by executing:
 
-    $ gem install UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG
+    $ gem install pinot
 
 ## Usage
 
-TODO: Write usage instructions here
+To configure our client, we must have some data about our pinot cluster, its broken and controller host, and which port/protocol both services are using. After having this information, we can setup the client to use it.
+
+```ruby
+host = "localhost"
+controller_host = "localhost"
+client = Pinot::Client.new(host: host, port: 443, protocol: :https, controller_host: controller_host, controller_port: 443)
+```
+
+After setting up your client, we can start using it to query the cluster.
+
+```ruby
+client.execute("select * from table;")
+```
+
+After querying, it returns a `HTTPX::ErrorResponse` in case of any HTTP error, like a timeout, or connection refused, we are returning the client's error to make easier to troubleshoot any problem and handle it accordingly.
+
+In case of success, it returns a `Pinot::Response`, an enumerable where you can interact over the returned rows.
+
+Each row is an array with row's values.
+
+```ruby
+result = client.execute("select id, name, age from people;")
+result.each do |row|
+  puts row[0] # id
+  puts row[1] # name
+  puts row[2] # age
+end
+```
+
+If the column names and types are needed, can call `columns` method on the response.
+
+```ruby
+result = client.execute("select count(*) from people;")
+resp.columns
+# {"count(*)"=>"LONG"}
+```
+
+It's not on the public API, but the instance variable `@payload` contains all information returned from the Pinot cluster in case can help with troubleshooting
+
+### Authentication
+
+In case your pinot cluster is using authentication bearer token, you can specify it on the client constructor
+
+```ruby
+client = Pinot::Client.new(bearer_token: "my-secret", **client_args)
+```
+
+### Socks5
+
+Using Pinot cluster inside a VPC and need to jump through a bastion host using socks5? We got your back!
+
+```ruby
+client = Pinot::Client.new(socks5_uri: "socks5://localhost:80", **client_args)
+```
 
 ## Development
 
@@ -28,8 +75,8 @@ To install this gem onto your local machine, run `bundle exec rake install`. To
 
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/pinot. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [code of conduct](https://github.com/[USERNAME]/pinot/blob/main/CODE_OF_CONDUCT.md).
+Bug reports and pull requests are welcome on GitHub at https://github.com/clickfunnels2/pinot. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [code of conduct](https://github.com/clickfunnels2/pinot/blob/main/CODE_OF_CONDUCT.md).
 
 ## Code of Conduct
 
-Everyone interacting in the Pinot project's codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/[USERNAME]/pinot/blob/main/CODE_OF_CONDUCT.md).
+Everyone interacting in the Pinot project's codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/clickfunnels2/pinot/blob/main/CODE_OF_CONDUCT.md).

data/lib/pinot/client.rb

@@ -1,17 +1,21 @@
 module Pinot
   class Client
-    attr_reader :host, :port, :controller_host, :controller_port, :protocol
+    attr_reader :host, :port, :controller_host, :controller_port, :protocol, :socks5_uri, :bearer_token
 
-    def initialize(host:, port:, controller_port:, controller_host: nil, protocol: :http)
+    def initialize(host:, port:, controller_port:, controller_host: nil, protocol: :http, socks5_uri: nil, bearer_token: nil)
       @host = host
       @port = port
       @controller_port = controller_port
       @controller_host = controller_host || host
       @protocol = protocol
+      @socks5_uri = socks5_uri
+      @bearer_token = bearer_token
     end
 
     def execute(sql)
-      Response.new(JSON.parse(http.post(query_sql_uri, json: {sql: sql})))
+      response = http.post(query_sql_uri, json: {sql: sql})
+      return response if response.is_a?(HTTPX::ErrorResponse)
+      Response.new(JSON.parse(response))
     end
 
     def schema(name)
@@ -62,8 +66,21 @@ module Pinot
       JSON.parse(response)
     end
 
+    def tables
+      url = "#{controller_uri}/tables"
+      response = http.get(url)
+      JSON.parse(response)
+    end
+
     def http(content_type: "application/json")
-      HTTPX.with(headers: {"Content-Type" => content_type})
+      return @http if !@http.nil?
+      default_headers = {"Content-Type" => content_type}
+      default_headers["Authorization"] = "Bearer #{bearer_token}" if bearer_token
+      @http = HTTPX.with(headers: default_headers, timeout: {connect_timeout: 5})
+      if socks5_uri
+        @http = @http.plugin(:proxy).with_proxy(uri: socks5_uri) if socks5_uri
+      end
+      @http
     end
 
     def uri

data/lib/pinot/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Pinot
-  VERSION = "0.1.1"
+  VERSION = "0.1.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pinot
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - Celso Fernandes
-autorequire:
+autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-05-16 00:00:00.000000000 Z
+date: 2024-07-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httpx
@@ -90,7 +90,7 @@ metadata:
   homepage_uri: https://github.com/fernandes/pinot-ruby
   source_code_uri: https://github.com/fernandes/pinot
   changelog_uri: https://github.com/fernandes/pinot/blob/main/CHANGELOG.md
-post_install_message:
+post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -105,8 +105,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.3
-signing_key:
+rubygems_version: 3.5.14
+signing_key: 
 specification_version: 4
 summary: Client for Apache Pinot
 test_files: []