Use generator service to register Azure Storage destination type. (#237)

* Use generator service to register Azure Storage destination type.

* Update DEPENDENCIES file.

* Update DEPENDENCIES file.

* Update DEPENDENCIES file.

* Update DEPENDENCIES file.

* Update DEPENDENCIES file.

* Update DEPENDENCIES file.

* Update DEPENDENCIES file to latest gen.

* Update DEPENDENCIES file to latest gen.

* Update DEPENDENCIES file to latest gen.

* Update DEPENDENCIES file to latest gen.

Author: bmg13

extensions/data-plane/data-plane-azure-storage/src/main/java/org/eclipse/edc/connector/dataplane/azure/storage/DataPlaneAzureStorageExtension.java

@@ -15,16 +15,20 @@
 package org.eclipse.edc.connector.dataplane.azure.storage;
 
 import dev.failsafe.RetryPolicy;
+import org.eclipse.edc.azure.blob.AzureBlobStoreSchema;
 import org.eclipse.edc.azure.blob.api.BlobStoreApi;
 import org.eclipse.edc.connector.dataplane.azure.storage.metadata.BlobMetadataProvider;
 import org.eclipse.edc.connector.dataplane.azure.storage.metadata.BlobMetadataProviderImpl;
 import org.eclipse.edc.connector.dataplane.azure.storage.metadata.CommonBlobMetadataDecorator;
 import org.eclipse.edc.connector.dataplane.azure.storage.pipeline.AzureStorageDataSinkFactory;
 import org.eclipse.edc.connector.dataplane.azure.storage.pipeline.AzureStorageDataSourceFactory;
+import org.eclipse.edc.connector.dataplane.spi.Endpoint;
+import org.eclipse.edc.connector.dataplane.spi.iam.PublicEndpointGeneratorService;
 import org.eclipse.edc.connector.dataplane.spi.pipeline.DataTransferExecutorServiceContainer;
 import org.eclipse.edc.connector.dataplane.spi.pipeline.PipelineService;
 import org.eclipse.edc.runtime.metamodel.annotation.Extension;
 import org.eclipse.edc.runtime.metamodel.annotation.Inject;
+import org.eclipse.edc.runtime.metamodel.annotation.Setting;
 import org.eclipse.edc.spi.security.Vault;
 import org.eclipse.edc.spi.system.ServiceExtension;
 import org.eclipse.edc.spi.system.ServiceExtensionContext;
@@ -36,6 +40,9 @@
 @Extension(value = DataPlaneAzureStorageExtension.NAME)
 public class DataPlaneAzureStorageExtension implements ServiceExtension {
 
+    @Setting(value = "Base url of the public API endpoint without the trailing slash.")
+    private static final String PUBLIC_ENDPOINT = "edc.dataplane.api.public.baseurl";
+
     public static final String NAME = "Data Plane Azure Storage";
     @Inject
     private RetryPolicy retryPolicy;
@@ -55,6 +62,9 @@ public class DataPlaneAzureStorageExtension implements ServiceExtension {
     @Inject
     private TypeManager typeManager;
 
+    @Inject
+    private PublicEndpointGeneratorService generatorService;
+
     @Override
     public String name() {
         return NAME;
@@ -68,6 +78,9 @@ public void initialize(ServiceExtensionContext context) {
         context.registerService(BlobMetadataProvider.class, metadataProvider);
         metadataProvider.registerDecorator(new CommonBlobMetadataDecorator(typeManager, context));
 
+        var endpoint = Endpoint.url(context.getSetting(PUBLIC_ENDPOINT, null));
+        generatorService.addGeneratorFunction(AzureBlobStoreSchema.TYPE, dataAddress -> endpoint);
+
         var sourceFactory = new AzureStorageDataSourceFactory(blobStoreApi, retryPolicy, monitor, vault);
         pipelineService.registerFactory(sourceFactory);
         var sinkFactory = new AzureStorageDataSinkFactory(blobStoreApi, executorContainer.getExecutorService(), 5, monitor, vault, typeManager, metadataProvider);

extensions/data-plane/data-plane-azure-storage/src/test/java/org/eclipse/edc/connector/dataplane/azure/storage/DataPlaneAzureStorageExtensionTest.java

@@ -0,0 +1,62 @@
+/*
+ *  Copyright (c) 2024 Bayerische Motoren Werke Aktiengesellschaft (BMW AG)
+ *
+ *  This program and the accompanying materials are made available under the
+ *  terms of the Apache License, Version 2.0 which is available at
+ *  https://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  SPDX-License-Identifier: Apache-2.0
+ *
+ *  Contributors:
+ *       Bayerische Motoren Werke Aktiengesellschaft (BMW AG)
+ *
+ */
+
+package org.eclipse.edc.connector.dataplane.azure.storage;
+
+import org.eclipse.edc.azure.blob.AzureBlobStoreSchema;
+import org.eclipse.edc.connector.dataplane.azure.storage.pipeline.AzureStorageDataSinkFactory;
+import org.eclipse.edc.connector.dataplane.azure.storage.pipeline.AzureStorageDataSourceFactory;
+import org.eclipse.edc.connector.dataplane.spi.iam.PublicEndpointGeneratorService;
+import org.eclipse.edc.connector.dataplane.spi.pipeline.PipelineService;
+import org.eclipse.edc.junit.extensions.DependencyInjectionExtension;
+import org.eclipse.edc.spi.system.ServiceExtensionContext;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+
+import java.util.function.Function;
+
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.ArgumentMatchers.isA;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+
+@ExtendWith(DependencyInjectionExtension.class)
+class DataPlaneAzureStorageExtensionTest {
+
+    private final PipelineService pipelineService = mock();
+    private final PublicEndpointGeneratorService generatorService = mock();
+
+    @BeforeEach
+    void setup(ServiceExtensionContext context) {
+        context.registerService(PipelineService.class, pipelineService);
+        context.registerService(PublicEndpointGeneratorService.class, generatorService);
+    }
+
+    @Test
+    void shouldProvidePipelineServices(DataPlaneAzureStorageExtension extension, ServiceExtensionContext context) {
+        extension.initialize(context);
+
+        verify(pipelineService).registerFactory(isA(AzureStorageDataSourceFactory.class));
+        verify(pipelineService).registerFactory(isA(AzureStorageDataSinkFactory.class));
+    }
+
+    @Test
+    void shouldInvokePublicEndpointGeneratorService(DataPlaneAzureStorageExtension extension, ServiceExtensionContext context) {
+        extension.initialize(context);
+
+        verify(generatorService).addGeneratorFunction(eq(AzureBlobStoreSchema.TYPE), isA(Function.class));
+    }
+
+}