trino-client 1.0.2 → 2.0.1

data/publish.rb

@@ -1,14 +0,0 @@
-#!/usr/bin/env ruby
-require File.expand_path 'lib/trino/client/version', File.dirname(__FILE__)
-
-def run(cmd)
-  puts cmd
-  system cmd
-end
-
-run("gem build trino-client.gemspec")
-run("gem push trino-client-#{Trino::Client::VERSION}.gem")
-
-run("gem build trino-client-ruby/trino-client-ruby.gemspec")
-run("gem push trino-client-ruby/trino-client-ruby-#{Trino::Client::VERSION}.gem")
-

data/release.rb

@@ -1,56 +0,0 @@
-#!/usr/bin/env ruby
-
-require 'fileutils'
-
-PREFIX = 'https://github.com/treasure-data/trino-client-ruby'
-RELEASE_NOTES_FILE = "ChangeLog.md"
-
-last_tag = `git describe --tags --abbrev=0`.chomp
-last_version = last_tag.sub("v", "")
-puts "last version: #{last_version}"
-
-print "next version? "
-next_version = STDIN.gets.chomp
-
-abort("Can't use empty version string") if next_version.empty?
-
-logs = `git log #{last_tag}..HEAD --pretty=format:'%h %s'`
-# Add links to GitHub issues
-logs = logs.gsub(/\#([0-9]+)/, "[#\\1](#{PREFIX}/issues/\\1)")
-
-new_release_notes = []
-new_release_notes <<= "\#\# #{next_version}\n"
-new_release_notes <<= logs.split(/\n/)
-  .reject{|line| line.include?("#{last_version} release notes")}
-  .map{|x|
-    rev = x[0..6]
-    "- #{x[8..-1]} [[#{rev}](#{PREFIX}/commit/#{rev})]\n"
-  }
-
-release_notes = []
-notes = File.readlines(RELEASE_NOTES_FILE)
-
-release_notes <<= notes[0..1]
-release_notes <<= new_release_notes
-release_notes <<= "\n"
-release_notes <<= notes[2..-1]
-
-TMP_RELEASE_NOTES_FILE = "#{RELEASE_NOTES_FILE}.tmp"
-File.delete(TMP_RELEASE_NOTES_FILE) if File.exists?(TMP_RELEASE_NOTES_FILE)
-File.write("#{TMP_RELEASE_NOTES_FILE}", release_notes.join)
-system("cat #{TMP_RELEASE_NOTES_FILE} | vim - -c ':f #{TMP_RELEASE_NOTES_FILE}' -c ':9'")
-
-abort("The release note file is not saved. Aborted") unless File.exists?(TMP_RELEASE_NOTES_FILE)
-
-def run(cmd)
-  puts cmd
-  system cmd
-end
-
-FileUtils.cp(TMP_RELEASE_NOTES_FILE, RELEASE_NOTES_FILE)
-File.delete(TMP_RELEASE_NOTES_FILE)
-
-# run "git commit #{RELEASE_NOTES_FILE} -m \"Add #{next_version} release notes\""
-# run "git tag v#{next_version}"
-# run "git push"
-# run "git push --tags"

data/spec/basic_query_spec.rb

@@ -1,82 +0,0 @@
-require 'spec_helper'
-
-describe Trino::Client::Client do
-  before(:all) do
-    WebMock.disable!
-    @cluster = TinyPresto::Cluster.new()
-    @container = @cluster.run
-    @client = Trino::Client.new(server: 'localhost:8080', catalog: 'memory', user: 'test-user', schema: 'default')
-    loop do
-      begin
-        @client.run('show schemas')
-        break
-      rescue StandardError => exception
-        puts "Waiting for cluster ready... #{exception}"
-        sleep(3)
-      end
-    end
-    puts 'Cluster is ready'
-  end
-
-  after(:all) do
-    @cluster.stop
-    WebMock.enable!
-  end
-
-  it 'show schemas' do
-    columns, rows = run_with_retry(@client, 'show schemas')
-    expect(columns.length).to be(1)
-    expect(rows.length).to be(2)
-  end
-
-  it 'ctas' do
-    expected = [[1, 'a'], [2, 'b']]
-    run_with_retry(@client, "create table ctas1 as select * from (values (1, 'a'), (2, 'b')) t(c1, c2)")
-    columns, rows = run_with_retry(@client, 'select * from ctas1')
-    expect(columns.map(&:name)).to match_array(%w[c1 c2])
-    expect(rows).to eq(expected)
-  end
-
-  it 'next_uri' do
-    @client.query('show schemas') do |q|
-      expect(q.next_uri).to start_with('http://localhost:8080/v1/statement/')
-    end
-  end
-
-  it 'advance' do
-    @client.query('show schemas') do |q|
-      expect(q.advance).to be(true)
-    end
-  end
-
-  it 'current query result' do
-    @client.query('show schemas') do |q|
-      expect(q.current_results.info_uri).to start_with('http://localhost:8080/ui/query.html')
-    end
-  end
-
-  it 'statement stats' do
-    @client.query('show schemas') do |q|
-      stats = q.current_results.stats
-      # Immediate subsequent request should get queued result
-      expect(stats.queued).to be(true)
-      expect(stats.scheduled).to be(false)
-    end
-  end
-
-  it 'partial cancel' do
-    @client.query('show schemas') do |q|
-      q.cancel
-      expect { q.query_info }.to raise_error(Trino::Client::TrinoHttpError, /Error 410 Gone/)
-    end
-  end
-
-  it 'row chunk' do
-    expected_schemas = %w[default information_schema]
-    @client.query('show schemas') do |q|
-      q.each_row do |r|
-        expect(expected_schemas).to include(r[0])
-      end
-    end
-  end
-end

data/spec/client_spec.rb

@@ -1,75 +0,0 @@
-require 'spec_helper'
-
-describe Trino::Client::Client do
-  let(:client) { Trino::Client.new({}) }
-
-  describe 'rehashes' do
-    let(:columns) do
-      [
-        Models::Column.new(name: 'animal', type: 'string'),
-        Models::Column.new(name: 'score', type: 'integer'),
-        Models::Column.new(name: 'name', type: 'string')
-      ]
-    end
-
-    it 'multiple rows' do
-      rows = [
-        ['dog', 1, 'Lassie'],
-        ['horse', 5, 'Mr. Ed'],
-        ['t-rex', 37, 'Doug']
-      ]
-      client.stub(:run).and_return([columns, rows])
-
-      rehashed = client.run_with_names('fake query')
-
-      rehashed.length.should == 3
-
-      rehashed[0]['animal'].should == 'dog'
-      rehashed[0]['score'].should == 1
-      rehashed[0]['name'].should == 'Lassie'
-
-      rehashed[0].values[0].should == 'dog'
-      rehashed[0].values[1].should == 1
-      rehashed[0].values[2].should == 'Lassie'
-
-      rehashed[1]['animal'].should == 'horse'
-      rehashed[1]['score'].should == 5
-      rehashed[1]['name'].should == 'Mr. Ed'
-
-      rehashed[1].values[0].should == 'horse'
-      rehashed[1].values[1].should == 5
-      rehashed[1].values[2].should == 'Mr. Ed'
-    end
-
-    it 'empty results' do
-      rows = []
-      client.stub(:run).and_return([columns, rows])
-
-      rehashed = client.run_with_names('fake query')
-
-      rehashed.length.should == 0
-    end
-
-    it 'handles too few result columns' do
-      rows = [['wrong', 'count']]
-      client.stub(:run).and_return([columns, rows])
-
-      client.run_with_names('fake query').should == [{
-        "animal" => "wrong",
-        "score" => "count",
-        "name" => nil,
-      }]
-    end
-
-    it 'handles too many result columns' do
-      rows = [['wrong', 'count', 'too', 'much', 'columns']]
-      client.stub(:run).and_return([columns, rows])
-
-      client.run_with_names('fake query').should == [{
-        "animal" => "wrong",
-        "score" => "count",
-        "name" => 'too',
-      }]
-    end
-  end
-end

data/spec/gzip_spec.rb

@@ -1,40 +0,0 @@
-require 'spec_helper'
-
-describe Trino::Client::Client do
-  before(:all) do
-    @spec_path = File.dirname(__FILE__)
-    WebMock.disable!
-    @cluster = TinyPresto::Cluster.new()
-    @container = @cluster.run
-    @client = Trino::Client.new(server: 'localhost:8080', catalog: 'tpch', user: 'test-user', schema: 'tiny', gzip: true, http_debug: true)
-    loop do
-      begin
-        # Make sure to all workers are available.
-        @client.run('show schemas')
-        break
-      rescue StandardError => exception
-        puts "Waiting for cluster ready... #{exception}"
-        sleep(3)
-      end
-    end
-    puts 'Cluster is ready'
-  end
-
-  after(:all) do
-    @cluster.stop
-    WebMock.enable!
-  end
-
-  it 'tpch q01 with gzip option' do
-    $stdout = StringIO.new
-    begin
-      q = File.read("#{@spec_path}/tpch/q01.sql")
-      columns, rows = run_with_retry(@client, q)
-      expect(columns.length).to be(10)
-      expect(rows.length).to be(4)
-      expect($stdout.string).to include ('content-encoding: "gzip"')
-    ensure
-      $stdout = STDOUT
-    end
-  end
-end

data/spec/model_spec.rb

@@ -1,35 +0,0 @@
-require 'spec_helper'
-
-describe Trino::Client::Models do
-  describe 'rehash of BlockedReason' do
-    h = {
-      "operatorId" => 0,
-      "planNodeId" => "47",
-      "operatorType" => "ScanFilterAndProjectOperator",
-      "addInputCalls" => 0,
-      "addInputWall" => "0.00ns",
-      "addInputCpu" => "0.00ns",
-      "addInputUser" => "0.00ns",
-      "inputDataSize" => "9.46MB",
-      "inputPositions" => 440674,
-      "getOutputCalls" => 734,
-      "getOutputWall" => "7.29s",
-      "getOutputCpu" => "0.00ns",
-      "getOutputUser" => "0.00ns",
-      "outputDataSize" => "36.99MB",
-      "outputPositions" => 440674,
-      "blockedWall" => "0.00ns",
-      "finishCalls" => 0,
-      "finishWall" => "0.00ns",
-      "finishCpu" => "0.00ns",
-      "finishUser" => "0.00ns",
-      "memoryReservation" => "0B",
-      "systemMemoryReservation" => "0b",
-      "blockedReason" => "WAITING_FOR_MEMORY",
-      "info" => {"k" => "v"}
-    }
-
-    stats = Models::OperatorStats.decode(h)
-    stats.blocked_reason.should == :waiting_for_memory
-  end
-end

data/spec/spec_helper.rb

@@ -1,42 +0,0 @@
-require 'bundler'
-
-begin
-  Bundler.setup(:default, :test)
-rescue Bundler::BundlerError => e
-  $stderr.puts e.message
-  $stderr.puts "Run `bundle install` to install missing gems"
-  exit e.status_code
-end
-
-require 'simplecov'
-SimpleCov.start
-
-require 'json'
-require 'webmock/rspec'
-
-require 'trino-client'
-include Trino::Client
-
-require 'tiny-presto'
-
-MAX_RETRY_COUNT = 5
-RETRYABLE_ERRORS = [
-  /No nodes available to run query/
-]
-
-def run_with_retry(client, sql)
-  i = 0
-  while i < MAX_RETRY_COUNT
-    begin
-      columns, rows = @client.run(sql)
-      return columns, rows
-    rescue Trino::Client::TrinoQueryError => e
-      if RETRYABLE_ERRORS.any? { |error| e.message =~ error }
-        sleep(i)
-        i += 1
-        next
-      end
-      raise "Fail to run query: #{e}"
-    end
-  end
-end