Merge pull request #68 from tomarv2/add_lib

Add lib

Author: tomarv2

cluster.tf

@@ -3,6 +3,23 @@ locals {
   cluster_name      = var.cluster_name != null ? var.cluster_name : "${var.teamid}-${var.prjid} (Terraform managed)"
 }
 
+resource "databricks_library" "maven" {
+  for_each = try(var.libraries.maven, null) != null ? var.libraries.maven : {}
+
+  cluster_id = databricks_cluster.cluster[local.cluster_name].id
+  maven {
+    coordinates = each.key
+    exclusions = try(each.value.exclusion, [])
+  }
+}
+
+resource "databricks_library" "python_wheel" {
+  for_each = try(var.libraries.python_wheel, null) != null ? var.libraries.python_wheel: {}
+
+  cluster_id = databricks_cluster.cluster[local.cluster_name].id
+  whl        = each.key
+}
+
 resource "databricks_cluster" "cluster" {
   for_each = var.deploy_cluster == true ? toset([local.cluster_name]) : toset([])
 

examples/clusters/1_single_node_cluster/main.tf

@@ -19,6 +19,14 @@ module "databricks_workspace_management" {
   # ------------------------------------------------
   deploy_cluster = true
   fixed_value    = 0
+
+  libraries = {
+    maven = {
+      "com.microsoft.azure:azure-eventhubs-spark_2.12:2.3.21" = {}
+    }
+    python_wheel = {}
+  }
+
   # ------------------------------------------------
   # Do not change the teamid, prjid once set.
   teamid = var.teamid

examples/jobs/4_job_on_existing_cluster_with_new_notebooks/main.tf

@@ -15,11 +15,12 @@ terraform {
 
 module "databricks_workspace_management" {
   source = "../../../"
+
   # ------------------------------------------------
   # JOB
   # ------------------------------------------------
   deploy_jobs               = true
-  cluster_id                = "0606-193951-30y13dr5"
+  cluster_id                = "0907-052446-bike152"
   fixed_value               = 1
   retry_on_timeout          = false
   max_retries               = 3

variables.tf

@@ -358,3 +358,9 @@ variable "custom_tags" {
   description = "Extra custom tags"
   default     = null
 }
+
+variable "libraries" {
+  type = map(any)
+  description = "Installs a library on databricks_cluster"
+  default = {}
+}