pce/src/main/java/org/opendaylight/transportpce/pce/gnpy/consumer/RequestMessageBodyWriter.java

   1 /*
   2  * Copyright © 2022 Orange, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.transportpce.pce.gnpy.consumer;
   9
  10 import java.io.IOException;
  11 import java.io.OutputStream;
  12 import java.lang.annotation.Annotation;
  13 import java.lang.reflect.Type;
  14 import java.nio.charset.StandardCharsets;
  15 import javax.ws.rs.WebApplicationException;
  16 import javax.ws.rs.core.MediaType;
  17 import javax.ws.rs.core.MultivaluedMap;
  18 import javax.ws.rs.ext.MessageBodyWriter;
  19 import org.opendaylight.transportpce.common.converter.JsonStringConverter;
  20 import org.opendaylight.yang.gen.v1.gnpy.gnpy.api.rev220221.Request;
  21 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  22 import org.opendaylight.yangtools.yang.data.codec.gson.JSONCodecFactorySupplier;
  23 import org.slf4j.Logger;
  24 import org.slf4j.LoggerFactory;
  25
  26 public class RequestMessageBodyWriter implements MessageBodyWriter<Request> {
  27     private static final Logger LOG = LoggerFactory.getLogger(RequestMessageBodyWriter.class);
  28     private final InstanceIdentifier<Request> idRequest = InstanceIdentifier.builder(Request.class).build();
  29     private final JsonStringConverter<Request> converter;
  30
  31     public RequestMessageBodyWriter(JsonStringConverter<Request> converter) {
  32         this.converter = converter;
  33     }
  34
  35     @Override
  36     @SuppressWarnings("java:S1872")
  37     public boolean isWriteable(Class<?> type, Type genericType, Annotation[] annotations, MediaType mediaType) {
  38         return "org.opendaylight.yang.gen.v1.gnpy.gnpy.api.rev220221.Request$$$codecImpl"
  39                 .equals(type.getName())
  40                 || "org.opendaylight.yang.gen.v1.gnpy.gnpy.api.rev220221.RequestBuilder$RequestImpl"
  41                         .equals(type.getName());
  42     }
  43
  44     @Override
  45     public void writeTo(Request request, Class<?> type, Type genericType, Annotation[] annotations,
  46                         MediaType mediaType, MultivaluedMap<String, Object> httpHeaders, OutputStream entityStream)
  47             throws IOException, WebApplicationException {
  48         // we have to use a string because GNPy does not support prefix
  49         String jsonStringFromDataObject = converter
  50                 .createJsonStringFromDataObject(idRequest, request,
  51                         JSONCodecFactorySupplier.DRAFT_LHOTKA_NETMOD_YANG_JSON_02)
  52                 .replace("gnpy-network-topology:", "");
  53         LOG.debug("Serialized request {}", jsonStringFromDataObject);
  54         entityStream.write(jsonStringFromDataObject
  55                 .getBytes(StandardCharsets.UTF_8));
  56     }
  57 }