Search in sources :

Example 46 with Client

use of com.aliyun.dysmsapi20170525.Client in project stream-services by Backbase.

the class DbsWebClientConfiguration method dbsWebClient.

/**
 * Default Reactive Web Client to be used when interacting with DBS Services. Requires OAuth2 client credentials set
 * in application.yml
 *
 * @param objectMapper                          The Jackson Object mapper to register serialization and deserialization json
 *                                              content.
 * @param reactiveOAuth2AuthorizedClientManager Client Manager managing OAuth2 tokens
 * @param builder                               THe Web Client Builder which is already preconfigured using MicroMeter
 *                                              instrumentation.
 * @return Preconfigured Web Client
 */
@Bean
public WebClient dbsWebClient(ObjectMapper objectMapper, ReactiveOAuth2AuthorizedClientManager reactiveOAuth2AuthorizedClientManager, WebClient.Builder builder, DbsWebClientConfigurationProperties dbsWebClientConfigurationProperties) {
    ServerOAuth2AuthorizedClientExchangeFilterFunction oauth2ClientFilter = new ServerOAuth2AuthorizedClientExchangeFilterFunction(reactiveOAuth2AuthorizedClientManager);
    oauth2ClientFilter.setDefaultClientRegistrationId(dbsWebClientConfigurationProperties.getDefaultClientRegistrationId());
    builder.defaultHeader("Content-Type", MediaType.APPLICATION_JSON.toString()).defaultHeader("Accept", MediaType.APPLICATION_JSON.toString()).filter((clientRequest, exchangeFunction) -> {
        final ClientRequest newRequest = Optional.ofNullable(dbsWebClientConfigurationProperties.getAdditionalHeaders()).map(additionalHeaders -> {
            log.debug("Adding additional headers: {} from configuration  Request: {}", additionalHeaders, clientRequest.url());
            return ClientRequest.from(clientRequest).headers(httpHeaders -> httpHeaders.addAll(additionalHeaders)).build();
        }).orElse(clientRequest);
        return Mono.subscriberContext().flatMap(context -> {
            Optional<MultiValueMap<String, String>> forwardHeaders = context.<MultiValueMap<String, String>>getOrEmpty(CONTEXT_KEY_FORWARDED_HEADERS);
            log.debug("context contains headers? " + forwardHeaders.isPresent());
            log.debug("forward headers:" + forwardHeaders.map(MultiValueMap::toString).orElse("null"));
            ClientRequest contextRequest = context.<MultiValueMap<String, String>>getOrEmpty("headers").map(headers -> {
                log.debug("Adding additional headers: {} from Reactive subscriber context to Request: {}", headers, clientRequest.url());
                return ClientRequest.from(newRequest).headers(httpHeaders -> httpHeaders.addAll(headers)).build();
            }).orElse(newRequest);
            return exchangeFunction.exchange(contextRequest);
        });
    }).filter(new CsrfClientExchangeFilterFunction()).filter(oauth2ClientFilter);
    if (log.isDebugEnabled()) {
        HttpClient httpClient = HttpClient.create().wiretap("reactor.netty.http.client.HttpClient", LogLevel.DEBUG, AdvancedByteBufFormat.TEXTUAL);
        builder.clientConnector(new ReactorClientHttpConnector(httpClient));
    }
    // ensure correct exchange strategy is installed
    ExchangeStrategies strategies = ExchangeStrategies.builder().codecs(clientDefaultCodecsConfigurer -> {
        Jackson2JsonEncoder encoder = new Jackson2JsonEncoder(objectMapper, MediaType.APPLICATION_JSON);
        Jackson2JsonDecoder decoder = new Jackson2JsonDecoder(objectMapper, MediaType.APPLICATION_JSON);
        clientDefaultCodecsConfigurer.defaultCodecs().jackson2JsonEncoder(encoder);
        clientDefaultCodecsConfigurer.defaultCodecs().jackson2JsonDecoder(decoder);
    }).build();
    builder.exchangeStrategies(strategies);
    return builder.build();
}
Also used : java.util(java.util) DbsWebClientConfigurationProperties(com.backbase.stream.webclient.configuration.DbsWebClientConfigurationProperties) Jackson2JsonEncoder(org.springframework.http.codec.json.Jackson2JsonEncoder) DeserializationFeature(com.fasterxml.jackson.databind.DeserializationFeature) ServerOAuth2AuthorizedClientExchangeFilterFunction(org.springframework.security.oauth2.client.web.reactive.function.client.ServerOAuth2AuthorizedClientExchangeFilterFunction) JavaTimeModule(com.fasterxml.jackson.datatype.jsr310.JavaTimeModule) org.springframework.web.reactive.function.client(org.springframework.web.reactive.function.client) EnableConfigurationProperties(org.springframework.boot.context.properties.EnableConfigurationProperties) OAuth2ClientProperties(org.springframework.boot.autoconfigure.security.oauth2.client.OAuth2ClientProperties) OAuth2ClientPropertiesRegistrationAdapter(org.springframework.boot.autoconfigure.security.oauth2.client.OAuth2ClientPropertiesRegistrationAdapter) DateFormat(java.text.DateFormat) ResponseCookie(org.springframework.http.ResponseCookie) ReactorClientHttpConnector(org.springframework.http.client.reactive.ReactorClientHttpConnector) AdvancedByteBufFormat(reactor.netty.transport.logging.AdvancedByteBufFormat) Jackson2JsonDecoder(org.springframework.http.codec.json.Jackson2JsonDecoder) ReactiveClientRegistrationRepository(org.springframework.security.oauth2.client.registration.ReactiveClientRegistrationRepository) MediaType(org.springframework.http.MediaType) org.springframework.security.oauth2.client(org.springframework.security.oauth2.client) ObjectMapper(com.fasterxml.jackson.databind.ObjectMapper) MultiValueMap(org.springframework.util.MultiValueMap) Mono(reactor.core.publisher.Mono) ClientRegistration(org.springframework.security.oauth2.client.registration.ClientRegistration) Configuration(org.springframework.context.annotation.Configuration) InMemoryReactiveClientRegistrationRepository(org.springframework.security.oauth2.client.registration.InMemoryReactiveClientRegistrationRepository) Slf4j(lombok.extern.slf4j.Slf4j) LogLevel(io.netty.handler.logging.LogLevel) JsonInclude(com.fasterxml.jackson.annotation.JsonInclude) StdDateFormat(com.fasterxml.jackson.databind.util.StdDateFormat) Bean(org.springframework.context.annotation.Bean) HttpClient(reactor.netty.http.client.HttpClient) Jackson2JsonDecoder(org.springframework.http.codec.json.Jackson2JsonDecoder) HttpClient(reactor.netty.http.client.HttpClient) ServerOAuth2AuthorizedClientExchangeFilterFunction(org.springframework.security.oauth2.client.web.reactive.function.client.ServerOAuth2AuthorizedClientExchangeFilterFunction) MultiValueMap(org.springframework.util.MultiValueMap) ReactorClientHttpConnector(org.springframework.http.client.reactive.ReactorClientHttpConnector) Jackson2JsonEncoder(org.springframework.http.codec.json.Jackson2JsonEncoder) Bean(org.springframework.context.annotation.Bean)

