elastic-esql 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d509f46ee96d9ce934b04bdf01c14589e77a0e7130bbf86386e3050668865f10
-  data.tar.gz: 1631dc08bc5cb8a83175a1e3ab226d27093906751babf9fe220f0d61b54e669b
+  metadata.gz: 99e718442d20298997082926d68a0a81fb7fa4caca2469438b1661f21134ea9f
+  data.tar.gz: 6baf28b918cfead958e96b59dc2062e02d5702c20b232cfe7fdaf6bdb9aa5285
 SHA512:
-  metadata.gz: fd106b0e92d4e951b3b6f685e926f538e79d727634e3f805554cbf021b8c4e92c8eb12ae2be25372e0fcfaf669ef11d372ba6c2bc53a253c2fe177e33adf14e9
-  data.tar.gz: 490df5f4fc2fae7d0505ce7a411db8a85fa32f93e6609844bbe6c0270638e7c17e58c0ba815549de28acb6fbc09cbe07fd5bc47c4604ca68408ff778a4b6594e
+  metadata.gz: e4dffbd00b997031a36c27a7edf0640e8f726e1d719ba35a98d37a3f15235b3c92a2a8e38ee5159557a97dc775177e0ba08ef4df35cfed1921cc53b682fe129e
+  data.tar.gz: e1d9bd655f5d34016da2caf6a976c74dc9f57244782cbfc6700a8993f8dbbce37722c1cafe978c99a60f5b88daa6e99e1ae9015ed3347ca65d59f362d15f5dcc

data/lib/elastic/esql.rb

@@ -22,8 +22,10 @@ require_relative 'drop'
 require_relative 'enrich'
 require_relative 'eval'
 require_relative 'grok'
-require_relative 'limit'
 require_relative 'keep'
+require_relative 'limit'
+require_relative 'lookup_join'
+require_relative 'metadata'
 require_relative 'rename'
 require_relative 'row'
 require_relative 'show'
@@ -43,6 +45,8 @@ module Elastic
     include Grok
     include Keep
     include Limit
+    include LookupJoin
+    include Metadata
     include Rename
     include Row
     include Show
@@ -53,6 +57,8 @@ module Elastic
     def initialize
       @query = {}
       @custom = []
+      @metadata = []
+      @lookup_joins = []
     end
 
     # Function to build the ES|QL formatted query and return it as a String.
@@ -62,10 +68,8 @@ module Elastic
       raise ArgumentError, 'No source command found' unless source_command_present?
 
       @query[:enrich] = @enriches.join('| ') if @enriches
-      string_query = @query.map do |k, v|
-        "#{k.upcase} #{v}"
-      end.join(' | ')
-
+      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
@@ -134,7 +138,7 @@ module Elastic
     # Error raised when a function expects a Hash and something else is passed in, with explanation
     def raise_hash_error(name)
       raise ArgumentError, "#{name.to_s.upcase} needs a Hash as a parameter where the keys are the " \
-                          'column names and the value is the function or expression to calculate.'
+                           'column names and the value is the function or expression to calculate.'
     end
 
     # Used when building the query from hash params function
@@ -157,5 +161,25 @@ module Elastic
       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.1.0
+  version: 0.2.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.6.9
+rubygems_version: 3.7.1
 specification_version: 4
 summary: Elastic ES|QL Query builder
 test_files: []