azure-kusto-ruby 0.0.5

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 825e171c28ef4a5b0f5a133cbffa427623e5e28a71519eda8d523f373ed76c58
+  data.tar.gz: 415378a81832dc2369b495da8302f85b1bb1b941704fe72df18020e2f56ae84c
+SHA512:
+  metadata.gz: e1fd7e66bea05a43e7ed11db5c83aed4e742cc16aa8e9cee9c4f99cd39e79e1a9203fe266ee65f9ac77c143a2738e35145b1f299805308f28f8611384c94f226
+  data.tar.gz: f0abd0dbf4138e2bd681738b094b2a2e869243dfc8aaf8da47ad980de05e1eca6be8d1fd6a91bbb765f58c8b7f7a39e0586c418c02c596b349579da98e5c453d

data/lib/kusto/rest/context/context.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require "faraday"
+require_relative "../result/data_set"
+
+module Kusto
+  module Rest
+    class Context
+      attr_reader :path
+
+      class Error < StandardError; end
+
+      def initialize(url:, token:)
+        @url = url
+        @token = token
+        @connector = Faraday.new(url:)
+      end
+
+      def post(query)
+        response = @connector.post(@path) do |req|
+          req.headers["Content-Type"] = "application/json"
+          req.headers["Authorization"] = "Bearer #{@token}"
+          req.body = query.request_body.to_json
+        end
+
+        unless response.success?
+          raise Error, "Request failed with status #{response.status} - #{response.reason_phrase} #{response.body}"
+        end
+
+        if response.body.empty? || response.body.nil?
+          raise Error, "Request with status #{response.status} returned an empty response body"
+        end
+
+        Kusto::Rest::DataSet.new(response.body)
+      end
+
+      def endpoint
+        @connector.url_prefix.to_s + @path
+      end
+    end
+  end
+end

data/lib/kusto/rest/context/management_context.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+require_relative "context"
+
+module Kusto
+  module Rest
+    class ManagementContext < Context
+      attr_reader :path
+
+      class Error < StandardError; end
+
+      def initialize(url:, token:)
+        super(url:, token:)
+        @path = "v1/rest/mgmt"
+      end
+    end
+  end
+end

data/lib/kusto/rest/context/query_context.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+require_relative "context"
+
+module Kusto
+  module Rest
+    class QueryContext < Context
+      attr_reader :path
+
+      class Error < StandardError; end
+
+      def initialize(url:, token:)
+        super(url:, token:)
+        @path = "v2/rest/query"
+      end
+    end
+  end
+end

data/lib/kusto/rest/context.rb

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+require_relative "context/query_context"
+require_relative "context/management_context"

data/lib/kusto/rest/query/data.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require "securerandom"
+require_relative "query"
+
+module Kusto
+  module Rest
+    class DataQuery < Query
+      def initialize(db:, statement:, parameters: {}, options: {}, request_id: nil)
+        default_options = { "results_progressive_enabled": false }
+        @options = default_options.merge!(options)
+        @parameters = parameters
+        @db = db
+        super(statement:, request_id:)
+      end
+
+      def request_body
+        @request_body ||= {
+          db: @db,
+          properties:,
+          csl: @statement
+        }
+      end
+    end
+  end
+end

data/lib/kusto/rest/query/query.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+require "securerandom"
+
+module Kusto
+  module Rest
+    class Query
+      def initialize(statement:, options: {}, parameters: {}, request_id: nil)
+        @options = options
+        @parameters = parameters
+        @request_id = request_id || get_request_id
+        @statement = statement
+      end
+
+      def properties
+        @properties ||= {
+          "Options": @options,
+          "Parameters": @parameters,
+          "ClientRequestId": @request_id
+        }
+      end
+
+      def request_body
+        @request_body ||= {
+          properties:,
+          csl: @statement
+        }
+      end
+
+      private
+
+      def get_request_id
+        @request_id ||= "#{self.class.name}::#{SecureRandom.uuid}"
+      end
+    end
+  end
+end

data/lib/kusto/rest/query.rb

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+require_relative "query/query"
+require_relative "query/data"

data/lib/kusto/rest/result/data_frame.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require "date"
+
+module Kusto
+  module Rest
+    class DataFrame
+      def initialize(data)
+        @data = data
+      end
+
+      def type
+        @data["FrameType"]
+      end
+
+      def kind
+        @data["TableKind"]
+      end
+
+      def name
+        @data["TableName"]
+      end
+
+      def cast(value, type)
+        case type
+        when "datetime"
+          value ? DateTime.parse(value) : nil
+        else
+          value
+        end
+      end
+    end
+  end
+end

