main/src/main/java/org/onap/policy/api/main/config/WebConfig.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2022 Bell Canada. All rights reserved.
   4  * ================================================================================
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  * SPDX-License-Identifier: Apache-2.0
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.policy.api.main.config;
  22
  23 import com.google.gson.Gson;
  24 import com.google.gson.GsonBuilder;
  25 import com.google.gson.JsonParser;
  26 import com.google.gson.JsonSerializer;
  27 import java.util.Arrays;
  28 import java.util.List;
  29 import org.onap.policy.api.main.config.converter.StringToEnumConverter;
  30 import org.onap.policy.api.main.config.converter.YamlHttpMessageConverter;
  31 import org.springframework.context.annotation.Configuration;
  32 import org.springframework.format.FormatterRegistry;
  33 import org.springframework.http.MediaType;
  34 import org.springframework.http.converter.HttpMessageConverter;
  35 import org.springframework.http.converter.json.GsonHttpMessageConverter;
  36 import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
  37 import springfox.documentation.spring.web.json.Json;
  38
  39 /**
  40  * Register custom converters to Spring configuration.
  41  */
  42 @Configuration
  43 public class WebConfig implements WebMvcConfigurer {
  44     @Override
  45     public void addFormatters(FormatterRegistry registry) {
  46         registry.addConverter(new StringToEnumConverter());
  47     }
  48
  49     @Override
  50     public void extendMessageConverters(List<HttpMessageConverter<?>> converters) {
  51         YamlHttpMessageConverter yamlConverter = new YamlHttpMessageConverter();
  52         yamlConverter.setSupportedMediaTypes(Arrays.asList(MediaType.parseMediaType("application/yaml")));
  53         converters.add(yamlConverter);
  54
  55         GsonHttpMessageConverter converter = buildGsonConverter();
  56         converters.removeIf(c -> c instanceof GsonHttpMessageConverter);
  57         converters.add(0, converter);
  58     }
  59
  60     /**
  61      * Swagger uses {{@link springfox.documentation.spring.web.json.Json}} which leads to Gson serialization errors.
  62      * Hence, we customize a typeAdapter on the Gson bean in the Gson http message converter.
  63      *
  64      * @return customised GSON HttpMessageConverter instance.
  65      */
  66     private GsonHttpMessageConverter buildGsonConverter() {
  67         JsonSerializer<Json> serializer = (json, type, jsonSerializationContext) ->
  68             JsonParser.parseString(json.value());
  69         Gson gson = new GsonBuilder().registerTypeAdapter(Json.class, serializer).create();
  70         return new GsonHttpMessageConverter(gson);
  71     }
  72 }