net.stickycode.resource.codec.JaxbResourceCodec Maven / Gradle / Ivy

Go to download
package net.stickycode.resource.codec;

import java.io.InputStream;
import java.io.OutputStream;
import java.nio.charset.Charset;

import javax.inject.Inject;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;
import javax.xml.bind.Unmarshaller;
import javax.xml.bind.annotation.XmlRootElement;
import javax.xml.transform.stream.StreamSource;

import net.stickycode.coercion.CoercionTarget;
import net.stickycode.resource.ResourceCodec;
import net.stickycode.stereotype.plugin.StickyExtension;
import net.stickycode.xml.jaxb.JaxbFactory;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

@StickyExtension
public class JaxbResourceCodec
    implements ResourceCodec {

  private Logger log = LoggerFactory.getLogger(getClass());

  @Inject
  private JaxbFactory jaxbFactory;

  @Override
  public void store(CoercionTarget sourceType, T resource, OutputStream output, Charset characterSet) {
    try {
      log.debug("storing {} of type {}", resource, sourceType);
      Marshaller m = jaxbFactory.createMarshaller(sourceType);
      m.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, Boolean.TRUE);
      m.marshal(resource, output);
    }
    catch (JAXBException e) {
      throw new ResourceEncodingFailureException(e, sourceType, this);
    }
  }

  @Override
  public T load(CoercionTarget resourceTarget, InputStream input, Charset characterSet) {
    log.debug("loading {} from {}", resourceTarget);
    @SuppressWarnings("unchecked")
    Class type = (Class) resourceTarget.getType();
    try {
      Unmarshaller u = jaxbFactory.createUnmarshaller(resourceTarget);
      return u.unmarshal(new StreamSource(input), type).getValue();
    }
    catch (JAXBException e) {
      throw new ResourceDecodingFailureException(e, type, this);
    }
  }

  @Override
  public boolean isApplicableTo(CoercionTarget type) {
    return type.getType().isAnnotationPresent(XmlRootElement.class);
  }

  @Override
  public String toString() {
    return getClass().getSimpleName();
  }

  @Override
  public String getDefaultFileSuffix() {
    return ".xml";
  }

}