restconf/restconf-openapi/src/main/java/org/opendaylight/restconf/openapi/jaxrs/OpenApiBodyWriter.java

   1 /*
   2  * Copyright (c) 2023 PANTHEON.tech, s.r.o. 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.restconf.openapi.jaxrs;
   9
  10 import com.fasterxml.jackson.core.JsonGenerator;
  11 import java.io.ByteArrayOutputStream;
  12 import java.io.IOException;
  13 import java.io.OutputStream;
  14 import java.lang.annotation.Annotation;
  15 import java.lang.reflect.Type;
  16 import javax.ws.rs.Produces;
  17 import javax.ws.rs.core.MediaType;
  18 import javax.ws.rs.core.MultivaluedMap;
  19 import javax.ws.rs.ext.MessageBodyWriter;
  20 import javax.ws.rs.ext.Provider;
  21 import org.opendaylight.restconf.openapi.model.OpenApiEntity;
  22
  23 /**
  24  * A {@link MessageBodyWriter} capable of turning {@link OpenApiEntity} objects into JSON body.
  25  */
  26 @Provider
  27 @Produces(MediaType.APPLICATION_JSON)
  28 public final class OpenApiBodyWriter implements MessageBodyWriter<OpenApiEntity> {
  29     private final JsonGenerator generator;
  30     private final ByteArrayOutputStream stream;
  31
  32     public OpenApiBodyWriter(final JsonGenerator generator, final ByteArrayOutputStream stream) {
  33         this.generator = generator;
  34         this.stream = stream;
  35     }
  36
  37     @Override
  38     public boolean isWriteable(final Class<?> type, final Type genericType, final Annotation[] annotations,
  39             final MediaType mediaType) {
  40         return OpenApiEntity.class.isAssignableFrom(type);
  41     }
  42
  43     // TODO IMO we can get rid of implementing MessageBodyWriter
  44     @Override
  45     public void writeTo(final OpenApiEntity entity, final Class<?> type, final Type genericType,
  46             final Annotation[] annotations, final MediaType mediaType, final MultivaluedMap<String, Object> httpHeaders,
  47             final OutputStream entityStream) throws IOException {
  48         entity.generate(generator);
  49         generator.flush();
  50     }
  51
  52     public byte[] readFrom() {
  53         final var bytes = stream.toByteArray();
  54         stream.reset();
  55         return bytes;
  56     }
  57 }