com.microsoft.azure.toolkit.lib.cosmos.mongo.MongoCosmosDBAccount 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.resourcemanager.resources.fluentcore.arm.ResourceId;
import com.microsoft.azure.toolkit.lib.common.model.AbstractAzResourceModule;
import com.microsoft.azure.toolkit.lib.cosmos.CosmosDBAccount;
import com.microsoft.azure.toolkit.lib.cosmos.CosmosDBAccountModule;
import com.microsoft.azure.toolkit.lib.cosmos.model.CosmosDBAccountConnectionString;
import com.microsoft.azure.toolkit.lib.cosmos.model.MongoDatabaseAccountConnectionString;
import com.mongodb.ConnectionString;
import com.mongodb.MongoClient;
import com.mongodb.MongoClientURI;

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

public class MongoCosmosDBAccount extends CosmosDBAccount {
    private MongoClient mongoClient;
    private final MongoDatabaseModule mongoDatabaseModule;

    public MongoCosmosDBAccount(@Nonnull String name, @Nonnull String resourceGroupName, @Nonnull CosmosDBAccountModule module) {
        super(name, resourceGroupName, module);
        this.mongoDatabaseModule = new MongoDatabaseModule(this);
    }

    public MongoCosmosDBAccount(@Nonnull MongoCosmosDBAccount account) {
        super(account);
        this.mongoDatabaseModule = account.mongoDatabaseModule;
        this.mongoClient = account.mongoClient;
    }

    public MongoCosmosDBAccount(@Nonnull com.azure.resourcemanager.cosmos.models.CosmosDBAccount remote, @Nonnull CosmosDBAccountModule module) {
        super(remote.name(), ResourceId.fromString(remote.id()).resourceGroupName(), module);
        this.mongoDatabaseModule = new MongoDatabaseModule(this);
    }

    @Nullable
    public ConnectionString getPrimaryConnectionString() {
        return Optional.ofNullable(listConnectionStrings().getPrimaryConnectionString()).map(ConnectionString::new).orElse(null);
    }

    @Nonnull
    public MongoDatabaseAccountConnectionString getMongoConnectionString() {
        return Optional.ofNullable(listConnectionStrings().getPrimaryConnectionString())
                .map(MongoDatabaseAccountConnectionString::fromConnectionString)
                .orElseGet(MongoDatabaseAccountConnectionString::new);
    }

    @Nullable
    public List getHosts() {
        return getMongoConnectionString().getHosts();
    }

    @Nullable
    public String getContactPoint() {
        return getMongoConnectionString().getHost();
    }

    @Nullable
    public Integer getPort() {
        return getMongoConnectionString().getPort();
    }

    @Nullable
    public String getUserName() {
        return getMongoConnectionString().getUsername();
    }

    @Nullable
    public Boolean isSslEnabled() {
        return getMongoConnectionString().getSslEnabled();
    }

    public MongoDatabaseModule mongoDatabases() {
        return this.mongoDatabaseModule;
    }

    @Nonnull
    @Override
    public CosmosDBAccountConnectionString getCosmosDBAccountPrimaryConnectionString() {
        return getMongoConnectionString();
    }

    @Override
    public @Nonnull List> getSubModules() {
        return Collections.singletonList(mongoDatabaseModule);
    }

    public MongoClient getClient() {
        if (Objects.isNull(this.mongoClient)) {
            this.mongoClient = getMongoClient();
        }
        return this.mongoClient;
    }

    @Override
    protected void updateAdditionalProperties(com.azure.resourcemanager.cosmos.models.CosmosDBAccount newRemote, com.azure.resourcemanager.cosmos.models.CosmosDBAccount oldRemote) {
        super.updateAdditionalProperties(newRemote, oldRemote);
        if (Objects.nonNull(newRemote)) {
            this.mongoClient = getMongoClient();
        } else {
            try {
                Optional.ofNullable(this.mongoClient).ifPresent(MongoClient::close);
                this.mongoClient = null;
            } catch (final RuntimeException e) {
                // swallow exception during close client
            }
        }
    }

    private MongoClient getMongoClient() {
        try {
            final MongoDatabaseAccountConnectionString mongoConnectionString = Objects.requireNonNull(this.getMongoConnectionString());
            return new MongoClient(new MongoClientURI(mongoConnectionString.getConnectionString()));
        } catch (Throwable e) {
            // swallow exception to load data client
            return null;
        }
    }
}