databricks 2.0.0 → 2.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 06a826e00c327ae1b293c0da9f57632ec9103e128f253fcf2f8ce1229a64074c
-  data.tar.gz: 6830f2c2fc4fe380b92856f28d827e8573ccfbb78f86280d1d11269ccf602fbb
+  metadata.gz: 5180797f8cb0ab700566b70bbb20fe1e54a46241162a2efc7c128846998aaaf4
+  data.tar.gz: 910d6984b316a2b09b90a94ae267e157aadc7197aa5f897bd50b2a5b21e21da2
 SHA512:
-  metadata.gz: 601a0e36f9fb7bc79c9d4f42ecf9f3ff32eb1a531ce9d706f38b6731e3ef373da32b7cd2c404414e4747f1d2ec2b9833e0aa913ac8e1dac84f7e72fb8113d151
-  data.tar.gz: 7ef05a8b17ff9167e346cd129fbda117fd75289e9ac77d1b0679340b56bddcfec17d66c437a8c2fc95fa8278ee653a439936b408d1b7dacfde41aa69c85f8a36
+  metadata.gz: da5dcd7a5f419079a845c43fe2c4f3d073351abad9c5f8313d826124389c5d801bf528622ef06bdb81ca58d41bdd341d0487dfa0740632d1a50162548b263324
+  data.tar.gz: 8d850d1cabacf5c325ee094560473055c14da9278cc5d4412b3a85ae2fbd914c01ede639458db3ffab9ba3fb8001e174bbc326a7aaf2b86421141d3d3b24d065

data/lib/databricks/resources/clusters.rb

@@ -9,13 +9,15 @@ module Databricks
       # List clusters
       #
       # Result::
-      # * Array<Hash>: List of clusters information
+      # * Array<Cluster>: List of clusters
       def list
         (get_json('clusters/list')['clusters'] || []).map { |properties| new_resource(:cluster, properties) }
       end
 
       # Get a cluster based on its cluster_id
       #
+      # Parameters::
+      # * *cluster_id* (String): The cluster id to get
       # Result::
       # * Cluster: The cluster
       def get(cluster_id)
@@ -26,6 +28,8 @@ module Databricks
       #
       # Parameters::
       # * *properties* (Hash<Symbol,Object>): Properties to create the cluster
+      # Result::
+      # * Cluster: The new cluster created
       def create(**properties)
         cluster = new_resource(:cluster, post_json('clusters/create', properties))
         cluster.add_properties(properties)

data/lib/databricks/resources/instance_pool.rb

@@ -0,0 +1,27 @@
+module Databricks
+
+  module Resources
+
+    class InstancePool < Resource
+
+      # Edit properties of this instance pool.
+      #
+      # Parameters::
+      # * *properties* (Hash<Symbol,Object>): Properties of this cluster
+      def edit(**properties)
+        # Make sure we don't change its ID
+        properties[:instance_pool_id] = instance_pool_id
+        post_json('instance-pools/edit', properties)
+        add_properties(properties)
+      end
+
+      # Delete this instance pool
+      def delete
+        post_json('instance-pools/delete', { instance_pool_id: instance_pool_id })
+      end
+
+    end
+
+  end
+
+end

data/lib/databricks/resources/instance_pools.rb

@@ -0,0 +1,43 @@
+module Databricks
+
+  module Resources
+
+    # Provide the Instance Pools API
+    # cf. https://docs.databricks.com/dev-tools/api/latest/instance-pools.html
+    class InstancePools < Resource
+
+      # List instance pools
+      #
+      # Result::
+      # * Array<InstancePool>: List of instance pools
+      def list
+        (get_json('instance-pools/list')['instance_pools'] || []).map { |properties| new_resource(:instance_pool, properties) }
+      end
+
+      # Get an instance pool based on its instance_pool_id
+      #
+      # Parameters::
+      # * *instance_pool_id* (String): The instance pool id to get
+      # Result::
+      # * InstancePool: The instance pool
+      def get(instance_pool_id)
+        new_resource(:instance_pool, get_json('instance-pools/get', { instance_pool_id: instance_pool_id }))
+      end
+
+      # Create a new instance pool.
+      #
+      # Parameters::
+      # * *properties* (Hash<Symbol,Object>): Properties to create the instance pool
+      # Result::
+      # * InstancePool: The new instance pool created
+      def create(**properties)
+        instance_pool = new_resource(:instance_pool, post_json('instance-pools/create', properties))
+        instance_pool.add_properties(properties)
+        instance_pool
+      end
+
+    end
+
+  end
+
+end

data/lib/databricks/resources/jobs.rb

@@ -9,7 +9,7 @@ module Databricks
       # List jobs
       #
       # Result::
-      # * Array<Hash>: List of jobs information
+      # * Array<Job>: List of jobs information
       def list
         (get_json('jobs/list')['jobs'] || []).map { |properties| new_resource(:job, properties) }
       end
@@ -18,6 +18,8 @@ module Databricks
       #
       # Parameters::
       # * *properties* (Hash<Symbol,Object>): Properties to create the job
+      # Result::
+      # * Job: The new job created
       def create(**properties)
         job = new_resource(:job, post_json('jobs/create', properties))
         job.add_properties(properties)

data/lib/databricks/resources/root.rb

@@ -11,6 +11,7 @@ module Databricks
       sub_resources %i[
         clusters
         dbfs
+        instance_pools
         jobs
       ]
 

data/lib/databricks/version.rb

@@ -1,5 +1,5 @@
 module Databricks
 
-  VERSION = '2.0.0'
+  VERSION = '2.1.0'
 
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: databricks
 version: !ruby/object:Gem::Version
-  version: 2.0.0
+  version: 2.1.0
 platform: ruby
 authors:
 - Muriel Salvan
@@ -80,6 +80,8 @@ files:
 - lib/databricks/resources/clusters.rb
 - lib/databricks/resources/dbfs.rb
 - lib/databricks/resources/file.rb
+- lib/databricks/resources/instance_pool.rb
+- lib/databricks/resources/instance_pools.rb
 - lib/databricks/resources/job.rb
 - lib/databricks/resources/jobs.rb
 - lib/databricks/resources/root.rb