Example 47 with Client

use of com.aliyun.dysmsapi20170525.Client in project submarine by apache.

the class ModelManager method transferDescription.

private void transferDescription(ServeSpec spec) {
    Client s3Client = new Client();
    String modelUniquePath = String.format("%s-%d-%s", spec.getModelName(), spec.getModelVersion(), spec.getModelId());
    String res = new String(s3Client.downloadArtifact(String.format("registry/%s/%s/%d/description.json", modelUniquePath, spec.getModelName(), spec.getModelVersion())));
    JSONObject description = new JSONObject(res);
    TritonModelConfig.ModelConfig.Builder modelConfig = TritonModelConfig.ModelConfig.newBuilder();
    modelConfig.setPlatform("pytorch_libtorch");
    JSONArray inputs = (JSONArray) description.get("input");
    for (int idx = 0; idx < inputs.length(); idx++) {
        JSONArray dims = (JSONArray) ((JSONObject) inputs.get(idx)).get("dims");
        TritonModelConfig.ModelInput.Builder modelInput = TritonModelConfig.ModelInput.newBuilder();
        modelInput.setName("INPUT__" + idx);
        modelInput.setDataType(TritonModelConfig.DataType.valueOf("TYPE_FP32"));
        dims.forEach(dim -> modelInput.addDims((Integer) dim));
        modelConfig.addInput(modelInput);
    }
    JSONArray outputs = (JSONArray) description.get("output");
    for (int idx = 0; idx < outputs.length(); idx++) {
        JSONArray dims = (JSONArray) ((JSONObject) outputs.get(idx)).get("dims");
        TritonModelConfig.ModelOutput.Builder modelOutput = TritonModelConfig.ModelOutput.newBuilder();
        modelOutput.setName("OUTPUT__" + idx);
        modelOutput.setDataType(TritonModelConfig.DataType.valueOf("TYPE_FP32"));
        dims.forEach(dim -> modelOutput.addDims((Integer) dim));
        modelConfig.addOutput(modelOutput);
    }
    s3Client.logArtifact(String.format("registry/%s/%s/config.pbtxt", modelUniquePath, spec.getModelName()), modelConfig.toString().getBytes());
}
Also used : TritonModelConfig(org.apache.submarine.server.api.proto.TritonModelConfig) JSONObject(org.json.JSONObject) JSONArray(org.json.JSONArray) Client(org.apache.submarine.server.s3.Client)

Aggregations

Client (org.orcid.jaxb.model.client_v2.Client)18 Test (org.junit.Test)13 Client (com.aliyun.dysmsapi20170525.Client)12 ClientDetailsEntity (org.orcid.persistence.jpa.entities.ClientDetailsEntity)10 Client (com.predic8.membrane.core.interceptor.oauth2.Client)9 Config (com.aliyun.teaopenapi.models.Config)7 HashSet (java.util.HashSet)6 ClientRedirectUri (org.orcid.jaxb.model.client_v2.ClientRedirectUri)6 SessionManager (com.predic8.membrane.core.interceptor.authentication.session.SessionManager)5 BaseTest (org.orcid.core.BaseTest)5 ScopePathType (org.orcid.jaxb.model.message.ScopePathType)5 SendSmsRequest (com.aliyun.dysmsapi20170525.models.SendSmsRequest)3 NoResponse (com.predic8.membrane.core.interceptor.oauth2.request.NoResponse)3 IOException (java.io.IOException)3 ArrayList (java.util.ArrayList)3 Date (java.util.Date)3 SendSmsResponse (com.aliyun.dysmsapi20170525.models.SendSmsResponse)2 Client (com.aliyun.ons20190214.Client)2 OnsTopicListRequest (com.aliyun.ons20190214.models.OnsTopicListRequest)2 OnsTopicListResponse (com.aliyun.ons20190214.models.OnsTopicListResponse)2