com.atlan.model.core.AsyncCreationResponse Maven / Gradle / Ivy

Go to download
// Generated by delombok at Wed Oct 16 22:16:04 UTC 2024
/* SPDX-License-Identifier: Apache-2.0
   Copyright 2022 Atlan Pte. Ltd. */
package com.atlan.model.core;

import com.atlan.exception.ApiException;
import com.atlan.exception.AtlanException;
import com.atlan.exception.ErrorCode;
import com.atlan.exception.PermissionException;
import com.atlan.model.assets.Asset;
import com.atlan.model.assets.AtlanCollection;
import com.atlan.model.assets.Connection;
import com.atlan.net.HttpClient;
import com.atlan.net.RequestOptions;
import java.util.ArrayList;
import java.util.List;

public class AsyncCreationResponse extends AssetMutationResponse implements AtlanAsyncMutator {
    @java.lang.SuppressWarnings("all")
    @lombok.Generated
    private static final org.slf4j.Logger log = org.slf4j.LoggerFactory.getLogger(AsyncCreationResponse.class);
    private static final long serialVersionUID = 2L;

    /**
     * Block until the connection that was created is confirmed to be accessible,
     * since policies linked to the connection are created asynchronously.
     * @throws ApiException if the retry loop is interrupted or the maximum number of retries is hit
     */
    @Override
    public AssetMutationResponse block() throws ApiException {
        List toCheck = getCreatedAssets();
        if (toCheck != null && !toCheck.isEmpty()) {
            try {
                retrieveAndCheck(toCheck, 0);
            } catch (InterruptedException e) {
                throw new ApiException(ErrorCode.RETRIES_INTERRUPTED, e);
            }
        }
        return this;
    }

    /**
     * Continually iterate through the list of created connections to confirm they have been created.
     * Only retry up to the maximum defined by Atlan.getMaxNetworkRetries()
     *
     * @param toCheck list of assets to confirm are accessible
     * @param retryCount current retry
     * @throws InterruptedException if the retry loop is interrupted
     * @throws ApiException if the maximum number of retries is hit without confirming accessibility
     */
    private void retrieveAndCheck(List toCheck, int retryCount) throws InterruptedException, ApiException {
        List leftovers = new ArrayList<>();
        for (Asset one : toCheck) {
            if (one instanceof Connection || one instanceof AtlanCollection) {
                log.debug(" ... blocking to confirm {} accessibility: {}", one.getTypeName(), one.getGuid());
                // Only even attempt to look at an asset if it is a connection or a collection,
                // otherwise skip it entirely
                try {
                    AssetResponse candidate = client.assets.get(one.getGuid(), false, false, RequestOptions.from(client).maxNetworkRetries(MAX_ASYNC_RETRIES).build());
                    if (candidate == null || candidate.getAsset() == null) {
                        // Since the retry logic in this case is actually embedded in the retrieveMinimal
                        // call, if we get to this point without retrieving the connection we have by
                        // definition overrun the retry limit
                        throw new ApiException(ErrorCode.RETRY_OVERRUN, null);
                    }
                } catch (PermissionException e) {
                    // If we get a permission exception after the built-in retries above, throw it
                    // onwards as a retry overrun
                    throw new ApiException(ErrorCode.RETRY_OVERRUN, e);
                } catch (AtlanException e) {
                    // If there was some other exception, we should try again
                    leftovers.add(one);
                }
            }
        }
        if (!leftovers.isEmpty()) {
            if (retryCount == client.getMaxNetworkRetries()) {
                throw new ApiException(ErrorCode.RETRY_OVERRUN, null);
            } else {
                Thread.sleep(HttpClient.waitTime(retryCount).toMillis());
                retrieveAndCheck(leftovers, retryCount + 1);
            }
        }
    }

    @java.lang.Override
    @java.lang.SuppressWarnings("all")
    @lombok.Generated
    public boolean equals(final java.lang.Object o) {
        if (o == this) return true;
        if (!(o instanceof AsyncCreationResponse)) return false;
        final AsyncCreationResponse other = (AsyncCreationResponse) o;
        if (!other.canEqual((java.lang.Object) this)) return false;
        return true;
    }

    @java.lang.SuppressWarnings("all")
    @lombok.Generated
    protected boolean canEqual(final java.lang.Object other) {
        return other instanceof AsyncCreationResponse;
    }

    @java.lang.Override
    @java.lang.SuppressWarnings("all")
    @lombok.Generated
    public int hashCode() {
        final int result = 1;
        return result;
    }

    @java.lang.Override
    @java.lang.SuppressWarnings("all")
    @lombok.Generated
    public java.lang.String toString() {
        return "AsyncCreationResponse(super=" + super.toString() + ")";
    }
}