osam-core/async-jobs/src/main/java/org/onap/osam/job/command/BaseWatchingCommand.java

/*-
 * ============LICENSE_START=======================================================
 * OSAM
 * ================================================================================
 * Copyright (C) 2018 AT&T
 * ================================================================================
 * 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.
 * ============LICENSE_END=========================================================
 */
package org.onap.osam.job.command;

import lombok.extern.slf4j.Slf4j;
import org.onap.osam.job.dao.job.JobStatus;
import org.onap.osam.job.dao.job.OsamJob;
import org.onap.osam.job.IJobCommand;
import org.onap.osam.job.NextCommand;
import org.onap.osam.job.AsyncJobService;
import org.onap.osam.job.impl.JobSharedData;
import org.onap.osam.job.repository.job.OsamJobRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.CollectionUtils;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import java.util.stream.Collectors;

@Slf4j
public abstract class BaseWatchingCommand extends CommandBase implements IJobCommand {

    @Autowired
    protected AsyncJobService asyncInstantiationBL;

    @Autowired
    private OsamJobRepository osamJobRepository;

    private List<UUID> childrenJobsIds;

    protected boolean isRoot;

    public BaseWatchingCommand() {}

    public BaseWatchingCommand(JobSharedData sharedData, List<UUID> childrenJobsIds, boolean isRoot) {
        init(sharedData, childrenJobsIds, isRoot);
    }

    @Override
    public BaseWatchingCommand init(JobSharedData sharedData, Map<String, Object> commandData) {
        return init(
                sharedData,
                ((List<String>) commandData.get("childrenJobs")).stream().map(x -> UUID.fromString(x)).collect(Collectors.toList()),
                (boolean) commandData.get("isRoot")
        );
    }

    protected BaseWatchingCommand init(JobSharedData sharedData, List<UUID> childrenJobsIds, boolean isRoot) {
        super.init(sharedData);
        this.childrenJobsIds = CollectionUtils.isEmpty(childrenJobsIds) ? new ArrayList<>() : childrenJobsIds;
        this.isRoot = isRoot;
        return this;
    }

    @Override
    public NextCommand call() {
        Map<UUID, OsamJob> jobs = getAllChildrenJobs();

        boolean isAllChildrenFinal = true;
        boolean hasFailedChild = false;
        for (UUID jobId: childrenJobsIds) {
            OsamJob job = jobs.get(jobId);


            //if job not found - we assume it failed
            if (job == null || job.getStatus() == null) {
                hasFailedChild = true;
                continue;
            }

            if (!job.getStatus().isFinal()) {
                isAllChildrenFinal = false;
            } else if (!job.getStatus().equals(JobStatus.COMPLETED)) {
                //if job status is final - check if it failed status
                hasFailedChild = true;
            }
        }

        return getNextCommand(isAllChildrenFinal, hasFailedChild);
    }

    private Map<UUID, OsamJob> getAllChildrenJobs() {
        List<OsamJob> jobs = osamJobRepository.findAllByUuid(childrenJobsIds);
        return jobs.stream().collect(Collectors.toMap(OsamJob::getUuid, item -> item));
    }

    protected abstract NextCommand getNextCommand(boolean isAllChildrenFinal, boolean hasFailedChild);


    @Override
    public Map<String, Object> getData() {
        Map<String, Object> data = new HashMap<>();
        data.put("childrenJobs", childrenJobsIds);
        data.put("isRoot", isRoot);
        return data;
    }
}