feat: basic hudi support (#437)

## Summary

## Checklist
- [ ] Added Unit Tests
- [ ] Covered by existing CI
- [ ] Integration tested
- [ ] Documentation update
<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

- **Chores**
- Consolidated and streamlined build dependencies for improved
integration with AWS services and data processing libraries.
- Expanded the set of supported third-party libraries, including new
artifacts for enhanced performance and compatibility.
- Added new dependencies for Hudi, Jackson, and Zookeeper to enhance
functionality.
- Introduced additional Hudi artifacts for Scala 2.12 and 2.13 to
broaden available functionalities.

- **Tests**
- Added a new test class to verify reliable write/read operations on
Hudi tables using a Spark session.

- **Refactor**
- Enhanced serialization registration to support a broader range of data
types, improving overall processing stability.
- Introduced a new variable for shared library dependencies to simplify
dependency management.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

<!-- av pr metadata
This information is embedded by the av CLI when creating PRs to track
the status of stacks when using Aviator. Please do not delete or edit
this section of the PR.
```
{"parent":"main","parentHead":"","trunk":"main"}
```
-->

---------

Co-authored-by: Thomas Chow <[email protected]>

Author: tchow-zlai

cloud_aws/BUILD.bazel

@@ -1,3 +1,24 @@
+shared_libs = [
+    maven_artifact("software.amazon.awssdk:dynamodb"),
+    maven_artifact("software.amazon.awssdk:regions"),
+    maven_artifact("software.amazon.awssdk:aws-core"),
+    maven_artifact("software.amazon.awssdk:sdk-core"),
+    maven_artifact("software.amazon.awssdk:utils"),
+    maven_artifact("com.google.guava:guava"),
+    maven_artifact("org.slf4j:slf4j-api"),
+    maven_artifact("org.apache.hudi:hudi-aws-bundle"),
+    maven_artifact("junit:junit"),
+    maven_artifact("com.novocode:junit-interface"),
+    maven_artifact("com.fasterxml.jackson.module:jackson-module-afterburner"),
+    scala_artifact_with_suffix("org.apache.hudi:hudi-spark3.5-bundle"),
+    scala_artifact_with_suffix("org.scala-lang.modules:scala-collection-compat"),
+    "//api:lib",
+    "//api:thrift_java",
+    "//online:lib",
+    "//spark:lib",
+    "//tools/build_rules/spark:spark-exec",
+]
+
 scala_library(
     name = "cloud_aws_lib",
     srcs = glob(["src/main/**/*.scala"]),
@@ -6,54 +27,24 @@ scala_library(
         "//conditions:default": True,  # Enable for other versions
     }),
     visibility = ["//visibility:public"],
-    deps = [
-        maven_artifact("software.amazon.awssdk:dynamodb"),
-        maven_artifact("software.amazon.awssdk:regions"),
-        maven_artifact("software.amazon.awssdk:aws-core"),
-        maven_artifact("software.amazon.awssdk:sdk-core"),
-        maven_artifact("software.amazon.awssdk:utils"),
-        maven_artifact("com.google.guava:guava"),
-        maven_artifact("org.slf4j:slf4j-api"),
-        scala_artifact_with_suffix("org.scala-lang.modules:scala-collection-compat"),
-        "//api:lib",
-        "//api:thrift_java",
-        "//online:lib",
-        "//spark:lib",
-    ],
+    deps = shared_libs,
 )
 
 test_deps = [
-    ":cloud_aws_lib",
-    "//api:lib",
-    "//online:lib",
-    maven_artifact("software.amazon.awssdk:dynamodb"),
-    maven_artifact("software.amazon.awssdk:regions"),
-    maven_artifact("software.amazon.awssdk:aws-core"),
-    maven_artifact("software.amazon.awssdk:sdk-core"),
-    maven_artifact("software.amazon.awssdk:utils"),
+    maven_artifact("com.amazonaws:DynamoDBLocal"),
+    maven_artifact("org.mockito:mockito-core"),
     maven_artifact("software.amazon.awssdk:auth"),
     maven_artifact("software.amazon.awssdk:identity-spi"),
-    scala_artifact_with_suffix("org.typelevel:cats-core"),
-    maven_artifact("com.amazonaws:DynamoDBLocal"),
     scala_artifact_with_suffix("com.chuusai:shapeless"),
+    scala_artifact_with_suffix("org.typelevel:cats-core"),
+    scala_artifact_with_suffix("org.mockito:mockito-scala"),
 ] + _CIRCE_DEPS + _SCALA_TEST_DEPS
 
-scala_library(
-    name = "test_lib",
-    srcs = glob(["src/test/**/*.scala"]),
-    format = select({
-        "//tools/config:scala_2_13": False,  # Disable for 2.13
-        "//conditions:default": True,  # Enable for other versions
-    }),
-    visibility = ["//visibility:public"],
-    deps = test_deps,
-)
-
 scala_test_suite(
     name = "tests",
     srcs = glob(["src/test/**/*.scala"]),
     # defined in prelude_bazel file
     jvm_flags = _JVM_FLAGS_FOR_ACCESSING_BASE_JAVA_CLASSES,
     visibility = ["//visibility:public"],
-    deps = test_deps + [":test_lib"],
+    deps = shared_libs + test_deps + [":cloud_aws_lib"],
 )

cloud_aws/src/test/scala/ai/chronon/integrations/aws/GlueCatalogTest.scala

@@ -0,0 +1,38 @@
+package ai.chronon.integrations.aws
+
+
+import ai.chronon.spark.{SparkSessionBuilder, TableUtils, ChrononHudiKryoRegistrator}
+import org.apache.spark.sql.{SaveMode, SparkSession}
+import org.junit.Assert.assertEquals
+import org.scalatest.flatspec.AnyFlatSpec
+import org.scalatestplus.mockito.MockitoSugar
+
+class GlueCatalogTest extends AnyFlatSpec with MockitoSugar {
+
+  lazy val spark: SparkSession = SparkSessionBuilder.build(
+    classOf[GlueCatalogTest].getSimpleName,
+    local = true,
+    additionalConfig = Some(
+      Map(
+        "spark.sql.catalog.spark_catalog" -> "org.apache.spark.sql.hudi.catalog.HoodieCatalog",
+        "spark.sql.extensions" -> "org.apache.spark.sql.hudi.HoodieSparkSessionExtension",
+         "spark.kryo.registrator"-> classOf[ChrononHudiKryoRegistrator].getName,
+      ))
+  )
+  lazy val tableUtils: TableUtils = TableUtils(spark)
+
+  "basic round trip hudi table" should "work with local metastore" in {
+    import spark.implicits._
+
+    val input = Set(1,2,3,4)
+    val sourceDF = spark.sparkContext.parallelize(input.toSeq).toDF("id")
+
+    sourceDF.write.format("hudi")
+      .mode(SaveMode.Overwrite)
+      .saveAsTable("test_hudi_table")
+
+    val back = spark.table("test_hudi_table").select("id").as[Int].collect()
+    assertEquals(input, back.toSet)
+
+  }
+}