elastic-esql 0.2.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 99e718442d20298997082926d68a0a81fb7fa4caca2469438b1661f21134ea9f
-  data.tar.gz: 6baf28b918cfead958e96b59dc2062e02d5702c20b232cfe7fdaf6bdb9aa5285
+  metadata.gz: f7837c640aa915808d6c720d4cd53bdcfe3210fbff0a5e9b900a0c0adcc3ed7a
+  data.tar.gz: 6f3cc673984d2968b098a36137a44780e279be68c3ec04d37737044e29cf1af5
 SHA512:
-  metadata.gz: e4dffbd00b997031a36c27a7edf0640e8f726e1d719ba35a98d37a3f15235b3c92a2a8e38ee5159557a97dc775177e0ba08ef4df35cfed1921cc53b682fe129e
-  data.tar.gz: e1d9bd655f5d34016da2caf6a976c74dc9f57244782cbfc6700a8993f8dbbce37722c1cafe978c99a60f5b88daa6e99e1ae9015ed3347ca65d59f362d15f5dcc
+  metadata.gz: 90fbc7b576d768c0d2282d87e2ae0bcbd7c8f82ba19a910edfd084fa15cb5ef1b86423fe4eae5ddd07c0910ab0dcde4a6e67516147b54f5102a093c018d26237
+  data.tar.gz: a535a2fa6d19d8a5bb41ac4c70e27f66559699fe07073383ee7a3af295416d0df8a1f47e7a7ff0a1fbe20d0c6dbb315e6842e4fbf111351de2ee54d4bd774a56

data/lib/elastic/esql.rb

@@ -15,50 +15,44 @@
 # specific language governing permissions and limitations
 # under the License.
 
+require_relative 'branch'
 require_relative 'change_point'
 require_relative 'custom'
 require_relative 'dissect'
 require_relative 'drop'
 require_relative 'enrich'
 require_relative 'eval'
+require_relative 'functions'
+require_relative 'fork'
+require_relative 'fuse'
 require_relative 'grok'
 require_relative 'keep'
-require_relative 'limit'
 require_relative 'lookup_join'
 require_relative 'metadata'
+require_relative 'queryable'
 require_relative 'rename'
 require_relative 'row'
 require_relative 'show'
-require_relative 'sort'
-require_relative 'where'
+require_relative 'stats'
+require_relative 'ts'
+require_relative 'util'
 
 module Elastic
   # @example
   #    Elastic::ESQL.from('sample_data').sort_descending('@timestamp').limit(3).to_s
   #    # => FROM 'sample_data' | SORT @timestamp desc | LIMIT 3
   class ESQL
-    include ChangePoint
-    include Custom
-    include Dissect
-    include Drop
-    include Eval
-    include Grok
-    include Keep
-    include Limit
-    include LookupJoin
-    include Metadata
-    include Rename
-    include Row
-    include Show
-    include Sort
-    include Where
-    SOURCE_COMMANDS = [:from, :row, :show].freeze
+    [
+      ChangePoint, Custom, Dissect, Drop, Eval, Fork, Fuse, Grok, Keep, LookupJoin, Metadata,
+      Queryable, Rename, Row, Show, Stats, TS, Util
+    ].each { |m| include m }
+
+    SOURCE_COMMANDS = [:from, :row, :show, :ts].freeze
 
     def initialize
       @query = {}
       @custom = []
       @metadata = []
-      @lookup_joins = []
     end
 
     # Function to build the ES|QL formatted query and return it as a String.
@@ -67,15 +61,14 @@ module Elastic
     def query
       raise ArgumentError, 'No source command found' unless source_command_present?
 
-      @query[:enrich] = @enriches.join('| ') if @enriches
+      @query[:enrich] = @enriches.map(&:to_query).join('| ') if @enriches
       string_query = build_string_query
-      string_query.concat(build_lookup_joins) unless @lookup_joins.empty?
       string_query.concat(" #{@custom.join(' ')}") unless @custom.empty?
       string_query
     end
 
-    # Creates a new Enrich object to chain with +on+ and +with+. If other methods are chained to the
-    # Enrich object, it returns calls it upon the ESQL object that instantiated it, and returns it.
+    # Creates a new Enrich object to chain with +on+ and +with+. If other method is chained to the
+    # Enrich object, it calls it upon the ESQL object that instantiated it, and returns it.
     # @return [Elastic::Enrich]
     def enrich(policy)
       @enriches ||= []
@@ -109,6 +102,10 @@ module Elastic
       new.row(*params)
     end
 
+    def self.ts(*params)
+      new.ts(*params)
+    end
+
     # Instance method to allow to update +from+ with +esql.from('different_source')+.
     # @param [String] index_pattern A list of indices, data streams or aliases. Supports wildcards and date math.
     def from(index_pattern)
@@ -122,6 +119,26 @@ module Elastic
       query
     end
 
+    # rubocop:disable Naming/MethodName, Naming/BinaryOperatorParameterName
+    def self.🐔(message)
+      "ROW CHICKEN(\"#{message}\")"
+    end
+
+    def 🐔(message)
+      self.class.🐔(message)
+    end
+
+    alias chicken 🐔
+
+    class << self
+      alias chicken 🐔
+    end
+    # rubocop:enable Naming/MethodName, Naming/BinaryOperatorParameterName
+
+    def self.branch
+      Branch.new
+    end
+
     private
 
     # Function for eval, row, and other functions that have one or more columns with values specified
@@ -152,34 +169,5 @@ module Elastic
 
       false
     end
-
-    # Helper method to return a copy of the object when functions are called without `!`, so the
-    # object is not mutated.
-    def method_copy(name, *params)
-      esql = clone
-      esql.instance_variable_set('@query', esql.instance_variable_get('@query').clone)
-      esql.send("#{name}!", *params)
-      esql
-    end
-
-    # Helper to build the LOOKUP JOIN part of the query.
-    def build_lookup_joins
-      joins = @lookup_joins.map { |a| a.map { |k, v| "LOOKUP JOIN #{k} ON #{v}" } }.flatten.join(' | ')
-      " | #{joins}"
-    end
-
-    # Helper to build the String for the simpler functions.
-    # These are of the form 'key.upcase value' like 'DROP value'
-    # If metadata has been set, it needs to be added to FROM. There's a possibility there'll be more
-    # special cases like this in the future, they can be added here.
-    def build_string_query
-      @query.map do |k, v|
-        if k == :from && !@metadata.empty?
-          "#{k.upcase} #{v} METADATA #{@metadata.join(', ')}"
-        else
-          "#{k.upcase} #{v}"
-        end
-      end.join(' | ')
-    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: elastic-esql
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Fernando Briano
@@ -104,7 +104,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.7.1
+rubygems_version: 4.0.3
 specification_version: 4
 summary: Elastic ES|QL Query builder
 test_files: []