001/*
002 * (C) Copyright 2016 Nuxeo SA (http://nuxeo.com/) and others.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 *     http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 *
016 * Contributors:
017 *     "Guillaume Renard"
018 */
019
020package org.nuxeo.elasticsearch;
021
022import java.io.IOException;
023
024import javax.servlet.Filter;
025import javax.servlet.FilterChain;
026import javax.servlet.FilterConfig;
027import javax.servlet.ServletException;
028import javax.servlet.ServletRequest;
029import javax.servlet.ServletResponse;
030import javax.servlet.http.HttpServletRequest;
031
032import org.nuxeo.elasticsearch.listener.ElasticSearchInlineListener;
033
034/**
035 * @since 8.3
036 */
037public class ElasticSearchFilter implements Filter {
038
039    public static final String ES_SYNC_FLAG = "nx-es-sync";
040
041    protected static final String ES_SYNC_FLAG_COMPAT = "nx_es_sync";
042
043    @Override
044    public void init(FilterConfig filterConfig) throws ServletException {
045    }
046
047    @Override
048    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
049            throws IOException, ServletException {
050        if (!(request instanceof HttpServletRequest)) {
051            chain.doFilter(request, response);
052            return;
053        }
054        final HttpServletRequest httpRequest = (HttpServletRequest) request;
055        final boolean esSync = Boolean.parseBoolean(httpRequest.getHeader(ES_SYNC_FLAG))
056                // NXP-28075: keep compatibility with old header with underscores
057                || Boolean.parseBoolean(httpRequest.getHeader(ES_SYNC_FLAG_COMPAT));
058        if (!esSync) {
059            chain.doFilter(request, response);
060            return;
061        }
062        ElasticSearchInlineListener.useSyncIndexing.set(true);
063        try {
064            chain.doFilter(request, response);
065        } finally {
066            ElasticSearchInlineListener.useSyncIndexing.set(false);
067        }
068    }
069
070    @Override
071    public void destroy() {
072    }
073
074}