com.francelabs.datafari.servlets.admin.ZooKeeperConf.java Source code

Introduction

Here is the source code for com.francelabs.datafari.servlets.admin.ZooKeeperConf.java

Source

/*******************************************************************************
 * Copyright 2015 France Labs
 *
 * 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 com.francelabs.datafari.servlets.admin;

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.nio.file.Paths;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.io.FileUtils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.apache.solr.client.solrj.SolrServerException;
import org.json.simple.JSONObject;

import com.francelabs.datafari.exception.CodesReturned;
import com.francelabs.datafari.service.indexer.IndexerServer;
import com.francelabs.datafari.service.indexer.IndexerServerManager;
import com.francelabs.datafari.service.indexer.IndexerServerManager.Core;
import com.francelabs.datafari.servlets.constants.OutputConstants;
import com.francelabs.datafari.utils.Environment;
import com.francelabs.datafari.utils.ExecutionEnvironment;

@WebServlet("/SearchExpert/zookeeperConf")
public class ZooKeeperConf extends HttpServlet {
    private final String env;
    private final String downloadFolder;
    private static final String confname = "FileShare";
    private static final long serialVersionUID = 1L;
    private final static Logger LOGGER = LogManager.getLogger(ZooKeeperConf.class.getName());

    /**
     * @see HttpServlet#HttpServlet()
     */
    public ZooKeeperConf() {
        super();

        String environnement = Environment.getEnvironmentVariable("DATAFARI_HOME");

        if (environnement == null) { // If in development environment
            environnement = ExecutionEnvironment.getDevExecutionEnvironment();
        }
        env = environnement + "/solr/solrcloud/FileShare/conf/";
        downloadFolder = environnement + "/bin/backup/solr";

    }

    /**
     * @throws IOException
     * @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse
     *      response)
     */
    @Override
    protected void doGet(final HttpServletRequest request, final HttpServletResponse response)
            throws ServletException, IOException {
        final JSONObject jsonResponse = new JSONObject();
        request.setCharacterEncoding("utf8");
        response.setContentType("application/json");
        final String actionParam = request.getParameter("action");

        IndexerServer server = null;
        try {
            server = IndexerServerManager.getIndexerServer(Core.FILESHARE);
        } catch (final IOException e1) {
            final PrintWriter out = response.getWriter();
            out.append(
                    "Error while getting the Solr core, please make sure the core dedicated to PromoLinks has booted up. Error code : 69000");
            out.close();
            LOGGER.error(
                    "Error while getting the Solr core in doGet, admin servlet, make sure the core dedicated to Promolink has booted up and is still called promolink or that the code has been changed to match the changes. Error 69000 ",
                    e1);
            return;

        } catch (final Exception e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

        try {
            if (actionParam.toLowerCase().equals("download")) {
                final File folderConf = new File(downloadFolder);
                FileUtils.cleanDirectory(folderConf);
                server.downloadConfig(Paths.get(downloadFolder), Core.FILESHARE.toString());
            } else if (actionParam.toLowerCase().equals("upload")) {
                server.uploadConfig(Paths.get(env), Core.FILESHARE.toString());
            } else if (actionParam.toLowerCase().equals("reload")) {
                server.reloadCollection(Core.FILESHARE.toString());
            } else if (actionParam.toLowerCase().equals("upload_and_reload")) {
                server.uploadConfig(Paths.get(env), Core.FILESHARE.toString());
                Thread.sleep(3000);
                server.reloadCollection(Core.FILESHARE.toString());
            }

            jsonResponse.put(OutputConstants.CODE, CodesReturned.ALLOK.getValue());
        } catch (final IOException | SolrServerException | InterruptedException e) {
            LOGGER.error("Exception during action " + actionParam, e);
            jsonResponse.put(OutputConstants.CODE, CodesReturned.GENERALERROR.getValue());
        }
        final PrintWriter out = response.getWriter();
        out.print(jsonResponse);
    }

    /**
     * @throws IOException
     * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse
     *      response)
     */
    @Override
    protected void doPost(final HttpServletRequest request, final HttpServletResponse response)
            throws ServletException, IOException {
        request.setCharacterEncoding("utf8");
        response.setContentType("application/json");
        final JSONObject jsonResponse = new JSONObject();

        final PrintWriter out = response.getWriter();
        out.print(jsonResponse);
    }
}