com.azure.resourcemanager.compute.implementation.ComputeSkusImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of azure-resourcemanager-compute Show documentation

This package contains Microsoft Azure Compute Management SDK. For documentation on how to use this package, please see https://aka.ms/azsdk/java/mgmt

There is a newer version: 2.46.0

Show newest version

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

package com.azure.resourcemanager.compute.implementation;

import com.azure.core.http.rest.PagedFlux;
import com.azure.core.http.rest.PagedIterable;
import com.azure.resourcemanager.compute.ComputeManager;
import com.azure.resourcemanager.compute.models.ComputeResourceType;
import com.azure.resourcemanager.compute.models.ComputeSku;
import com.azure.resourcemanager.compute.models.ComputeSkus;
import com.azure.resourcemanager.compute.fluent.models.ResourceSkuInner;
import com.azure.resourcemanager.compute.fluent.ResourceSkusClient;
import com.azure.core.management.Region;
import com.azure.resourcemanager.resources.fluentcore.arm.collection.implementation.ReadableWrappersImpl;
import com.azure.resourcemanager.resources.fluentcore.utils.PagedConverter;
import reactor.core.publisher.Mono;

/** The implementation for {@link ComputeSkus}. */
public final class ComputeSkusImpl extends ReadableWrappersImpl
    implements ComputeSkus {
    private final ComputeManager manager;

    public ComputeSkusImpl(ComputeManager computeManager) {
        this.manager = computeManager;
    }

    @Override
    protected ComputeSkuImpl wrapModel(ResourceSkuInner inner) {
        return new ComputeSkuImpl(inner);
    }

    @Override
    public PagedIterable list() {
        return wrapList(this.inner().list());
    }

    @Override
    public PagedFlux listAsync() {
        return wrapPageAsync(this.inner().listAsync());
    }

    @Override
    public PagedIterable listByRegion(String regionName) {
        return this.listByRegion(Region.fromName(regionName));
    }

    @Override
    public PagedIterable listByRegion(Region region) {
        return new PagedIterable<>(listByRegionAsync(region));
    }

    @Override
    public PagedFlux listByRegionAsync(String regionName) {
        return this.listByRegionAsync(Region.fromName(regionName));
    }

    @Override
    public PagedFlux listByRegionAsync(final Region region) {
        return PagedConverter.mapPage(
            inner().listAsync(String.format("location eq '%s'", region.name()), null),
            this::wrapModel);
    }

    public ResourceSkusClient inner() {
        return this.manager.serviceClient().getResourceSkus();
    }

    @Override
    public ComputeManager manager() {
        return this.manager;
    }

    @Override
    public PagedIterable listByResourceType(ComputeResourceType resourceType) {
        return new PagedIterable<>(listByResourceTypeAsync(resourceType));
    }

    @Override
    public PagedFlux listByResourceTypeAsync(final ComputeResourceType resourceType) {
        return PagedConverter
            .flatMapPage(
                wrapPageAsync(inner().listAsync()),
                computeSku -> {
                    if (computeSku.resourceType() != null && computeSku.resourceType().equals(resourceType)) {
                        return Mono.just(computeSku);
                    } else {
                        return Mono.empty();
                    }
                });
    }

    @Override
    public PagedIterable listByRegionAndResourceType(Region region, ComputeResourceType resourceType) {
        return new PagedIterable<>(listByRegionAndResourceTypeAsync(region, resourceType));
    }

    @Override
    public PagedFlux listByRegionAndResourceTypeAsync(
        final Region region, final ComputeResourceType resourceType) {
        return PagedConverter
            .flatMapPage(
                wrapPageAsync(inner().listAsync(String.format("location eq '%s'", region.name()), null)),
                computeSku -> {
                    if (computeSku.resourceType() != null && computeSku.resourceType().equals(resourceType)) {
                        return Mono.just(computeSku);
                    } else {
                        return Mono.empty();
                    }
                });
    }
}