| /**************************************************************** |
| * 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.james.modules.mailbox; |
| |
| import static org.apache.james.modules.mailbox.OpenSearchMailboxModule.OPENSEARCH_CONFIGURATION_NAME; |
| |
| import java.io.FileNotFoundException; |
| import java.io.IOException; |
| |
| import javax.inject.Inject; |
| |
| import org.apache.commons.configuration2.Configuration; |
| import org.apache.commons.configuration2.ex.ConfigurationException; |
| import org.apache.james.backends.opensearch.OpenSearchConfiguration; |
| import org.apache.james.backends.opensearch.OpenSearchIndexer; |
| import org.apache.james.backends.opensearch.ReactorOpenSearchClient; |
| import org.apache.james.events.EventListener; |
| import org.apache.james.lifecycle.api.Startable; |
| import org.apache.james.mailbox.quota.QuotaRootResolver; |
| import org.apache.james.quota.search.QuotaSearcher; |
| import org.apache.james.quota.search.opensearch.OpenSearchQuotaConfiguration; |
| import org.apache.james.quota.search.opensearch.OpenSearchQuotaSearcher; |
| import org.apache.james.quota.search.opensearch.QuotaSearchIndexCreationUtil; |
| import org.apache.james.quota.search.opensearch.UserRoutingKeyFactory; |
| import org.apache.james.quota.search.opensearch.events.OpenSearchQuotaMailboxListener; |
| import org.apache.james.quota.search.opensearch.json.QuotaRatioToOpenSearchJson; |
| import org.apache.james.utils.InitializationOperation; |
| import org.apache.james.utils.InitilizationOperationBuilder; |
| import org.apache.james.utils.PropertiesProvider; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| import com.google.inject.AbstractModule; |
| import com.google.inject.Provides; |
| import com.google.inject.Singleton; |
| import com.google.inject.multibindings.Multibinder; |
| import com.google.inject.multibindings.ProvidesIntoSet; |
| |
| public class OpenSearchQuotaSearcherModule extends AbstractModule { |
| |
| static class OpenSearchQuotaIndexCreator implements Startable { |
| private final OpenSearchConfiguration configuration; |
| private final OpenSearchQuotaConfiguration quotaConfiguration; |
| private final ReactorOpenSearchClient client; |
| |
| @Inject |
| OpenSearchQuotaIndexCreator(OpenSearchConfiguration configuration, |
| OpenSearchQuotaConfiguration quotaConfiguration, |
| ReactorOpenSearchClient client) { |
| this.configuration = configuration; |
| this.quotaConfiguration = quotaConfiguration; |
| this.client = client; |
| } |
| |
| void createIndex() throws IOException { |
| QuotaSearchIndexCreationUtil.prepareClient(client, |
| quotaConfiguration.getReadAliasQuotaRatioName(), |
| quotaConfiguration.getWriteAliasQuotaRatioName(), |
| quotaConfiguration.getIndexQuotaRatioName(), |
| configuration); |
| } |
| } |
| |
| private static final Logger LOGGER = LoggerFactory.getLogger(OpenSearchQuotaSearcherModule.class); |
| |
| @Override |
| protected void configure() { |
| Multibinder.newSetBinder(binder(), EventListener.ReactiveGroupEventListener.class) |
| .addBinding() |
| .to(OpenSearchQuotaMailboxListener.class); |
| } |
| |
| @Provides |
| @Singleton |
| public QuotaSearcher provideSearcher(ReactorOpenSearchClient client, OpenSearchQuotaConfiguration configuration) { |
| return new OpenSearchQuotaSearcher(client, |
| configuration.getReadAliasQuotaRatioName()); |
| } |
| |
| @Provides |
| @Singleton |
| private OpenSearchQuotaConfiguration getOpenSearchQuotaConfiguration(PropertiesProvider propertiesProvider) throws ConfigurationException { |
| try { |
| Configuration configuration = propertiesProvider.getConfiguration(OPENSEARCH_CONFIGURATION_NAME); |
| return OpenSearchQuotaConfiguration.fromProperties(configuration); |
| } catch (FileNotFoundException e) { |
| LOGGER.warn("Could not find " + OPENSEARCH_CONFIGURATION_NAME + " configuration file. Providing a default OpenSearchQuotaConfiguration"); |
| return OpenSearchQuotaConfiguration.DEFAULT_CONFIGURATION; |
| } |
| } |
| |
| @Provides |
| @Singleton |
| public OpenSearchQuotaMailboxListener provideListener(ReactorOpenSearchClient client, |
| OpenSearchQuotaConfiguration configuration, |
| QuotaRootResolver quotaRootResolver) { |
| return new OpenSearchQuotaMailboxListener( |
| new OpenSearchIndexer(client, |
| configuration.getWriteAliasQuotaRatioName()), |
| new QuotaRatioToOpenSearchJson(quotaRootResolver), |
| new UserRoutingKeyFactory(), quotaRootResolver); |
| } |
| |
| @ProvidesIntoSet |
| InitializationOperation createIndex(OpenSearchQuotaIndexCreator instance) { |
| return InitilizationOperationBuilder |
| .forClass(OpenSearchQuotaIndexCreator.class) |
| .init(instance::createIndex); |
| } |
| } |