LogAdminResource.java

package restx.log.admin;

import ch.qos.logback.classic.Level;
import ch.qos.logback.classic.Logger;
import ch.qos.logback.classic.LoggerContext;
import com.google.common.base.Charsets;
import com.google.common.io.Files;
import org.slf4j.ILoggerFactory;
import org.slf4j.LoggerFactory;
import restx.WebException;
import restx.admin.AdminModule;
import restx.annotations.GET;
import restx.annotations.PUT;
import restx.annotations.RestxResource;
import restx.common.MorePreconditions;
import restx.factory.Component;
import restx.security.RolesAllowed;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

/**
 * Date: 17/11/13
 * Time: 10:24
 */
@RestxResource(group = "restx-admin") @Component
public class LogAdminResource {
    public static class Logger {
        public String name;
        public String level;

        public Logger() {
        }

        public Logger(String name, String level) {
            this.name = name;
            this.level = level;
        }
    }

    @RolesAllowed(AdminModule.RESTX_ADMIN_ROLE)
    @GET("/@/logs")
    public String getLogs() {
        // quick and dirty basic implementation to get logs from the default log file.
        // this doesn't scale at all, and is very limited
        try {
            File appLog = new File(System.getProperty("logs.base", "logs"), "app.log");
            String logs = Files.toString(
                    appLog, Charsets.UTF_8);
            // limit to around 30k
            int limit = 30000;
            if (logs.length() > limit) {
                int length = logs.length();
                logs = logs.substring(logs.length() - limit);
                logs = logs.substring(logs.indexOf('\n') + 1);
                logs = "[... " + ((length - logs.length()) / 1024)  + " kB truncated ...]\n" + logs;
            }
            logs = "LOGS FROM: " + appLog.getAbsolutePath() + "\n" +
                    "------------------------------------------------------------------------------------------------\n" +
                    logs;
            return logs;
        } catch (IOException e) {
            return e.getMessage();
        }
    }

    @RolesAllowed(AdminModule.RESTX_ADMIN_ROLE)
    @GET("/@/loggers")
    public Iterable<Logger> getLoggers() {
        ILoggerFactory loggerFactory = LoggerFactory.getILoggerFactory();
        if (loggerFactory instanceof LoggerContext) {
            LoggerContext context = (LoggerContext) loggerFactory;

            List<Logger> loggers = new ArrayList<>();
            for (ch.qos.logback.classic.Logger l : context.getLoggerList()) {
                loggers.add(new Logger(l.getName(), l.getEffectiveLevel().toString()));
            }

            return loggers;
        } else {
            return Collections.emptyList();
        }
    }

    @RolesAllowed(AdminModule.RESTX_ADMIN_ROLE)
    @PUT("/@/loggers/{name}")
    public Logger updateLogger(String name, Logger logger) {
        logger.name = name;

        ILoggerFactory loggerFactory = LoggerFactory.getILoggerFactory();
        if (loggerFactory instanceof LoggerContext) {
            LoggerContext context = (LoggerContext) loggerFactory;
            context.getLogger(name).setLevel(Level.valueOf(logger.level));
        }

        return logger;
    }
}