org.jboss.hal.meta.processing.RrdTask Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hal-meta Show documentation
Metadata and registries
The newest version!
/*
 *  Copyright 2022 Red Hat
 *
 *  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
 *
 *      https://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.jboss.hal.meta.processing;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

import org.jboss.hal.config.Environment;
import org.jboss.hal.config.Settings;
import org.jboss.hal.dmr.Composite;
import org.jboss.hal.dmr.CompositeResult;
import org.jboss.hal.dmr.Operation;
import org.jboss.hal.dmr.dispatch.Dispatcher;
import org.jboss.hal.flow.Flow;
import org.jboss.hal.flow.Task;
import org.jboss.hal.meta.StatementContext;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.collect.Lists;

import elemental2.promise.Promise;

import static java.util.stream.Collectors.toList;

/** Creates, executes and parses the {@code read-resource-description} operations to read the missing metadata. */
final class RrdTask implements Task {

    private static final Logger logger = LoggerFactory.getLogger(RrdTask.class);

    private final Dispatcher dispatcher;
    private final int batchSize;
    private final CreateRrdOperations rrdOps;

    RrdTask(Environment environment, Dispatcher dispatcher, StatementContext statementContext, Settings settings,
            int batchSize, int depth) {
        this.dispatcher = dispatcher;
        this.batchSize = batchSize;
        this.rrdOps = new CreateRrdOperations(environment, statementContext, settings.get(Settings.Key.LOCALE).value(),
                depth);
    }

    @Override
    public Promise apply(final LookupContext context) {
        boolean recursive = context.recursive;
        List> tasks = new ArrayList<>();

        // create and partition non-optional operations
        List operations = rrdOps.create(context, recursive, false);
        List> piles = Lists.partition(operations, batchSize);
        List composites = piles.stream().map(Composite::new).collect(toList());
        for (Composite composite : composites) {
            tasks.add((LookupContext c) -> dispatcher.execute(composite).then(result -> {
                parseRrdAction(context, composite, result);
                return Promise.resolve(c);
            }));
        }

        // create optional operations w/o partitioning!
        List optionalOperations = rrdOps.create(context, recursive, true);
        // Do not refactor to
        // List optionalComposites = optionalOperations.stream().map(Composite::new).collect(toList());
        // the GWT compiler will crash with an ArrayIndexOutOfBoundsException!
        List optionalComposites = new ArrayList<>();
        optionalOperations.forEach(operation -> optionalComposites.add(new Composite(operation)));
        for (Composite composite : optionalComposites) {
            tasks.add((LookupContext c) -> dispatcher.execute(composite)
                    .then(result -> {
                        parseRrdAction(context, composite, result);
                        return Promise.resolve(c);
                    })
                    .catch_(error -> {
                        logger.debug("Ignore errors on optional resource operation {}", composite.asCli());
                        return Promise.resolve(c);
                    }));
        }

        if (!tasks.isEmpty()) {
            if (logger.isDebugEnabled()) {
                logger.debug("About to execute {} ({}+{}) composite operations (regular+optional)",
                        composites.size() + optionalComposites.size(), composites.size(), optionalComposites.size());
                String compositeOps = composites.stream().map(Composite::asCli).collect(Collectors.joining(", "));
                logger.debug("Composite operations: {}", compositeOps);
                if (!optionalComposites.isEmpty()) {
                    String optionalOps = optionalComposites.stream()
                            .map(Composite::asCli)
                            .collect(Collectors.joining(", "));
                    logger.debug("Optional operations: {}", optionalOps);
                }
            }
            return Flow.sequential(context, tasks).promise();
        } else {
            logger.debug("No DMR operations necessary");
            return Promise.resolve(context);
        }
    }

    private void parseRrdAction(LookupContext context, Composite composite, CompositeResult compositeResult) {
        RrdResult rrdResult = new CompositeRrdParser(composite).parse(compositeResult);
        context.toResourceDescriptionRegistry.putAll(rrdResult.resourceDescriptions);
        context.toResourceDescriptionDatabase.putAll(rrdResult.resourceDescriptions);
        context.toSecurityContextRegistry.putAll(rrdResult.securityContexts);
        context.toSecurityContextDatabase.putAll(rrdResult.securityContexts);
    }
}