class Blazer::Adapters::OpensearchAdapter

Public Instance Methods

preview_statement() click to toggle source
# File lib/blazer/adapters/opensearch_adapter.rb, line 37
def preview_statement
  "SELECT * FROM `{table}` LIMIT 10"
end
quoting() click to toggle source
# File lib/blazer/adapters/opensearch_adapter.rb, line 41
def quoting
  # unknown
end
run_statement(statement, comment) click to toggle source
# File lib/blazer/adapters/opensearch_adapter.rb, line 4
def run_statement(statement, comment)
  columns = []
  rows = []
  error = nil

  begin
    response = client.transport.perform_request("POST", "_plugins/_sql", {}, {query: "#{statement} /*#{comment}*/"}).body
    columns = response["schema"].map { |v| v["name"] }
    # TODO typecast more types
    # https://github.com/opensearch-project/sql/blob/main/docs/user/general/datatypes.rst
    date_indexes = response["schema"].each_index.select { |i| response["schema"][i]["type"] == "timestamp" }
    if columns.any?
      rows = response["datarows"]
      utc = ActiveSupport::TimeZone["Etc/UTC"]
      date_indexes.each do |i|
        rows.each do |row|
          row[i] &&= utc.parse(row[i])
        end
      end
    end
  rescue => e
    error = e.message
  end

  [columns, rows, error]
end
tables() click to toggle source
# File lib/blazer/adapters/opensearch_adapter.rb, line 31
def tables
  indices = client.cat.indices(format: "json").map { |v| v["index"] }
  aliases = client.cat.aliases(format: "json").map { |v| v["alias"] }
  (indices + aliases).uniq.sort
end

Protected Instance Methods

client() click to toggle source
# File lib/blazer/adapters/opensearch_adapter.rb, line 47
def client
  @client ||= OpenSearch::Client.new(url: settings["url"])
end