|  | 
|  | 1 | +package io.quarkus.amazon.kms.deployment; | 
|  | 2 | + | 
|  | 3 | +import java.util.List; | 
|  | 4 | + | 
|  | 5 | +import org.jboss.jandex.DotName; | 
|  | 6 | + | 
|  | 7 | +import io.quarkus.amazon.common.deployment.AbstractAmazonServiceProcessor; | 
|  | 8 | +import io.quarkus.amazon.common.deployment.AmazonClientBuildItem; | 
|  | 9 | +import io.quarkus.amazon.common.deployment.AmazonClientBuilderBuildItem; | 
|  | 10 | +import io.quarkus.amazon.common.deployment.AmazonClientBuilderConfiguredBuildItem; | 
|  | 11 | +import io.quarkus.amazon.common.deployment.AmazonClientInterceptorsPathBuildItem; | 
|  | 12 | +import io.quarkus.amazon.common.deployment.AmazonClientTransportsBuildItem; | 
|  | 13 | +import io.quarkus.amazon.common.runtime.AmazonClientRecorder; | 
|  | 14 | +import io.quarkus.amazon.common.runtime.AmazonClientTransportRecorder; | 
|  | 15 | +import io.quarkus.amazon.kms.runtime.KmsBuildTimeConfig; | 
|  | 16 | +import io.quarkus.amazon.kms.runtime.KmsClientProducer; | 
|  | 17 | +import io.quarkus.amazon.kms.runtime.KmsConfig; | 
|  | 18 | +import io.quarkus.amazon.kms.runtime.KmsRecorder; | 
|  | 19 | +import io.quarkus.arc.deployment.AdditionalBeanBuildItem; | 
|  | 20 | +import io.quarkus.arc.deployment.BeanContainerBuildItem; | 
|  | 21 | +import io.quarkus.arc.deployment.BeanRegistrationPhaseBuildItem; | 
|  | 22 | +import io.quarkus.deployment.annotations.BuildProducer; | 
|  | 23 | +import io.quarkus.deployment.annotations.BuildStep; | 
|  | 24 | +import io.quarkus.deployment.annotations.ExecutionTime; | 
|  | 25 | +import io.quarkus.deployment.annotations.Record; | 
|  | 26 | +import io.quarkus.deployment.builditem.ExtensionSslNativeSupportBuildItem; | 
|  | 27 | +import io.quarkus.deployment.builditem.FeatureBuildItem; | 
|  | 28 | +import io.quarkus.deployment.builditem.ShutdownContextBuildItem; | 
|  | 29 | +import software.amazon.awssdk.services.kms.KmsAsyncClient; | 
|  | 30 | +import software.amazon.awssdk.services.kms.KmsClient; | 
|  | 31 | + | 
|  | 32 | +public class KmsProcessor extends AbstractAmazonServiceProcessor { | 
|  | 33 | + | 
|  | 34 | +    KmsBuildTimeConfig buildTimeConfig; | 
|  | 35 | + | 
|  | 36 | +    @Override | 
|  | 37 | +    protected String amazonServiceClientName() { | 
|  | 38 | +        return FeatureBuildItem.AMAZON_KMS; | 
|  | 39 | +    } | 
|  | 40 | + | 
|  | 41 | +    @Override | 
|  | 42 | +    protected String configName() { | 
|  | 43 | +        return "kms"; | 
|  | 44 | +    } | 
|  | 45 | + | 
|  | 46 | +    @Override | 
|  | 47 | +    protected DotName syncClientName() { | 
|  | 48 | +        return DotName.createSimple(KmsClient.class.getName()); | 
|  | 49 | +    } | 
|  | 50 | + | 
|  | 51 | +    @Override | 
|  | 52 | +    protected DotName asyncClientName() { | 
|  | 53 | +        return DotName.createSimple(KmsAsyncClient.class.getName()); | 
|  | 54 | +    } | 
|  | 55 | + | 
|  | 56 | +    @Override | 
|  | 57 | +    protected String builtinInterceptorsPath() { | 
|  | 58 | +        return "software/amazon/awssdk/services/kms/execution.interceptors"; | 
|  | 59 | +    } | 
|  | 60 | + | 
|  | 61 | +    @BuildStep | 
|  | 62 | +    AdditionalBeanBuildItem producer() { | 
|  | 63 | +        return AdditionalBeanBuildItem.unremovableOf(KmsClientProducer.class); | 
|  | 64 | +    } | 
|  | 65 | + | 
|  | 66 | +    @BuildStep | 
|  | 67 | +    void setup(BeanRegistrationPhaseBuildItem beanRegistrationPhase, | 
|  | 68 | +            BuildProducer<ExtensionSslNativeSupportBuildItem> extensionSslNativeSupport, | 
|  | 69 | +            BuildProducer<FeatureBuildItem> feature, | 
|  | 70 | +            BuildProducer<AmazonClientInterceptorsPathBuildItem> interceptors, | 
|  | 71 | +            BuildProducer<AmazonClientBuildItem> clientProducer) { | 
|  | 72 | + | 
|  | 73 | +        setupExtension(beanRegistrationPhase, extensionSslNativeSupport, feature, interceptors, clientProducer, | 
|  | 74 | +                buildTimeConfig.sdk, buildTimeConfig.syncClient); | 
|  | 75 | +    } | 
|  | 76 | + | 
|  | 77 | +    @BuildStep | 
|  | 78 | +    @Record(ExecutionTime.RUNTIME_INIT) | 
|  | 79 | +    void setupTransport(List<AmazonClientBuildItem> amazonClients, KmsRecorder recorder, | 
|  | 80 | +            AmazonClientTransportRecorder transportRecorder, | 
|  | 81 | +            KmsConfig runtimeConfig, BuildProducer<AmazonClientTransportsBuildItem> clientTransportBuildProducer) { | 
|  | 82 | + | 
|  | 83 | +        createTransportBuilders(amazonClients, | 
|  | 84 | +                transportRecorder, | 
|  | 85 | +                buildTimeConfig.syncClient, | 
|  | 86 | +                recorder.getSyncConfig(runtimeConfig), | 
|  | 87 | +                recorder.getAsyncConfig(runtimeConfig), | 
|  | 88 | +                clientTransportBuildProducer); | 
|  | 89 | +    } | 
|  | 90 | + | 
|  | 91 | +    @BuildStep | 
|  | 92 | +    @Record(ExecutionTime.RUNTIME_INIT) | 
|  | 93 | +    void createClientBuilders(List<AmazonClientTransportsBuildItem> transportBuildItems, KmsRecorder recorder, | 
|  | 94 | +            KmsConfig runtimeConfig, BuildProducer<AmazonClientBuilderBuildItem> builderProducer) { | 
|  | 95 | + | 
|  | 96 | +        createClientBuilders(transportBuildItems, builderProducer, | 
|  | 97 | +                (syncTransport) -> recorder.createSyncBuilder(runtimeConfig, syncTransport), | 
|  | 98 | +                (asyncTransport) -> recorder.createAsyncBuilder(runtimeConfig, asyncTransport)); | 
|  | 99 | +    } | 
|  | 100 | + | 
|  | 101 | +    @BuildStep | 
|  | 102 | +    @Record(ExecutionTime.RUNTIME_INIT) | 
|  | 103 | +    void configureClient(List<AmazonClientBuilderBuildItem> clients, KmsRecorder recorder, | 
|  | 104 | +            AmazonClientRecorder commonRecorder, | 
|  | 105 | +            KmsConfig runtimeConfig, | 
|  | 106 | +            BuildProducer<AmazonClientBuilderConfiguredBuildItem> producer) { | 
|  | 107 | + | 
|  | 108 | +        initClientBuilders(clients, commonRecorder, recorder.getAwsConfig(runtimeConfig), recorder.getSdkConfig(runtimeConfig), | 
|  | 109 | +                buildTimeConfig.sdk, producer); | 
|  | 110 | +    } | 
|  | 111 | + | 
|  | 112 | +    @BuildStep | 
|  | 113 | +    @Record(ExecutionTime.RUNTIME_INIT) | 
|  | 114 | +    void buildClients(List<AmazonClientBuilderConfiguredBuildItem> configuredClients, KmsRecorder recorder, | 
|  | 115 | +            BeanContainerBuildItem beanContainer, | 
|  | 116 | +            ShutdownContextBuildItem shutdown) { | 
|  | 117 | + | 
|  | 118 | +        buildClients(configuredClients, | 
|  | 119 | +                (syncBuilder) -> recorder.buildClient(syncBuilder, beanContainer.getValue(), shutdown), | 
|  | 120 | +                (asyncBuilder) -> recorder.buildAsyncClient(asyncBuilder, beanContainer.getValue(), shutdown)); | 
|  | 121 | +    } | 
|  | 122 | +} | 
0 commit comments