| /* |
| * 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.syncope.core.provisioning.java.job; |
| |
| import co.elastic.clients.elasticsearch.ElasticsearchClient; |
| import co.elastic.clients.elasticsearch._types.mapping.TypeMapping; |
| import co.elastic.clients.elasticsearch.core.IndexRequest; |
| import co.elastic.clients.elasticsearch.core.IndexResponse; |
| import co.elastic.clients.elasticsearch.indices.IndexSettings; |
| import java.io.IOException; |
| import java.util.Map; |
| import org.apache.syncope.common.lib.types.AnyTypeKind; |
| import org.apache.syncope.core.persistence.api.dao.AnyDAO; |
| import org.apache.syncope.core.persistence.api.dao.AnyObjectDAO; |
| import org.apache.syncope.core.persistence.api.dao.GroupDAO; |
| import org.apache.syncope.core.persistence.api.dao.UserDAO; |
| import org.apache.syncope.core.persistence.api.entity.task.SchedTask; |
| import org.apache.syncope.core.persistence.api.entity.task.TaskExec; |
| import org.apache.syncope.core.spring.security.AuthContextUtils; |
| import org.apache.syncope.ext.elasticsearch.client.ElasticsearchIndexManager; |
| import org.apache.syncope.ext.elasticsearch.client.ElasticsearchUtils; |
| import org.quartz.JobExecutionContext; |
| import org.quartz.JobExecutionException; |
| import org.springframework.beans.factory.annotation.Autowired; |
| |
| /** |
| * Remove and rebuild all Elasticsearch indexes with information from existing users, groups and any objects. |
| */ |
| public class ElasticsearchReindex extends AbstractSchedTaskJobDelegate<SchedTask> { |
| |
| @Autowired |
| protected ElasticsearchClient client; |
| |
| @Autowired |
| protected ElasticsearchIndexManager indexManager; |
| |
| @Autowired |
| protected ElasticsearchUtils utils; |
| |
| @Autowired |
| protected UserDAO userDAO; |
| |
| @Autowired |
| protected GroupDAO groupDAO; |
| |
| @Autowired |
| protected AnyObjectDAO anyObjectDAO; |
| |
| protected IndexSettings userSettings() throws IOException { |
| return indexManager.defaultSettings(); |
| } |
| |
| protected IndexSettings groupSettings() throws IOException { |
| return indexManager.defaultSettings(); |
| } |
| |
| protected IndexSettings anyObjectSettings() throws IOException { |
| return indexManager.defaultSettings(); |
| } |
| |
| protected TypeMapping userMapping() throws IOException { |
| return indexManager.defaultMapping(); |
| } |
| |
| protected TypeMapping groupMapping() throws IOException { |
| return indexManager.defaultMapping(); |
| } |
| |
| protected TypeMapping anyObjectMapping() throws IOException { |
| return indexManager.defaultMapping(); |
| } |
| |
| @Override |
| protected String doExecute(final boolean dryRun, final String executor, final JobExecutionContext context) |
| throws JobExecutionException { |
| |
| if (!dryRun) { |
| LOG.debug("Start rebuilding indexes"); |
| |
| try { |
| indexManager.createIndex( |
| AuthContextUtils.getDomain(), AnyTypeKind.USER, userSettings(), userMapping()); |
| |
| indexManager.createIndex( |
| AuthContextUtils.getDomain(), AnyTypeKind.GROUP, groupSettings(), groupMapping()); |
| |
| indexManager.createIndex( |
| AuthContextUtils.getDomain(), AnyTypeKind.ANY_OBJECT, anyObjectSettings(), anyObjectMapping()); |
| |
| LOG.debug("Indexing users..."); |
| for (int page = 1; page <= (userDAO.count() / AnyDAO.DEFAULT_PAGE_SIZE) + 1; page++) { |
| for (String user : userDAO.findAllKeys(page, AnyDAO.DEFAULT_PAGE_SIZE)) { |
| IndexRequest<Map<String, Object>> request = new IndexRequest.Builder<Map<String, Object>>(). |
| index(ElasticsearchUtils.getContextDomainName( |
| AuthContextUtils.getDomain(), AnyTypeKind.USER)). |
| id(user). |
| document(utils.document(userDAO.find(user), AuthContextUtils.getDomain())). |
| build(); |
| try { |
| IndexResponse response = client.index(request); |
| LOG.debug("Index successfully created for {}: {}", user, response); |
| } catch (Exception e) { |
| LOG.error("Could not create index for {} {}", AnyTypeKind.USER, user, e); |
| } |
| } |
| } |
| |
| LOG.debug("Indexing groups..."); |
| for (int page = 1; page <= (groupDAO.count() / AnyDAO.DEFAULT_PAGE_SIZE) + 1; page++) { |
| for (String group : groupDAO.findAllKeys(page, AnyDAO.DEFAULT_PAGE_SIZE)) { |
| IndexRequest<Map<String, Object>> request = new IndexRequest.Builder<Map<String, Object>>(). |
| index(ElasticsearchUtils.getContextDomainName( |
| AuthContextUtils.getDomain(), AnyTypeKind.GROUP)). |
| id(group). |
| document(utils.document(groupDAO.find(group), AuthContextUtils.getDomain())). |
| build(); |
| try { |
| IndexResponse response = client.index(request); |
| LOG.debug("Index successfully created for {}: {}", group, response); |
| } catch (Exception e) { |
| LOG.error("Could not create index for {} {}", AnyTypeKind.GROUP, group, e); |
| } |
| } |
| } |
| |
| LOG.debug("Indexing any objects..."); |
| for (int page = 1; page <= (anyObjectDAO.count() / AnyDAO.DEFAULT_PAGE_SIZE) + 1; page++) { |
| for (String anyObject : anyObjectDAO.findAllKeys(page, AnyDAO.DEFAULT_PAGE_SIZE)) { |
| IndexRequest<Map<String, Object>> request = new IndexRequest.Builder<Map<String, Object>>(). |
| index(ElasticsearchUtils.getContextDomainName( |
| AuthContextUtils.getDomain(), AnyTypeKind.ANY_OBJECT)). |
| id(anyObject). |
| document(utils.document(anyObjectDAO.find(anyObject), AuthContextUtils.getDomain())). |
| build(); |
| try { |
| IndexResponse response = client.index(request); |
| LOG.debug("Index successfully created for {}: {}", anyObject, response); |
| } catch (Exception e) { |
| LOG.error("Could not create index for {} {}", AnyTypeKind.ANY_OBJECT, anyObject, e); |
| } |
| } |
| } |
| |
| LOG.debug("Rebuild indexes for domain {} successfully completed", AuthContextUtils.getDomain()); |
| } catch (Exception e) { |
| throw new JobExecutionException("While rebuilding index for domain " + AuthContextUtils.getDomain(), e); |
| } |
| } |
| |
| return "SUCCESS"; |
| } |
| |
| @Override |
| protected boolean hasToBeRegistered(final TaskExec<?> execution) { |
| return true; |
| } |
| } |