com.microsoft.azure.toolkit.lib.cosmos.mongo.MongoCollectionDraft Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of azure-toolkit-cosmos-lib Show documentation
There is a newer version: 0.48.0
Show newest version
/*
 * Copyright (c) Microsoft Corporation. All rights reserved.
 * Licensed under the MIT License. See License.txt in the project root for license information.
 */
package com.microsoft.azure.toolkit.lib.cosmos.mongo;

import com.azure.core.util.Context;
import com.azure.resourcemanager.cosmos.fluent.MongoDBResourcesClient;
import com.azure.resourcemanager.cosmos.fluent.models.MongoDBCollectionGetResultsInner;
import com.azure.resourcemanager.cosmos.models.MongoDBCollectionCreateUpdateParameters;
import com.azure.resourcemanager.cosmos.models.MongoDBCollectionResource;
import com.microsoft.azure.toolkit.lib.common.action.Action;
import com.microsoft.azure.toolkit.lib.common.action.AzureActionManager;
import com.microsoft.azure.toolkit.lib.common.bundle.AzureString;
import com.microsoft.azure.toolkit.lib.common.messager.AzureMessager;
import com.microsoft.azure.toolkit.lib.common.model.AzResource;
import com.microsoft.azure.toolkit.lib.common.operation.AzureOperation;
import com.microsoft.azure.toolkit.lib.common.utils.Utils;
import com.microsoft.azure.toolkit.lib.cosmos.ICosmosDocumentContainer;
import com.microsoft.azure.toolkit.lib.cosmos.model.ThroughputConfig;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.Setter;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.commons.lang3.StringUtils;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.Collections;
import java.util.Map;
import java.util.Objects;
import java.util.Optional;

public class MongoCollectionDraft extends MongoCollection implements
    AzResource.Draft {

    @Setter
    private MongoCollectionConfig config;

    protected MongoCollectionDraft(@Nonnull String name, @Nonnull String resourceGroupName, @Nonnull MongoCollectionModule module) {
        super(name, resourceGroupName, module);
    }

    @Override
    public void reset() {
        this.config = null;
    }

    @Nonnull
    @Override
    @AzureOperation(name = "azure/cosmos.create_mongo_collection.collection", params = {"this.getName()"})
    public MongoDBCollectionGetResultsInner createResourceInAzure() {
        final MongoDBResourcesClient mongoDBResourcesClient = Objects.requireNonNull(((MongoCollectionModule) Objects.requireNonNull(getModule())).getClient());
        final Map shardKey = StringUtils.isEmpty(ensureConfig().getShardKey()) ? null : Collections.singletonMap(ensureConfig().getShardKey(), "Hash");
        final MongoDBCollectionResource sqlContainerResource = new MongoDBCollectionResource()
                .withId(ensureConfig().getCollectionId())
                .withShardKey(shardKey);
        final MongoDBCollectionCreateUpdateParameters parameters = new MongoDBCollectionCreateUpdateParameters()
                .withLocation(Objects.requireNonNull(this.getParent().getRegion()).getName())
                .withResource(sqlContainerResource);
        parameters.withOptions(ensureConfig().toCreateUpdateOptions());
        AzureMessager.getMessager().info(AzureString.format("Start creating MongoDB collection({0})...", this.getName()));
        final MongoDBCollectionGetResultsInner result = mongoDBResourcesClient.createUpdateMongoDBCollection(this.getResourceGroupName(), this.getParent().getParent().getName(),
                this.getParent().getName(), this.getName(), parameters, Context.NONE);
        final Action> create = AzureActionManager.getInstance().getAction(CREATE_DOCUMENT).bind(this);
        final Action> upload = AzureActionManager.getInstance().getAction(IMPORT_DOCUMENT).bind(this);
        AzureMessager.getMessager().success(AzureString.format("MongoDB collection({0}) is successfully created.", this.getName()), create, upload);
        return result;
    }

    @Nonnull
    @Override
    @AzureOperation(name = "azure/cosmos.update_mongo_collection.collection", params = {"this.getName()"})
    public MongoDBCollectionGetResultsInner updateResourceInAzure(@Nonnull MongoDBCollectionGetResultsInner origin) {
        throw new UnsupportedOperationException("not support");
    }

    @Override
    public boolean isModified() {
        return config != null && ObjectUtils.anyNotNull(config.getCollectionId(), config.getShardKey(), config.getThroughput(), config.getMaxThroughput());
    }

    @Nullable
    @Override
    public MongoCollection getOrigin() {
        return null;
    }

    private MongoCollectionConfig ensureConfig() {
        this.config = Optional.ofNullable(config).orElseGet(MongoCollectionConfig::new);
        return this.config;
    }

    @Override
    public @Nullable String getSharedKey() {
        return Optional.ofNullable(config).map(MongoCollectionConfig::getShardKey).orElseGet(super::getSharedKey);
    }

    @Data
    @EqualsAndHashCode(callSuper = true)
    public static class MongoCollectionConfig extends ThroughputConfig {
        private String collectionId;
        private String shardKey;

        public static MongoCollectionConfig getDefaultConfig() {
            final MongoCollectionConfig result = new MongoCollectionConfig();
            result.setCollectionId(String.format("collection-%s", Utils.getTimestamp()));
            return result;
        }
    }
}