src/main/java/org/apache/sling/reqanalyzer/impl/RequestAnalysisLogger.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 org.apache.sling.reqanalyzer.impl;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.util.Hashtable;
import java.util.Iterator;

import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletResponse;

import org.apache.sling.api.SlingHttpServletRequest;
import org.apache.sling.api.SlingHttpServletResponse;
import org.apache.sling.api.wrappers.SlingHttpServletResponseWrapper;
import org.apache.sling.engine.EngineConstants;
import org.apache.sling.settings.SlingSettingsService;
import org.osgi.framework.BundleContext;
import org.osgi.framework.ServiceRegistration;
import org.osgi.service.component.annotations.Activate;
import org.osgi.service.component.annotations.Component;
import org.osgi.service.component.annotations.Deactivate;
import org.osgi.service.component.annotations.Reference;
import org.osgi.service.component.propertytypes.ServiceRanking;

@Component(service = {RequestAnalysisLogger.class},
        property = {
                EngineConstants.FILTER_NAME + "=RequestAnalysisLogger",
                EngineConstants.SLING_FILTER_SCOPE + "=" + EngineConstants.FILTER_SCOPE_REQUEST,
        })
@ServiceRanking(Integer.MAX_VALUE)
public class RequestAnalysisLogger implements Filter {

    @Reference
    private SlingSettingsService settings;

    private BufferedWriter logFile;

    private RequestAnalyzerWebConsole requestAnalyzerWebConsole;
    private ServiceRegistration webConsolePlugin;

    @SuppressWarnings({ "serial" })
    @Activate
    private void activate(final BundleContext ctx) throws IOException {
        final File logFile = new File(settings.getSlingHomePath(), "logs/requesttracker.txt");
        logFile.getParentFile().mkdirs();
        final FileOutputStream out = new FileOutputStream(logFile, true);
        this.logFile = new BufferedWriter(new OutputStreamWriter(out, "UTF-8"));

        this.requestAnalyzerWebConsole = new RequestAnalyzerWebConsole(logFile);
        this.webConsolePlugin = ctx.registerService("javax.servlet.Servlet", this.requestAnalyzerWebConsole,
                new Hashtable<String, Object>() {
                    {
                        put("felix.webconsole.label", "requestanalyzer");
                        put("felix.webconsole.title", "Request Analyzer");
                        put("felix.webconsole.category", "Sling");
                    }
                });
    }

    @Deactivate
    private void deactivate() throws IOException {
        if (this.webConsolePlugin != null) {
            this.webConsolePlugin.unregister();
            this.webConsolePlugin = null;
        }

        if (this.requestAnalyzerWebConsole != null) {
            this.requestAnalyzerWebConsole.dispose();
            this.requestAnalyzerWebConsole = null;
        }

        if (this.logFile != null) {
            this.logFile.close();
            this.logFile = null;
        }
    }

    public void init(FilterConfig filterConfig) throws ServletException {
    }

    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException,
            ServletException {

        if (request instanceof SlingHttpServletRequest) {
            final long start = System.currentTimeMillis();
            final AnylserSlingHttpServletResponse slingRes = new AnylserSlingHttpServletResponse(
                    (SlingHttpServletResponse) response);
            try {
                chain.doFilter(request, response);
            } finally {
                final long end = System.currentTimeMillis();
                final SlingHttpServletRequest slingReq = (SlingHttpServletRequest) request;

                StringBuilder pw = new StringBuilder(1024);
                pw.append(String.format(":%d:%d:%s:%s:%s:%d%n", start, (end - start), slingReq.getMethod(),
                        slingReq.getRequestURI(), slingRes.getContentType(), slingRes.getStatus()));

                final Iterator<String> entries = slingReq.getRequestProgressTracker().getMessages();
                while (entries.hasNext()) {
                    pw.append('!').append(entries.next());
                }

                BufferedWriter out = this.logFile;
                if (out != null) {
                    out.write(pw.toString());
                    out.flush();
                }
            }
        } else {
            chain.doFilter(request, response);
        }
    }

    public void destroy() {
    }

    private static class AnylserSlingHttpServletResponse extends SlingHttpServletResponseWrapper {

        private int status = 200;

        public AnylserSlingHttpServletResponse(SlingHttpServletResponse wrappedResponse) {
            super(wrappedResponse);
        }

        public int getStatus() {
            return status;
        }

        @Override
        public void setStatus(int sc) {
            this.status = sc;
            super.setStatus(sc);
        }

        @Override
        public void setStatus(int sc, String sm) {
            this.status = sc;
            super.setStatus(sc, sm);
        }

        @Override
        public void sendError(int sc) throws IOException {
            this.status = sc;
            super.sendError(sc);
        }

        @Override
        public void sendError(int sc, String msg) throws IOException {
            this.status = sc;
            super.sendError(sc, msg);
        }

        @Override
        public void sendRedirect(String location) throws IOException {
            this.status = HttpServletResponse.SC_FOUND;
            super.sendRedirect(location);
        }
    }
}