com.azure.cosmos.spark.PartitionKeyHelper.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-5_2-12 Show documentation

OLTP Spark 3.5 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

import com.azure.cosmos.implementation.InternalObjectNode
import com.azure.cosmos.models.{PartitionKey, PartitionKeyDefinition}
import com.fasterxml.jackson.databind.node.ObjectNode

private object PartitionKeyHelper {
  def getPartitionKeyPath(objectNode: ObjectNode,
                          partitionKeyDefinition: PartitionKeyDefinition): PartitionKey = {
      com.azure.cosmos.implementation.PartitionKeyHelper.extractPartitionKeyFromDocument(
          new InternalObjectNode(objectNode),
          partitionKeyDefinition)
  }
}