Azure Disk datafetcher

connector/azure/resourcemanager/src/main/java/com/blazebit/query/connector/azure/resourcemanager/AzureResourceManagerDisk.java

@@ -0,0 +1,22 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Blazebit
+ */
+package com.blazebit.query.connector.azure.resourcemanager;
+
+import com.azure.resourcemanager.compute.fluent.models.DiskInner;
+
+/**
+ * @author Martijn Sprengers
+ * @since 1.0.2
+ */
+public class AzureResourceManagerDisk extends AzureResourceManagerWrapper<DiskInner> {
+	public AzureResourceManagerDisk(String tenantId, String resourceId, DiskInner payload) {
+		super(tenantId, resourceId, payload);
+	}
+
+	@Override
+	public DiskInner getPayload() {
+		return super.getPayload();
+	}
+}

connector/azure/resourcemanager/src/main/java/com/blazebit/query/connector/azure/resourcemanager/AzureResourceManagerSchemaProvider.java

@@ -28,7 +28,8 @@ public Set<? extends DataFetcher<?>> resolveSchemaObjects(ConfigurationProvider
 				ResourceGroupDataFetcher.INSTANCE,
 				ManagedClusterDataFetcher.INSTANCE,
 				VaultDataFetcher.INSTANCE,
-				BlobServicePropertiesDataFetcher.INSTANCE
+				BlobServicePropertiesDataFetcher.INSTANCE,
+				DiskDataFetcher.INSTANCE
 		);
 	}
 }

connector/azure/resourcemanager/src/main/java/com/blazebit/query/connector/azure/resourcemanager/DiskDataFetcher.java

@@ -0,0 +1,56 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Blazebit
+ */
+package com.blazebit.query.connector.azure.resourcemanager;
+
+import com.azure.resourcemanager.AzureResourceManager;
+import com.azure.resourcemanager.compute.models.Disk;
+import com.blazebit.query.connector.base.DataFormats;
+import com.blazebit.query.spi.DataFetchContext;
+import com.blazebit.query.spi.DataFetcher;
+import com.blazebit.query.spi.DataFetcherException;
+import com.blazebit.query.spi.DataFormat;
+
+import java.io.Serializable;
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * @author Martijn Sprengers
+ * @since 1.0.2
+ */
+public class DiskDataFetcher implements DataFetcher<AzureResourceManagerDisk>, Serializable {
+
+	public static final DiskDataFetcher INSTANCE = new DiskDataFetcher();
+
+	private DiskDataFetcher() {
+	}
+
+	@Override
+	public DataFormat getDataFormat() {
+		return DataFormats.componentMethodConvention( AzureResourceManagerDisk.class, AzureResourceManagerConventionContext.INSTANCE );
+	}
+
+	@Override
+	public List<AzureResourceManagerDisk> fetch(DataFetchContext context) {
+		try {
+			List<AzureResourceManager> resourceManagers = AzureResourceManagerConnectorConfig.AZURE_RESOURCE_MANAGER.getAll(
+					context );
+			List<AzureResourceManagerDisk> list = new ArrayList<>();
+			for ( AzureResourceManager resourceManager : resourceManagers ) {
+				for ( Disk disk : resourceManager.disks().list() ) {
+					list.add( new AzureResourceManagerDisk(
+							resourceManager.tenantId(),
+							disk.id(),
+							disk.innerModel()
+					) );
+				}
+			}
+			return list;
+		}
+		catch (RuntimeException e) {
+			throw new DataFetcherException( "Could not fetch disk list", e );
+		}
+	}
+}