org.mule.module.apikit.validation.io.XmlSchemaResource.java Source code

Introduction

Here is the source code for org.mule.module.apikit.validation.io.XmlSchemaResource.java

Source

/*
 * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.txt file.
 */
package org.mule.module.apikit.validation.io;

import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.concurrent.ExecutionException;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.SchemaOutputResolver;
import javax.xml.transform.Result;
import javax.xml.transform.stream.StreamResult;

import org.springframework.core.io.AbstractFileResolvingResource;
import org.springframework.util.Assert;
import org.springframework.util.ClassUtils;

public class XmlSchemaResource extends AbstractFileResolvingResource {

    private String className;

    private ClassLoader classLoader;

    private static LoadingCache<Class<?>, byte[]> schemaCache;

    static {
        schemaCache = CacheBuilder.newBuilder().maximumSize(1000).build(new CacheLoader<Class<?>, byte[]>() {
            public byte[] load(Class<?> clazz) throws JAXBException, IOException {
                final ByteArrayOutputStream baos = new ByteArrayOutputStream();

                // grab the context
                JAXBContext context = JAXBContext.newInstance(clazz);
                context.generateSchema(new SchemaOutputResolver() {
                    @Override
                    public Result createOutput(String namespaceURI, String suggestedFileName) throws IOException {
                        StreamResult result = new StreamResult(baos);
                        result.setSystemId("001");
                        return result;
                    }
                });

                return baos.toByteArray();
            }
        });

    }

    public XmlSchemaResource(String className, ClassLoader classLoader) {
        Assert.notNull(className, "Class name must not be null");
        this.className = className;
        this.classLoader = (classLoader != null ? classLoader : ClassUtils.getDefaultClassLoader());
    }

    @Override
    public String getDescription() {
        StringBuilder builder = new StringBuilder("annotated class resource [");
        builder.append(this.className);
        builder.append(']');
        return builder.toString();
    }

    @Override
    public InputStream getInputStream() throws IOException {
        try {
            Class<?> clazz = this.classLoader.loadClass(className);

            return new ByteArrayInputStream(schemaCache.get(clazz));
        } catch (ClassNotFoundException e) {
            throw new IOException(getDescription() + " cannot be found", e);
        } catch (ExecutionException e) {
            throw new IOException(getDescription() + " execution exception", e.getCause());
        }
    }
}