opendaylight/md-sal/sal-rest-connector/src/main/java/org/opendaylight/controller/sal/rest/impl/StructuredDataToJsonProvider.java

   1 package org.opendaylight.controller.sal.rest.impl;
   2
   3 import static org.opendaylight.controller.sal.restconf.impl.MediaTypes.API;
   4
   5 import java.io.IOException;
   6 import java.io.OutputStream;
   7 import java.io.OutputStreamWriter;
   8 import java.lang.annotation.Annotation;
   9 import java.lang.reflect.Type;
  10
  11 import javax.ws.rs.Produces;
  12 import javax.ws.rs.WebApplicationException;
  13 import javax.ws.rs.core.MediaType;
  14 import javax.ws.rs.core.MultivaluedMap;
  15 import javax.ws.rs.core.Response;
  16 import javax.ws.rs.ext.MessageBodyWriter;
  17 import javax.ws.rs.ext.Provider;
  18
  19 import org.opendaylight.controller.sal.rest.api.RestconfService;
  20 import org.opendaylight.controller.sal.restconf.impl.StructuredData;
  21 import org.opendaylight.yangtools.yang.data.api.CompositeNode;
  22 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
  23
  24 import com.google.gson.stream.JsonWriter;
  25
  26 @Provider
  27 @Produces({API+RestconfService.JSON})
  28 public enum StructuredDataToJsonProvider implements MessageBodyWriter<StructuredData> {
  29     INSTANCE;
  30
  31     @Override
  32     public boolean isWriteable(Class<?> type, Type genericType, Annotation[] annotations, MediaType mediaType) {
  33         return true;
  34     }
  35
  36     @Override
  37     public long getSize(StructuredData t, Class<?> type, Type genericType, Annotation[] annotations, MediaType mediaType) {
  38         return -1;
  39     }
  40
  41     @Override
  42     public void writeTo(StructuredData t, Class<?> type, Type genericType, Annotation[] annotations,
  43             MediaType mediaType, MultivaluedMap<String, Object> httpHeaders, OutputStream entityStream)
  44             throws IOException, WebApplicationException {
  45         CompositeNode data = t.getData();
  46         if (data == null) {
  47             throw new WebApplicationException(Response.status(Response.Status.NOT_FOUND).build());
  48         }
  49
  50         JsonWriter writer = new JsonWriter(new OutputStreamWriter(entityStream, "UTF-8"));
  51         writer.setIndent("    ");
  52         JsonMapper jsonMapper = new JsonMapper();
  53         jsonMapper.write(writer, data, (DataNodeContainer) t.getSchema());
  54         writer.flush();
  55     }
  56
  57 }