io.soabase.core.rest.LoggingApis Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of soabase-core Show documentation
The newest version!
/**
 * Copyright 2014 Jordan Zimmerman
 *
 * 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.
 */
package io.soabase.core.rest;

import ch.qos.logback.classic.Level;
import ch.qos.logback.classic.LoggerContext;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.google.common.collect.Lists;
import io.soabase.core.features.logging.LoggingFile;
import io.soabase.core.features.logging.LoggingReader;
import io.soabase.core.rest.entities.LogLevel;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.inject.Inject;
import javax.ws.rs.Consumes;
import javax.ws.rs.GET;
import javax.ws.rs.PUT;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import java.util.List;

@Path("/soa/logging")
public class LoggingApis
{
    private final Logger log = LoggerFactory.getLogger(getClass());
    private final LoggingReader loggingReader;

    @Inject
    public LoggingApis(LoggingReader loggingReader)
    {
        this.loggingReader = loggingReader;
    }

    @GET
    @Path("files")
    @Produces(MediaType.APPLICATION_JSON)
    public List listLoggingFiles()
    {
        return loggingReader.listLoggingFiles();
    }

    @GET
    @Path("file/gzip/{key}")
    @Produces(MediaType.TEXT_PLAIN)
    public Response getFileZipped(@PathParam("key") String key)
    {
        Object entity = loggingReader.keyToEntity(key, true);
        if ( entity == null )
        {
            return Response.status(Response.Status.NOT_FOUND).build();
        }
        return Response.ok(entity).encoding("gzip").build();
    }

    @GET
    @Path("file/raw/{key}")
    @Produces(MediaType.TEXT_PLAIN)
    public Response getFile(@PathParam("key") String key)
    {
        Object entity = loggingReader.keyToEntity(key, false);
        if ( entity == null )
        {
            return Response.status(Response.Status.NOT_FOUND).build();
        }
        return Response.ok(entity).build();
    }

    @GET
    @Path("levels")
    @Produces(MediaType.APPLICATION_JSON)
    public List getLogLevels() throws JsonProcessingException
    {
        List levels = Lists.newArrayList();
        for ( ch.qos.logback.classic.Logger logger : getLoggerContext().getLoggerList() )
        {
            levels.add(new LogLevel(logger.getName(), (logger.getLevel() != null) ? logger.getLevel().toString() : null));
        }
        return levels;
    }

    @PUT
    @Path("levels")
    @Consumes(MediaType.APPLICATION_JSON)
    public Response setLogLevels(List levels) throws JsonProcessingException
    {
        LoggerContext loggerContext = getLoggerContext();
        for ( LogLevel logLevel : levels )
        {
            ch.qos.logback.classic.Logger logger = loggerContext.getLogger(logLevel.getName());
            if ( logger != null )
            {
                Level newLevel = Level.toLevel(logLevel.getLevel(), null);
                logger.setLevel(newLevel);
                log.info(String.format("Setting log level. Name (%s) - Level (%s)", logLevel.getName(), logLevel.getLevel()));
            }
            else
            {
                log.warn("Logger not found for: " + logLevel.getName());
            }
        }
        return Response.ok().build();
    }

    private static LoggerContext getLoggerContext()
    {
        return (LoggerContext)LoggerFactory.getILoggerFactory();
    }
}