com.azure.cosmos.CosmosItemProperties Maven / Gradle / Ivy

Go to download
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
package com.azure.cosmos;

import com.azure.cosmos.implementation.Document;
import com.azure.cosmos.implementation.Utils;
import com.fasterxml.jackson.databind.ObjectMapper;

import java.io.IOException;
import java.util.List;
import java.util.stream.Collectors;

public class CosmosItemProperties extends Resource {

    private static final ObjectMapper MAPPER = Utils.getSimpleObjectMapper();

    /**
     * Initialize an empty CosmosItemProperties object.
     */
    public CosmosItemProperties() {
    }

    /**
     * Sets the id
     *
     * @param id the name of the resource.
     * @return the cosmos item properties with id set
     */
    public CosmosItemProperties setId(String id) {
        super.setId(id);
        return this;
    }

    /**
     * Initialize a CosmosItemProperties object from json string.
     *
     * @param jsonString the json string that represents the document object.
     */
    public CosmosItemProperties(String jsonString) {
        super(jsonString);
    }

    /**
     * fromObject returns Document for compatibility with V2 sdk
     */
    static Document fromObject(Object cosmosItem) {
        Document typedItem;
        if (cosmosItem instanceof CosmosItemProperties) {
            typedItem = new Document(((CosmosItemProperties) cosmosItem).toJson());
        } else {
            try {
                return new Document(CosmosItemProperties.MAPPER.writeValueAsString(cosmosItem));
            } catch (IOException e) {
                throw new IllegalArgumentException("Can't serialize the object into the json string", e);
            }
        }
        return typedItem;
    }

    static String toJsonString(Object cosmosItem, ObjectMapper objectMapper) {
        if (cosmosItem instanceof CosmosItemProperties) {
            return ((CosmosItemProperties) cosmosItem).toJson();
        } else {
            if (cosmosItem instanceof Document) {
                return ((Document) cosmosItem).toJson();
            }
            try {
                return objectMapper.writeValueAsString(cosmosItem);
            } catch (IOException e) {
                throw new IllegalArgumentException("Can't serialize the object into the json string", e);
            }
        }
    }

    static List getFromV2Results(List results) {
        return results.stream().map(document -> new CosmosItemProperties(document.toJson()))
                   .collect(Collectors.toList());
    }

    static  List getTypedResultsFromV2Results(List results, Class klass) {
        return results.stream().map(document -> document.toObject(klass))
                   .collect(Collectors.toList());
    }

    /**
     * Gets object.
     *
     * @param  the type parameter
     * @param klass the klass
     * @return the object
     * @throws IOException the io exception
     */
    public  T getObject(Class klass) throws IOException {
        return (T) MAPPER.readValue(this.toJson(), klass);
    }

}