Merge branch 'main' into orch-2

Author: nikhil-zlai

build.sbt

@@ -226,7 +226,6 @@ lazy val cloud_gcp = project
     libraryDependencies += "org.json4s" %% "json4s-native" % "3.7.0-M11",
     libraryDependencies += "org.json4s" %% "json4s-core" % "3.7.0-M11",
     libraryDependencies += "org.yaml" % "snakeyaml" % "2.3",
-    libraryDependencies += "com.google.cloud.bigtable" % "bigtable-hbase-2.x" % "2.14.2",
     libraryDependencies ++= avro,
     libraryDependencies ++= spark_all_provided,
     dependencyOverrides ++= jackson,

cloud_gcp/src/main/scala/ai/chronon/integrations/cloud_gcp/ApiFutureUtils.scala

@@ -0,0 +1,33 @@
+package ai.chronon.integrations.cloud_gcp
+
+import com.google.api.core.ApiFuture
+import com.google.api.core.ApiFutureCallback
+import com.google.api.core.ApiFutures
+import com.google.common.util.concurrent.MoreExecutors
+
+import java.util.concurrent.CompletableFuture
+
+// Fork of the bigtable-hbase ApiFutureUtils class to avoid taking a dependency on bigtable-hbase for one class
+// BigTable hbase brings in a ton of dependencies that we don't need for this one class
+object ApiFutureUtils {
+  def toCompletableFuture[T](apiFuture: ApiFuture[T]): CompletableFuture[T] = {
+    val completableFuture: CompletableFuture[T] = new CompletableFuture[T]() {
+      override def cancel(mayInterruptIfRunning: Boolean): Boolean = {
+        val result: Boolean = apiFuture.cancel(mayInterruptIfRunning)
+        super.cancel(mayInterruptIfRunning)
+        result
+      }
+    }
+    val callback: ApiFutureCallback[T] = new ApiFutureCallback[T]() {
+      override def onFailure(throwable: Throwable): Unit = {
+        completableFuture.completeExceptionally(throwable)
+      }
+
+      override def onSuccess(t: T): Unit = {
+        completableFuture.complete(t)
+      }
+    }
+    ApiFutures.addCallback(apiFuture, callback, MoreExecutors.directExecutor)
+    completableFuture
+  }
+}

cloud_gcp/src/main/scala/ai/chronon/integrations/cloud_gcp/BigTableKVStoreImpl.scala

@@ -29,7 +29,6 @@ import com.google.cloud.bigtable.data.v2.models.Query
 import com.google.cloud.bigtable.data.v2.models.Range.ByteStringRange
 import com.google.cloud.bigtable.data.v2.models.RowMutation
 import com.google.cloud.bigtable.data.v2.models.{TableId => BTTableId}
-import com.google.cloud.bigtable.hbase2_x.ApiFutureUtils
 import com.google.protobuf.ByteString
 import org.slf4j.Logger
 import org.slf4j.LoggerFactory