data/lib/kusto/rest/result/data_set.rb

@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+
+require "json"
+require_relative "data_set_completion"
+require_relative "data_set_header"
+require_relative "data_table"
+
+module Kusto
+  module Rest
+    class DataSet
+      attr_reader :data
+
+      def initialize(response)
+        @data = JSON.parse(response)
+      end
+
+      def rows
+        primary_result.rows
+      end
+
+      def primary_result
+        @primary_result ||= frames.find_all do |frame|
+          frame.type == "DataTable" &&
+            frame.kind == "PrimaryResult" &&
+            frame.name == "PrimaryResult"
+        end.first
+      end
+
+      def header
+        @header ||= frames.find_all { |frame| frame.type == "DataSetHeader" }.first
+      end
+
+      def completion
+        @completion ||= frames.find_all do |frame|
+          frame.type == "DataSetCompletion"
+        end.first
+      end
+
+      def self.parse_frame(frame)
+        case frame["FrameType"]
+        when "DataSetHeader"
+          Kusto::Rest::DataSetHeader.new(frame)
+        when "DataTable"
+          Kusto::Rest::DataTable.new(frame)
+        when "DataSetCompletion"
+          Kusto::Rest::DataSetCompletion.new(frame)
+        else
+          Kusto::Result::DataFrame.new(frame)
+        end
+      end
+
+      private
+
+      def frames
+        @frames ||= @data.map do |frame|
+          self.class.parse_frame(frame)
+        end
+      end
+    end
+  end
+end

data/lib/kusto/rest/result/data_set_completion.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require_relative "data_frame"
+
+module Kusto
+  module Rest
+    class DataSetCompletion < DataFrame
+      def errors?
+        @data["HasErrors"]
+      end
+
+      def cancelled?
+        @data["Cancelled"]
+      end
+    end
+  end
+end

data/lib/kusto/rest/result/data_set_header.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+require_relative "data_frame"
+
+module Kusto
+  module Rest
+    class DataSetHeader < DataFrame
+      def progressive?
+        @data["IsProgressive"]
+      end
+
+      def fragmented?
+        @data["IsFragmented"]
+      end
+
+      def version
+        @data["Version"]
+      end
+
+      def error_reporting_placement
+        @data["ErrorReportingPlacement"]
+      end
+    end
+  end
+end

data/lib/kusto/rest/result/data_table.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+require_relative "data_frame"
+
+module Kusto
+  module Rest
+    class DataTable < DataFrame
+      def rows
+        columns = @data["Columns"]
+        row_data = @data["Rows"]
+        row_data.map do |row|
+          columns.each_with_index.to_h do |column, index|
+            [column["ColumnName"], cast(row[index], column["ColumnType"])]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kusto/rest/result.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+require_relative "result/data_frame"
+require_relative "result/data_set"
+require_relative "result/data_set_completion"
+require_relative "result/data_set_header"
+require_relative "result/data_table"

data/lib/kusto/rest.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module Kusto
+  module Rest
+    require_relative "rest/context"
+    require_relative "rest/query"
+    require_relative "rest/result"
+  end
+end

data/lib/version.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module Azure
+  module Kusto
+    module Ruby
+      VERSION = "0.0.5"
+    end
+  end
+end

metadata

@@ -0,0 +1,59 @@
+--- !ruby/object:Gem::Specification
+name: azure-kusto-ruby
+version: !ruby/object:Gem::Version
+  version: 0.0.5
+platform: ruby
+authors:
+- Carlton Brown
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2024-05-17 00:00:00.000000000 Z
+dependencies: []
+description: 
+email: 
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/kusto/rest.rb
+- lib/kusto/rest/context.rb
+- lib/kusto/rest/context/context.rb
+- lib/kusto/rest/context/management_context.rb
+- lib/kusto/rest/context/query_context.rb
+- lib/kusto/rest/query.rb
+- lib/kusto/rest/query/data.rb
+- lib/kusto/rest/query/query.rb
+- lib/kusto/rest/result.rb
+- lib/kusto/rest/result/data_frame.rb
+- lib/kusto/rest/result/data_set.rb
+- lib/kusto/rest/result/data_set_completion.rb
+- lib/kusto/rest/result/data_set_header.rb
+- lib/kusto/rest/result/data_table.rb
+- lib/version.rb
+homepage: https://github.com/carltonbrown/azure-kusto-ruby
+licenses:
+- MIT
+metadata:
+  github_repo: https://github.com/carltonbrown/azure-kusto-ruby
+  rubygems_mfa_required: 'true'
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.1.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.3
+signing_key: 
+specification_version: 4
+summary: A client for the Kusto REST API
+test_files: []