pinot 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 80d2be5dfc3b959fa6e7dce9eef3cd7600f0a6ad3f5d261230a7487e1482c2af
-  data.tar.gz: 83e396f7d954f381fe4b88326a8217e84c827793f331e229d47af0310e5cb8fe
+  metadata.gz: 3fa42f97735f850dc0f870866e7d6da882227a998533b3aaa94f7c6a320d20ec
+  data.tar.gz: '019daf46b029bdac3a7c196d0a5f2927234f63ba513873318c5e0f6d48e1d3c6'
 SHA512:
-  metadata.gz: ae37bb8d8c521f537603e6bcbf1f82df414e55072bb11bf83511e1f0780656720635d597b1ce0ab1b699a21a685988e3e7fe90f611e10625741d231bae9794a5
-  data.tar.gz: e0965c6c1a37d19e665a8545a01b35f346d5cd4955f08e6c3e51af4bac0b05a35e49213fc2e5192d76a3a4754e292de8029e76fec8a4e8ac289da6547b000284
+  metadata.gz: 03e285bd5884bae307bd58797738d478f96cf97466ec0e6992dba0de6ab5120f935fba9245c4c5f1466794cf83da4a42dcc30b6819289e736c40cc374a054639
+  data.tar.gz: 07345f9e2e93442855887c208d586d98a0922bc346e19cac7bc58efbea1b4a79aadafaf9abd70692957890211583b2202502fe92996278f7e87f28a503fcfd88

data/Guardfile

@@ -0,0 +1,42 @@
+# A sample Guardfile
+# More info at https://github.com/guard/guard#readme
+
+## Uncomment and set this to only include directories you want to watch
+# directories %w(app lib config test spec features) \
+#  .select{|d| Dir.exist?(d) ? d : UI.warning("Directory #{d} does not exist")}
+
+## Note: if you are using the `directories` clause above and you are not
+## watching the project directory ('.'), then you will want to move
+## the Guardfile to a watched dir and symlink it back, e.g.
+#
+#  $ mkdir config
+#  $ mv Guardfile config/
+#  $ ln -s config/Guardfile .
+#
+# and, you'll have to watch "config/Guardfile" instead of "Guardfile"
+
+guard :minitest do
+  # with Minitest::Unit
+  watch(%r{^test/(.*)/?test_(.*)\.rb$})
+  watch(%r{^lib/(.*/)?([^/]+)\.rb$}) { |m| "test/#{m[1]}test_#{m[2]}.rb" }
+  watch(%r{^test/test_helper\.rb$}) { "test" }
+
+  # with Minitest::Spec
+  # watch(%r{^spec/(.*)_spec\.rb$})
+  # watch(%r{^lib/(.+)\.rb$})         { |m| "spec/#{m[1]}_spec.rb" }
+  # watch(%r{^spec/spec_helper\.rb$}) { 'spec' }
+
+  # Rails 4
+  # watch(%r{^app/(.+)\.rb$})                               { |m| "test/#{m[1]}_test.rb" }
+  # watch(%r{^app/controllers/application_controller\.rb$}) { 'test/controllers' }
+  # watch(%r{^app/controllers/(.+)_controller\.rb$})        { |m| "test/integration/#{m[1]}_test.rb" }
+  # watch(%r{^app/views/(.+)_mailer/.+})                    { |m| "test/mailers/#{m[1]}_mailer_test.rb" }
+  # watch(%r{^lib/(.+)\.rb$})                               { |m| "test/lib/#{m[1]}_test.rb" }
+  # watch(%r{^test/.+_test\.rb$})
+  # watch(%r{^test/test_helper\.rb$}) { 'test' }
+
+  # Rails < 4
+  # watch(%r{^app/controllers/(.*)\.rb$}) { |m| "test/functional/#{m[1]}_test.rb" }
+  # watch(%r{^app/helpers/(.*)\.rb$})     { |m| "test/helpers/#{m[1]}_test.rb" }
+  # watch(%r{^app/models/(.*)\.rb$})      { |m| "test/unit/#{m[1]}_test.rb" }
+end

data/lib/pinot/client.rb

