All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • com.floragunn.searchguard.configuration.ConfigurationLoader Maven / Gradle / Ivy

    There is a newer version: 5.0.0-rc1-6d
    Show newest version
    /*
 * Copyright 2015 floragunn UG (haftungsbeschränkt)
 * 
 * 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.floragunn.searchguard.configuration;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.TimeUnit;

import org.elasticsearch.ElasticsearchException;
import org.elasticsearch.ExceptionsHelper;
import org.elasticsearch.action.ActionListener;
import org.elasticsearch.action.get.GetResponse;
import org.elasticsearch.action.get.MultiGetItemResponse;
import org.elasticsearch.action.get.MultiGetRequest;
import org.elasticsearch.action.get.MultiGetResponse;
import org.elasticsearch.client.Client;
import org.elasticsearch.common.bytes.BytesReference;
import org.elasticsearch.common.inject.Inject;
import org.elasticsearch.common.inject.Provider;
import org.elasticsearch.common.logging.ESLogger;
import org.elasticsearch.common.logging.Loggers;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.common.settings.loader.JsonSettingsLoader;
import org.elasticsearch.common.xcontent.XContentHelper;

import com.floragunn.searchguard.support.ConfigConstants;

public class ConfigurationLoader {

    protected final ESLogger log = Loggers.getLogger(this.getClass());
    private final Provider client;

    @Inject
    public ConfigurationLoader(final Provider client) {
        super();
        this.client = client;
    }

    public Map load(final String[] events) {

        final Map rs = new HashMap(events.length);
        final BlockingQueue queue = new ArrayBlockingQueue(events.length);
        final MultiGetRequest mget = new MultiGetRequest();

        for (int i = 0; i < events.length; i++) {
            final String event = events[i];
            mget.add("searchguard", event, "0");
        }

        mget.putHeader(ConfigConstants.SG_CONF_REQUEST_HEADER, "true"); //header needed here
        mget.refresh(true);
        mget.realtime(true);

        client.get().multiGet(mget, new ActionListener() {

            @Override
            public void onResponse(final MultiGetResponse mresponse) {

                final MultiGetItemResponse[] mres = mresponse.getResponses();

                for (int i = 0; i < mres.length; i++) {
                    final GetResponse response = mres[i].getResponse();
                    if (response == null) {
                        try {
                            queue.put("failure " + mres[i].getType() + " " + mres[i].getFailure().getMessage());
                        } catch (final InterruptedException e) {
                            Thread.currentThread().interrupt();
                        }
                    } else

                    if (response.isExists() && !response.isSourceEmpty()) {
                            try {
                                queue.put(response);
                            } catch (final InterruptedException e) {
                                Thread.currentThread().interrupt();
                            }
                        } else {
                            try {
                                queue.put(response.getType());
                            } catch (final InterruptedException e) {
                                Thread.currentThread().interrupt();
                            }
                        }
                }
            }

            @Override
            public void onFailure(final Throwable e) {
                try {
                    queue.put(e);
                } catch (final InterruptedException e1) {
                    Thread.currentThread().interrupt();
                }
            }
        });

        Object response = null;
        try {
            response = queue.poll(10, TimeUnit.SECONDS);

            if (queue.size() == 0 && response != null && response instanceof Throwable) {
                throw ExceptionsHelper.convertToElastic((Throwable) response);
            }

            if (response instanceof GetResponse && response != null) {
                final GetResponse gs = (GetResponse) response;

                if (gs.isExists() && !gs.isSourceEmpty()) {
                    rs.put(gs.getType(), toSettings(gs.getSourceAsBytesRef()));
                }

            } else {
                if(response != null && response.toString().contains("fail")) {
                    log.debug("Cannot retrieve {}", response);
                } else {
                    log.debug("Cannot retrieve {}", response);
                    //log.error("Cannot retrieve {}", response);
                }
            }

            for (int i = 0; i < events.length - 1; i++) {
                response = queue.poll(10, TimeUnit.SECONDS);
                if (response instanceof GetResponse && response != null) {
                    final GetResponse gs = (GetResponse) response;

                    if (gs.isExists() && !gs.isSourceEmpty()) {
                        rs.put(gs.getType(), toSettings(gs.getSourceAsBytesRef()));
                    }

                } else {
                    if(response != null && response.toString().contains("fail")) {
                        log.debug("Cannot retrieve {}", response);
                    } else {
                        log.debug("Cannot retrieve {}", response);
                        //log.error("Cannot retrieve {}", response);
                    }
                }
            }

        } catch (final InterruptedException e1) {
            Thread.currentThread().interrupt();
            throw ExceptionsHelper.convertToElastic(e1);
        }

        return rs;
    }

    private static Settings toSettings(final BytesReference ref) {
        if (ref == null || ref.length() == 0) {
            throw new ElasticsearchException("ref invalid");
        }

        try {
            return Settings.builder().put(new JsonSettingsLoader().load(XContentHelper.createParser(ref))).build();
        } catch (final IOException e) {
            throw ExceptionsHelper.convertToElastic(e);
        }
    }

}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy