From fed46b7b136004cb9f22194d6d343e9a79ce2fa1 Mon Sep 17 00:00:00 2001
From: Andrei Hagiescu <andrei.hagiescu@databricks.com>
Date: Fri, 22 Jan 2021 13:00:36 -0500
Subject: [PATCH] add volume tags

---
 launcher/src/Instance.scala | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/launcher/src/Instance.scala b/launcher/src/Instance.scala
index 9ffe0be..38fe568 100644
--- a/launcher/src/Instance.scala
+++ b/launcher/src/Instance.scala
@@ -175,6 +175,17 @@ object Instance{
                 .build()
             }:_*
           )
+          .build(),
+        TagSpecification.builder()
+          .resourceType(ResourceType.VOLUME)
+          .tags(
+            tags.map{case (k, v) =>
+              Tag.builder()
+                .key(k)
+                .value(v)
+                .build()
+            }:_*
+          )
           .build()
       )
       .userData(java.util.Base64.getEncoder.encodeToString(userdata.getBytes))