com.kixeye.chassis.transport.swagger.SwaggerRegistry.java Source code

Introduction

Here is the source code for com.kixeye.chassis.transport.swagger.SwaggerRegistry.java

Source

package com.kixeye.chassis.transport.swagger;

/*
 * #%L
 * Chassis Transport Swagger
 * %%
 * Copyright (C) 2014 KIXEYE, Inc
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */

import com.fasterxml.jackson.databind.Module;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.module.SimpleModule;
import com.mangofactory.swagger.configuration.SwaggerApiListingJsonSerializer;
import com.mangofactory.swagger.configuration.SwaggerResourceListingJsonSerializer;
import com.wordnik.swagger.model.ApiListing;
import com.wordnik.swagger.model.ResourceListing;
import org.eclipse.jetty.servlet.ServletContextHandler;
import org.eclipse.jetty.servlet.ServletHolder;
import org.springframework.stereotype.Component;
import java.util.Collection;

/**
 * Registers swagger with Jetty.
 * 
 * @author ebahtijaragic
 */
@Component
public class SwaggerRegistry {
    public static final String DEFAULT_SWAGGER_CONTEXT_PATH = "/swagger/*";

    /**
     * Registers Swagger with default context.
     */
    public void registerSwagger(ServletContextHandler context, Collection<ObjectMapper> objectMappers) {
        registerSwagger(context, DEFAULT_SWAGGER_CONTEXT_PATH, objectMappers);
    }

    /**
     * Registers Swagger with context.
     */
    public void registerSwagger(ServletContextHandler context, String swaggerContextPath,
            Collection<ObjectMapper> objectMappers) {
        context.addServlet(new ServletHolder(new SwaggerServlet(swaggerContextPath)), swaggerContextPath);
        for (ObjectMapper objectMapper : objectMappers) {
            configure(objectMapper);
        }
    }

    private static void configure(ObjectMapper objectMapper) {
        objectMapper.registerModule(swaggerSerializationModule());
    }

    private static Module swaggerSerializationModule() {
        SimpleModule module = new SimpleModule("SwaggerJacksonModule");
        module.addSerializer(ApiListing.class, new SwaggerApiListingJsonSerializer());
        module.addSerializer(ResourceListing.class, new SwaggerResourceListingJsonSerializer());
        return module;
    }

}