org.voltdb.iv2.MpRepairTask Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of voltdb Show documentation
VoltDB server libraries
There is a newer version: 10.1.1
/* This file is part of VoltDB.
 * Copyright (C) 2008-2020 VoltDB Inc.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * 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
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with VoltDB.  If not, see .
 */

package org.voltdb.iv2;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CancellationException;

import org.voltcore.logging.VoltLogger;
import org.voltcore.utils.CoreUtils;
import org.voltdb.SiteProcedureConnection;
import org.voltdb.VoltDB;
import org.voltdb.VoltZK;
import org.voltdb.rejoin.TaskLog;

import com.google_voltpatches.common.base.Suppliers;

/**
 * This task runs on the MPI's site threads after a topology change which doesn't
 * kill the current master MPI.
 *
 * Must run the repair while pausing the site task queue;
 * Otherwise, a new MP might immediately be blocked in a
 * confused world of semi-repair. So just do the repair
 * work on the site thread....
 *
 * When repairing while the MPI is executing concurrent reads,
 * we need to ensure that the full repair algorithm only runs on one of
 * the in-use RO MPI site threads.  That site will perform all of the message collection
 * and delivery.  All of the other MP RO sites need to block at this point.
 */
public class MpRepairTask extends SiteTasker
{
    static VoltLogger repairLogger = new VoltLogger("REPAIR");

    private InitiatorMailbox m_mailbox;
    private List m_spMasters;
    private Object m_lock = new Object();
    private boolean m_repairRan = false;
    private final String whoami;
    private final RepairAlgo algo;

    // Indicate if this repair is triggered via partition leader migration
    private final boolean m_leaderMigration;

    // Indicate if the round of leader promotion has been completed
    private final boolean m_partitionLeaderPromotionComplete;
    public MpRepairTask(InitiatorMailbox mailbox, List spMasters, boolean leaderMigration,
            boolean partitionLeaderPromotionComplete, boolean skipRepair)
    {
        m_mailbox = mailbox;
        m_spMasters = new ArrayList(spMasters);
        whoami = "MP leader repair " +
                CoreUtils.hsIdToString(m_mailbox.getHSId()) + " ";
        m_leaderMigration = leaderMigration;
        m_partitionLeaderPromotionComplete = partitionLeaderPromotionComplete;
        algo = mailbox.constructRepairAlgo(Suppliers.ofInstance(m_spMasters), Integer.MAX_VALUE, whoami, skipRepair);
    }

    @Override
    public void run(SiteProcedureConnection connection) {

        // When MP is processing reads, the task will be queued to all MpRoSite but the task is processed only on one of MpRoSite.
        synchronized (m_lock) {
            if (!m_repairRan) {
                try {
                    try {
                        algo.start().get();
                        repairLogger.info(whoami + "finished repair.");
                    } catch (CancellationException e) {
                        repairLogger.info(whoami + "interrupted during repair.  Retrying.");
                    }
                } catch (InterruptedException ie) {
                } catch (Exception e) {
                    VoltDB.crashLocalVoltDB("Terminally failed MPI repair.", true, e);
                } finally {
                    m_repairRan = true;

                    // At this point, all the repairs are completed. This should be the final repair task
                    // in the repair process. Remove the mp repair blocker
                    if (!m_leaderMigration && m_partitionLeaderPromotionComplete && m_mailbox.m_messenger != null) {
                        VoltZK.removeActionBlocker(m_mailbox.m_messenger.getZK(), VoltZK.mpRepairInProgress, repairLogger);
                    }
                }
            }
        }
    }

    @Override
    public void runForRejoin(SiteProcedureConnection siteConnection, TaskLog taskLog)
    throws IOException
    {
        throw new RuntimeException("Rejoin while repairing the MPI should be impossible.");
    }
}