org.graylog2.indexer.migration.RemoteReindexMigration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of graylog2-server Show documentation
Graylog
There is a newer version: 6.1.4
/*
 * Copyright (C) 2020 Graylog, Inc.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the Server Side Public License, version 1,
 * as published by MongoDB, Inc.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * Server Side Public License for more details.
 *
 * You should have received a copy of the Server Side Public License
 * along with this program. If not, see
 * .
 */
package org.graylog2.indexer.migration;

import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonProperty;
import io.jsonwebtoken.lang.Collections;
import jakarta.validation.constraints.NotNull;
import org.graylog2.indexer.datanode.RemoteReindexingMigrationAdapter.Status;

import java.util.Comparator;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Collectors;

/**
 * Caution: this object will be heavily mutated from outside as the migration progresses.
 */
@JsonInclude(JsonInclude.Include.NON_NULL)
public class RemoteReindexMigration {

    @JsonProperty("id")
    private final String id;
    private final List indices;

    private final List logs;

    @JsonProperty("error")
    private String error;

    public RemoteReindexMigration(@NotNull String migrationID, List indices, List logs) {
        this.id = migrationID;
        this.indices = indices;
        this.logs = logs;
    }

    public static RemoteReindexMigration nonExistent(String migrationID) {
        return new RemoteReindexMigration(migrationID, Collections.emptyList(), Collections.emptyList());
    }

    @JsonProperty("indices")
    public List indices() {
        return indices;
    }

    public String id() {
        return id;
    }

    @JsonProperty("status")
    public Status status() {
        if (indices.isEmpty()) {
            return Status.NOT_STARTED;
        }
        if (indices().stream().map(RemoteReindexIndex::status).anyMatch(i -> i == Status.RUNNING)) {
            return Status.RUNNING;
        }
        if (indices().stream().map(RemoteReindexIndex::status).anyMatch(i -> i == Status.ERROR)) {
            return Status.ERROR;
        }
        return Status.FINISHED;
    }

    /**
     * @return How much of the migration is done, in percent, int value between 0 a 100.
     */
    @JsonProperty("progress")
    public int progress() {
        if (indices.isEmpty()) {
            return 100; // avoid division by zero. No indices == migration is immediately done
        }

        final double indexPortion = 100.0 / indices.size();

        final double overallProgress = indices.stream()
                .filter(i -> i.progress() != null)
                .mapToDouble(i -> i.progress().progressPercent() / 100.0)
                .map(relativeProgress -> relativeProgress * indexPortion)
                .sum();

         return Math.min((int)Math.ceil(overallProgress), 100);
    }

    public List getLogs() {
        return Optional.ofNullable(logs).map(l -> l.stream().toList()).orElse(Collections.emptyList());
    }

    @JsonProperty("tasks_progress")
    public Map getTasksProgress() {
        return indices.stream()
                .filter(i -> i.status() == Status.RUNNING)
                .filter(i -> i.progress() != null)
                .sorted(Comparator.comparing(RemoteReindexIndex::name))
                .collect(Collectors.toMap(RemoteReindexIndex::name, i -> i.progress().progressPercent(), (integer, integer2) -> integer, LinkedHashMap::new));
    }
}