@@ -0,0 +1,81 @@
+module Pinot
+  class Client
+    attr_reader :host, :port, :controller_host, :controller_port, :protocol
+
+    def initialize(host:, port:, controller_port:, controller_host: nil, protocol: :http)
+      @host = host
+      @port = port
+      @controller_port = controller_port
+      @controller_host = controller_host || host
+      @protocol = protocol
+    end
+
+    def execute(sql)
+      Response.new(JSON.parse(http.post(query_sql_uri, json: {sql: sql})))
+    end
+
+    def schema(name)
+      url = "#{controller_uri}/schemas/#{name}"
+      response = http.get(url)
+      JSON.parse(response)
+    end
+
+    def delete_segments(name, type: :offline)
+      type = type.to_s.upcase
+      url = "#{controller_uri}/segments/#{name}?type=#{type}"
+      response = http.delete(url)
+      JSON.parse(response)
+    end
+
+    def create_table(schema)
+      url = "#{controller_uri}/tables"
+      response = http.post(url, body: schema)
+      JSON.parse(response)
+    end
+
+    def delete_table(name, type: :offline)
+      type = type.to_s.downcase
+      url = "#{controller_uri}/tables/#{name}?type=#{type}"
+      response = http.delete(url)
+      JSON.parse(response)
+    end
+
+    def ingest_json(file, table:)
+      url = "#{controller_uri}/ingestFromFile?tableNameWithType=#{table}&batchConfigMapStr=%7B%22inputFormat%22%3A%22json%22%7D"
+      content_type = "multipart/form-data"
+      response = HTTPX.post(
+        url,
+        form: {
+          file: {
+            filename: File.basename(file.path),
+            content_type: content_type,
+            body: file.read
+          }
+        }
+      )
+      JSON.parse(response)
+    end
+
+    def create_schema(schema, override: true, force: false)
+      url = "#{controller_uri}/schemas?override=#{override}&force=#{force}"
+      response = http.post(url, body: schema)
+      JSON.parse(response)
+    end
+
+    def http(content_type: "application/json")
+      HTTPX.with(headers: {"Content-Type" => content_type})
+    end
+
+    def uri
+      "#{protocol}://#{host}:#{port}"
+    end
+
+    def controller_uri
+      "#{protocol}://#{controller_host}:#{controller_port}"
+    end
+
+    def query_sql_uri
+      "#{uri}/query/sql"
+    end
+  end
+end

data/lib/pinot/response.rb

@@ -0,0 +1,32 @@
+module Pinot
+  class Response
+    include Enumerable
+
+    def initialize(payload)
+      @payload = payload
+    end
+
+    def rows
+      @payload.dig("resultTable", "rows")
+    end
+
+    def columns
+      names = @payload.dig("resultTable", "dataSchema", "columnNames")
+      types = @payload.dig("resultTable", "dataSchema", "columnDataTypes")
+      return {} if @payload["exceptions"].any?
+      ix = 0
+      names ||= []
+      names.map do |name|
+        ret = [name, types[ix]]
+        ix += 1
+        ret
+      end.to_h
+    end
+
+    def each(&block)
+      rows.each do |row|
+        block.call(row)
+      end
+    end
+  end
+end

data/lib/pinot/version.rb

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

data/lib/pinot.rb

@@ -1,6 +1,10 @@
 # frozen_string_literal: true
 
+require "httpx"
+
 require_relative "pinot/version"
+require_relative "pinot/response"
+require_relative "pinot/client"
 
 module Pinot
   class Error < StandardError; end

metadata

@@ -1,15 +1,71 @@
 --- !ruby/object:Gem::Specification
 name: pinot
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Celso Fernandes
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-01-16 00:00:00.000000000 Z
-dependencies: []
+date: 2024-05-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: httpx
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: guard
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: guard-minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: minitest-reporters
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Client for Apache Pinot
 email:
 - celso.fernandes@clickfunnels.com
@@ -20,11 +76,13 @@ files:
 - ".standard.yml"
 - CHANGELOG.md
 - CODE_OF_CONDUCT.md
+- Guardfile
 - README.md
 - Rakefile
 - lib/pinot.rb
+- lib/pinot/client.rb
+- lib/pinot/response.rb
 - lib/pinot/version.rb
-- pinot.gemspec
 - sig/pinot.rbs
 homepage: https://github.com/fernandes/pinot-ruby
 licenses: []
@@ -47,7 +105,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.21
+rubygems_version: 3.5.3
 signing_key:
 specification_version: 4
 summary: Client for Apache Pinot

data/pinot.gemspec

@@ -1,37 +0,0 @@
-# frozen_string_literal: true
-
-require_relative "lib/pinot/version"
-
-Gem::Specification.new do |spec|
-  spec.name = "pinot"
-  spec.version = Pinot::VERSION
-  spec.authors = ["Celso Fernandes"]
-  spec.email = ["celso.fernandes@clickfunnels.com"]
-
-  spec.summary = "Client for Apache Pinot"
-  spec.description = "Client for Apache Pinot"
-  spec.homepage = "https://github.com/fernandes/pinot-ruby"
-  spec.required_ruby_version = ">= 2.6.0"
-
-  spec.metadata["homepage_uri"] = spec.homepage
-  spec.metadata["source_code_uri"] = "https://github.com/fernandes/pinot"
-  spec.metadata["changelog_uri"] = "https://github.com/fernandes/pinot/blob/main/CHANGELOG.md"
-
-  # Specify which files should be added to the gem when it is released.
-  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
-  spec.files = Dir.chdir(__dir__) do
-    `git ls-files -z`.split("\x0").reject do |f|
-      (File.expand_path(f) == __FILE__) ||
-        f.start_with?(*%w[bin/ test/ spec/ features/ .git .circleci appveyor Gemfile])
-    end
-  end
-  spec.bindir = "exe"
-  spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
-  spec.require_paths = ["lib"]
-
-  # Uncomment to register a new dependency of your gem
-  # spec.add_dependency "example-gem", "~> 1.0"
-
-  # For more information and examples about making a new gem, check out our
-  # guide at: https://bundler.io/guides/creating_gem.html
-end