[BUG][Spring Data Cosmos] PartitionKey of type UUID broken since 3.7.0

[witi83]

Describe the bug
This change broke the support for queries with PartitionKey of type UUID.

Exception or Stack Trace

java.lang.IllegalArgumentException: Unable to construct PartitionKeyInternal from objects array

        at com.azure.cosmos.implementation.routing.PartitionKeyInternal.fromObjectArray(PartitionKeyInternal.java:137)

        at com.azure.cosmos.implementation.routing.PartitionKeyInternal.fromObjectArray(PartitionKeyInternal.java:100)

        at com.azure.cosmos.models.PartitionKey.<init>(PartitionKey.java:31)

        at com.azure.spring.data.cosmos.core.CosmosTemplate.lambda$findItemsAsFlux$28(CosmosTemplate.java:860)

To Reproduce

data class User(
    val id: String,
    @PartitionKey val partitionKey: UUID
)

@Repository
interface UserRepository : CosmosRepository<User, String> {
    fun findByPartitionKey(partitionKey: UUID): Optional<User>
}

When you try to query it using repository.findByPartitionKey(UUID.randomUUID()) you get the above mentioned exception.

Additional information
Please note that IDs of type UUID are still not supported by azure-spring-data, which is really sad given the fact that UUIDs are handled as simple Strings in CosmosDB

[witi83]

@chenrujun any news?

[kushagraThapar]

@witi83 - UUID PartitionKey was never supported by spring data cosmos SDK. It is just that the error started surfacing since that PR supported partition key support for single partition queries in the above PR.

UUID type is now supported for Id, I have introduced it here - #24198
Support for UUID type for Partition Key is not supported through underlying azure-cosmos Java SDK. So spring data cosmos SDK cannot support it yet.

The reason for this unsupported UUID type as partition key is because UUID cannot be serialized / deserialized by JSON properly. Also, UUID needs to be read by cosmos db in both upper case / lower case without any discrepancy.
Due to these above reasons, we recommend only using String as partition key.

[witi83]

@kushagraThapar Thanks for the explanation. It makes sense to me!