com.azure.cosmos.spark.udf.GetFeedRangeForHierarchicalPartitionKeyValues.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of azure-cosmos-spark_3-2_2-12 Show documentation

OLTP Spark 3.2 Connector for Azure Cosmos DB SQL API

There is a newer version: 4.34.0

Show newest version

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
package com.azure.cosmos.spark.udf

import com.azure.cosmos.implementation.SparkBridgeImplementationInternal
import com.azure.cosmos.spark.CosmosPredicates.requireNotNullOrEmpty
import org.apache.spark.sql.api.java.UDF2


@SerialVersionUID(1L)
class GetFeedRangeForHierarchicalPartitionKeyValues extends UDF2[String, Object, String] {
  override def call
  (
    partitionKeyDefinitionJson: String,
    partitionKeyValueJsonArray: Object
  ): String = {

    requireNotNullOrEmpty(partitionKeyDefinitionJson, "partitionKeyDefinitionJson")

    val range = SparkBridgeImplementationInternal
      .hierarchicalPartitionKeyValuesToNormalizedRange(partitionKeyValueJsonArray, partitionKeyDefinitionJson)

    s"${range.min}-${range.max}"
  }
}